From d1e7dccfaeae832dde072ed0759f3cb17217d342 Mon Sep 17 00:00:00 2001
From: Mylloon <kennel.anri@tutanota.com>
Date: Mon, 23 Aug 2021 12:22:31 +0200
Subject: [PATCH] add verbose option

---
 main.py | 36 ++++++++++++++++++++++++------------
 1 file changed, 24 insertions(+), 12 deletions(-)

diff --git a/main.py b/main.py
index db760b3..b12e1b2 100644
--- a/main.py
+++ b/main.py
@@ -4,38 +4,50 @@ from dotenv import load_dotenv
 from cloudscraper import CloudScraper
 
 class Scraper:
-    def __init__(self, pseudo, password, app):
+    def __init__(self, pseudo, password, app, debug = False):
+        self.debug = debug
         self.url = "https://forum.mobilism.org"
         self.requested_app = app
         self.loginData = {
             "username": pseudo,
             "password": password,
             "login": "Login",
-            "redirect": "./ucp.php?mode=login",
+            # "redirect": "./ucp.php?mode=login",
             "redirect": "index.php"
         }
 
+    def errorFormat(self, code: int) -> str:
+        return f"\Error: [{code}]"
+
     def connect(self):
-        with CloudScraper(browser = {"browser": "chrome", "platform": "windows"}, debug = True) as session:
+        with CloudScraper(browser = {"browser": "chrome", "platform": "windows"}) as session:
+            if self.debug: print("Retrieval of the login cookie...", end = " ")
             reponse = session.get(f"{self.url}/ucp.php?mode=login")
             if reponse.status_code != 200:
-                raise ConnectionRefusedError(f"[{reponse.status_code}]")
+                raise ConnectionRefusedError(self.errorFormat(reponse.status_code))
             self.loginData["sid"] = reponse.cookies.get_dict()["ppcw_29d3s_sid"]
+            if self.debug: print("Cookie retrieval done, connection attempt...", end = " ")
             reponse = session.post(f"{self.url}/ucp.php?mode=login", data = self.loginData)
             if reponse.status_code != 200:
-                raise ConnectionRefusedError(f"[{reponse.status_code}]")
-            print(self.loginData)
-            print(reponse.status_code, reponse.url)
+                raise ConnectionRefusedError(self.errorFormat(reponse.status_code))
+            if self.debug: print("Connection done.")
+            if self.debug: print(reponse.status_code, reponse.url)
             with open("temp.html", "w") as f:
                 f.writelines(reponse.text)
+            link = "No link for your application was found."
+            return link
 
 if __name__ == "__main__":
     argv = argv[1:]
     if len(argv) == 3:
         print(Scraper(*argv).connect())
     else:
-        #try:
-        load_dotenv()
-        print(Scraper(environ["PSEUDO_MOBILISM"], environ["PASSWORD_MOBILISM"], environ["APP_MOBILISM"]).connect())
-        #except:
-        #    print('Please fill in the username and password (with ") by args or with .env file.')
+        try:
+            load_dotenv()
+            try:
+                debug = environ["DEBUG_MOBILISM"].lower() in ("yes", "true", "1")
+            except:
+                debug = False
+            print(Scraper(environ["PSEUDO_MOBILISM"], environ["PASSWORD_MOBILISM"], environ["APP_MOBILISM"], debug).connect())
+        except:
+            print('Please fill in the username and password (with ") by args or with .env file.')