From a4772a94c6cebcaeabf64af4e70eb32140a98dad Mon Sep 17 00:00:00 2001
From: Mylloon <kennel.anri@tutanota.com>
Date: Mon, 23 Aug 2021 12:27:11 +0200
Subject: [PATCH] create methods

---
 main.py | 50 +++++++++++++++++++++++++++++++-------------------
 1 file changed, 31 insertions(+), 19 deletions(-)

diff --git a/main.py b/main.py
index b12e1b2..102c8ad 100644
--- a/main.py
+++ b/main.py
@@ -1,7 +1,7 @@
 from sys import argv
 from os import environ
 from dotenv import load_dotenv
-from cloudscraper import CloudScraper
+from cloudscraper import create_scraper
 
 class Scraper:
     def __init__(self, pseudo, password, app, debug = False):
@@ -20,27 +20,39 @@ class Scraper:
         return f"\Error: [{code}]"
 
     def connect(self):
-        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(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(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
+        session = create_scraper(browser = {"browser": "chrome", "platform": "windows"})
+        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(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(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)
+
+        return session
+
+    def search(self, session):
+
+        link = "No link for your application was found."
+        return link
+
+    def work(self):
+        session = self.connect()
+
+        link = self.search(session)
+
+        return link
 
 if __name__ == "__main__":
     argv = argv[1:]
     if len(argv) == 3:
-        print(Scraper(*argv).connect())
+        print(Scraper(*argv).work())
     else:
         try:
             load_dotenv()
@@ -48,6 +60,6 @@ if __name__ == "__main__":
                 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())
+            print(Scraper(environ["PSEUDO_MOBILISM"], environ["PASSWORD_MOBILISM"], environ["APP_MOBILISM"], debug).work())
         except:
             print('Please fill in the username and password (with ") by args or with .env file.')