adds postinit

WolfgangFahl · Oct 23, 2024 · eef7363 · eef7363
1 parent 204b88b
commit eef7363
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 2 deletions.
diff --git a/tests/test_wikiuser.py b/tests/test_wikiuser.py
@@ -65,7 +65,9 @@ def testCommandLine(self):
                 ]
                 wikibot3rd.wikiuser_cmd.main(args)
         finally:
-            if self.debug:
+            debug=self.debug
+            debug=True
+            if debug:
                 print(open(path, "r").read())
             props = WikiUser.readPropertyFile(path)
             rUser = WikiUser.ofDict(props, encrypted=True)

diff --git a/wikibot3rd/wikiclient.py b/wikibot3rd/wikiclient.py
@@ -236,6 +236,7 @@ def of_wiki_id(
         """
         wiki_user = WikiUser.ofWikiId(wiki_id, lenient=lenient)
         wikibot = WikiClient(wiki_user, debug=debug)
+        wikibot.is_smw_enabled=wiki_user.is_smw
         return wikibot
 
     @staticmethod
@@ -257,6 +258,7 @@ def of_wiki_user(wiki_user: WikiUser,debug:bool=False) -> "WikiClient":
             WikiClient: A WikiClient instance for the given WikiUser.
         """
         wikibot = WikiClient(wiki_user,debug=debug)
+        wikibot.is_smw_enabled=wiki_user.is_smw
         return wikibot
 
     @staticmethod

diff --git a/wikibot3rd/wikiuser.py b/wikibot3rd/wikiuser.py
@@ -98,6 +98,16 @@ class WikiUserData(WikiCredentials):
     email: str = None
     is_smw: bool = True
 
+    def __post_init__(self):
+        WikiCredentials.__post_init__(self)
+        if isinstance(self.is_smw, str):
+            normalized_value = self.is_smw.strip().lower()
+            if normalized_value in {"yes", "y", "true"}:
+                self.is_smw = True
+            elif normalized_value in {"no", "n", "false"}:
+                self.is_smw = False
+            else:
+                raise ValueError(f"Invalid value for is_smw: {self.is_smw}")
 
 @dataclass
 class WikiUser(WikiUserData):
@@ -294,7 +304,8 @@ def ofDict(cls, userDict: dict, encrypted=True, lenient=False, encrypt=True):
         WikiUser: the WikiUser created from the dictionary
         """
         if "url" in userDict and userDict["url"] is not None:
-            userDict["url"] = userDict["url"].replace("\:", ":")
+            # fix broken escapes e.g. http\://waihekepedia.bitplan.com
+            userDict["url"] = userDict["url"].replace(r"\:", ":")
 
         is_encrypted = "password" not in userDict