fix(captcha): add captcha support

.gitignore

@@ -165,3 +165,6 @@ poetry.toml
 
 # ruff
 .ruff_cache/
+
+captcha.png
+session

src/TSRDownload.py

@@ -1,4 +1,5 @@
 import requests, time, json, os
+from TSRSession import TSRSession
 from TSRUrl import TSRUrl
 from logger import Logger
 from exceptions import *
@@ -11,8 +12,10 @@
 
 class TSRDownload:
     @classmethod
-    def __init__(self, url: TSRUrl):
+    def __init__(self, url: TSRUrl, session: TSRSession):
         self.session: requests.Session = requests.Session()
+        self.session.cookies.set("tsrdlsession", session.tsrdlsession)
+
         self.url: TSRUrl = url
         self.ticketInitializedTime: float = -1.0
         self.__getTSRDLTicketCookie()

src/TSRSession.py

@@ -0,0 +1,79 @@
+import requests, webbrowser, os
+from exceptions import InvalidCaptchaCode
+from typing import Optional
+
+CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
+
+
+class TSRSession:
+    @classmethod
+    def __init__(self, sessionId: Optional[str] = None) -> None:
+        self.session = requests.Session()
+        if sessionId is not None:
+            if self.__isValidSessionId(sessionId):
+                self.tsrdlsession = sessionId
+                return
+
+        self.__getTSRDLTicketCookie()
+        self.__saveCaptchaImage()
+        self.__openImageInBrowser()
+        self.tsrdlsession = None
+
+        print("Please enter captcha code:")
+        captchaInput = input(">> ")
+        if self.__tryCaptchaCode(captchaInput):
+            self.tsrdlsession = self.session.cookies.get_dict().get("tsrdlsession")
+        else:
+            raise InvalidCaptchaCode
+
+    @classmethod
+    def __tryCaptchaCode(self, code: str) -> bool:
+        r = self.session.post(
+            "https://www.thesimsresource.com/downloads/session/itemId/1646133",
+            data={"captchavalue": code},
+            headers={
+                "Content-Type": "application/x-www-form-urlencoded",
+                "Origin": "https://www.thesimsresource.com",
+            },
+            allow_redirects=True,
+        )
+        return (
+            r.url == "https://www.thesimsresource.com/downloads/download/itemId/1646133"
+        )
+
+    @classmethod
+    def __isValidSessionId(self, sessionId: str) -> bool:
+        self.__getTSRDLTicketCookie()
+        r = self.session.get(
+            "https://www.thesimsresource.com/downloads/download/itemId/1646133",
+            cookies={"tsrdlsession": sessionId},
+        )
+        return (
+            r.url == "https://www.thesimsresource.com/downloads/download/itemId/1646133"
+        )
+
+    @classmethod
+    def __getCaptchaImage(self) -> requests.Request:
+        self.session.get(
+            "https://www.thesimsresource.com/downloads/session/itemId/1646133"
+        )
+        return self.session.get(
+            "https://www.thesimsresource.com/downloads/captcha-image"
+        )
+
+    @classmethod
+    def __saveCaptchaImage(self):
+        with open(f"{CURRENT_DIR}/captcha.png", "wb") as f:
+            for chunk in self.__getCaptchaImage().iter_content(1024 * 1024):
+                f.write(chunk)
+
+    @classmethod
+    def __openImageInBrowser(self) -> None:
+        webbrowser.open_new_tab(f"{CURRENT_DIR}/captcha.png")
+
+    @classmethod
+    def __getTSRDLTicketCookie(self) -> str:
+        response = self.session.get(
+            f"https://www.thesimsresource.com/ajax.php?c=downloads&a=initDownload&itemid=1646133&format=zip"
+        )
+        return response.cookies.get("tsrdlticket")

src/exceptions.py

@@ -10,3 +10,7 @@ class InvalidDownloadTicket(Exception):
     def __init__(self, url: str, cookies: RequestsCookieJar):
         self.url = url
         self.cookies = cookies
+
+
+class InvalidCaptchaCode(Exception):
+    pass

src/main.py

@@ -4,11 +4,12 @@
 from exceptions import *
 from typings import *
 from multiprocessing import Pool
+from TSRSession import TSRSession
 import clipboard, time, json, os
 
 
 def processTarget(url: TSRUrl):
-    downloader = TSRDownload(url)
+    downloader = TSRDownload(url, session)
     if downloader.download():
         Logger.info(f"Completed download for: {url.url}")
 
@@ -22,13 +23,25 @@ def callback(url: TSRUrl):
 
 
 if __name__ == "__main__":
-    CONFIG: CONFIG_DICT = json.load(
-        open(os.path.dirname(os.path.abspath(__file__)) + "/config.json", "r")
-    )
+    CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
+    CONFIG: CONFIG_DICT = json.load(open(CURRENT_DIR + "/config.json", "r"))
     lastPastedText = ""
     runningDownloads: list[str] = []
     downloadQueue: list[str] = []
 
+    session = None
+    sessionId = None
+    if os.path.exists(CURRENT_DIR + "/session"):
+        sessionId = open(CURRENT_DIR + "/session", "r").read()
+
+    while session is None:
+        try:
+            session = TSRSession(sessionId)
+            if hasattr(session, "tsrdlsession"):
+                open(CURRENT_DIR + "/session", "w").write(session.tsrdlsession)
+        except InvalidCaptchaCode:
+            sessionId = None
+
     while True:
         pastedText = clipboard.paste()
         if lastPastedText == pastedText: