1 files changed, 40 insertions, 35 deletions

diff --git a/testing/poe_account_create_test.py b/testing/poe_account_create_test.py
index 5d435b1f..7072597f 100644
--- a/testing/poe_account_create_test.py
+++ b/testing/poe_account_create_test.py
@@ -1,27 +1,28 @@
 from hashlib import md5
 from json import dumps
 from re import findall
+from typing import Optional
 
 from tls_client import Session as TLS
 from twocaptcha import TwoCaptcha
 
-from quora import extract_formkey
-from quora.mail import Emailnator
+from openai_rev.quora import extract_formkey
+from openai_rev.quora.mail import Emailnator
 
 solver = TwoCaptcha('72747bf24a9d89b4dcc1b24875efd358')
 
 
 class Account:
-    def create(proxy: None or str = None, logging: bool = False, enable_bot_creation: bool = False):
+    @staticmethod
+    def create(proxy: Optional[str] = None, logging: bool = False, enable_bot_creation: bool = False):
         client = TLS(client_identifier='chrome110')
-        client.proxies = {
-            'http': f'http://{proxy}',
-            'https': f'http://{proxy}'} if proxy else None
+        client.proxies = {'http': f'http://{proxy}', 'https': f'http://{proxy}'} if proxy else None
 
         mail_client = Emailnator()
         mail_address = mail_client.get_mail()
 
-        if logging: print('email', mail_address)
+        if logging:
+            print('email', mail_address)
 
         client.headers = {
             'authority': 'poe.com',
@@ -39,29 +40,30 @@ class Account:
             'sec-fetch-dest': 'empty',
             'sec-fetch-mode': 'cors',
             'sec-fetch-site': 'same-origin',
-            'user-agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36'
+            'user-agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36',
         }
 
         client.headers["poe-formkey"] = extract_formkey(client.get('https://poe.com/login').text)
         client.headers["poe-tchannel"] = client.get('https://poe.com/api/settings').json()['tchannelData']['channel']
 
         # token = reCaptchaV3('https://www.recaptcha.net/recaptcha/enterprise/anchor?ar=1&k=6LflhEElAAAAAI_ewVwRWI9hsyV4mbZnYAslSvlG&co=aHR0cHM6Ly9wb2UuY29tOjQ0Mw..&hl=en&v=4PnKmGB9wRHh1i04o7YUICeI&size=invisible&cb=bi6ivxoskyal')
-        token = solver.recaptcha(sitekey='6LflhEElAAAAAI_ewVwRWI9hsyV4mbZnYAslSvlG',
-                                 url='https://poe.com/login?redirect_url=%2F',
-                                 version='v3',
-                                 enterprise=1,
-                                 invisible=1,
-                                 action='login', )['code']
-
-        payload = dumps(separators=(',', ':'), obj={
-            'queryName': 'MainSignupLoginSection_sendVerificationCodeMutation_Mutation',
-            'variables': {
-                'emailAddress': mail_address,
-                'phoneNumber': None,
-                'recaptchaToken': token
+        token = solver.recaptcha(
+            sitekey='6LflhEElAAAAAI_ewVwRWI9hsyV4mbZnYAslSvlG',
+            url='https://poe.com/login?redirect_url=%2F',
+            version='v3',
+            enterprise=1,
+            invisible=1,
+            action='login',
+        )['code']
+
+        payload = dumps(
+            separators=(',', ':'),
+            obj={
+                'queryName': 'MainSignupLoginSection_sendVerificationCodeMutation_Mutation',
+                'variables': {'emailAddress': mail_address, 'phoneNumber': None, 'recaptchaToken': token},
+                'query': 'mutation MainSignupLoginSection_sendVerificationCodeMutation_Mutation(\n  $emailAddress: String\n  $phoneNumber: String\n  $recaptchaToken: String\n) {\n  sendVerificationCode(verificationReason: login, emailAddress: $emailAddress, phoneNumber: $phoneNumber, recaptchaToken: $recaptchaToken) {\n    status\n    errorMessage\n  }\n}\n',
             },
-            'query': 'mutation MainSignupLoginSection_sendVerificationCodeMutation_Mutation(\n  $emailAddress: String\n  $phoneNumber: String\n  $recaptchaToken: String\n) {\n  sendVerificationCode(verificationReason: login, emailAddress: $emailAddress, phoneNumber: $phoneNumber, recaptchaToken: $recaptchaToken) {\n    status\n    errorMessage\n  }\n}\n',
-        })
+        )
 
         base_string = payload + client.headers["poe-formkey"] + 'WpuLMiXEKKE98j56k'
         client.headers["poe-tag-id"] = md5(base_string.encode()).hexdigest()
@@ -74,31 +76,34 @@ class Account:
             print('please try using a proxy / wait for fix')
 
         if 'Bad Request' in response.text:
-            if logging: print('bad request, retrying...', response.json())
+            if logging:
+                print('bad request, retrying...', response.json())
             quit()
 
-        if logging: print('send_code', response.json())
+        if logging:
+            print('send_code', response.json())
 
         mail_content = mail_client.get_message()
         mail_token = findall(r';">(\d{6,7})</div>', mail_content)[0]
 
-        if logging: print('code', mail_token)
+        if logging:
+            print('code', mail_token)
 
-        payload = dumps(separators=(',', ':'), obj={
-            "queryName": "SignupOrLoginWithCodeSection_signupWithVerificationCodeMutation_Mutation",
-            "variables": {
-                "verificationCode": str(mail_token),
-                "emailAddress": mail_address,
-                "phoneNumber": None
+        payload = dumps(
+            separators=(',', ':'),
+            obj={
+                "queryName": "SignupOrLoginWithCodeSection_signupWithVerificationCodeMutation_Mutation",
+                "variables": {"verificationCode": str(mail_token), "emailAddress": mail_address, "phoneNumber": None},
+                "query": "mutation SignupOrLoginWithCodeSection_signupWithVerificationCodeMutation_Mutation(\n  $verificationCode: String!\n  $emailAddress: String\n  $phoneNumber: String\n) {\n  signupWithVerificationCode(verificationCode: $verificationCode, emailAddress: $emailAddress, phoneNumber: $phoneNumber) {\n    status\n    errorMessage\n  }\n}\n",
             },
-            "query": "mutation SignupOrLoginWithCodeSection_signupWithVerificationCodeMutation_Mutation(\n  $verificationCode: String!\n  $emailAddress: String\n  $phoneNumber: String\n) {\n  signupWithVerificationCode(verificationCode: $verificationCode, emailAddress: $emailAddress, phoneNumber: $phoneNumber) {\n    status\n    errorMessage\n  }\n}\n"
-        })
+        )
 
         base_string = payload + client.headers["poe-formkey"] + 'WpuLMiXEKKE98j56k'
         client.headers["poe-tag-id"] = md5(base_string.encode()).hexdigest()
 
         response = client.post('https://poe.com/api/gql_POST', data=payload)
-        if logging: print('verify_code', response.json())
+        if logging:
+            print('verify_code', response.json())
 
 
 Account.create(proxy='xtekky:wegwgwegwed_streaming-1@geo.iproyal.com:12321', logging=True)