xync-client 0.0.50.dev1__tar.gz → 0.0.50.dev2__tar.gz

{xync_client-0.0.50.dev1/xync_client.egg-info → xync_client-0.0.50.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xync-client
-Version: 0.0.50.dev1
+Version: 0.0.50.dev2
 Author-email: Mike Artemiev <mixartemev@gmail.com>
 Project-URL: Homepage, https://gitlab.com/XyncNet/client
 Project-URL: Repository, https://gitlab.com/XyncNet/client

{xync_client-0.0.50.dev1 → xync_client-0.0.50.dev2}/xync_client/Pms/Ozon/__init__.py

@@ -5,20 +5,12 @@ from xync_client.loader import PG_DSN
 import undetected_chromedriver as uc
 from selenium.webdriver.common.by import By
 from selenium.webdriver.common.action_chains import ActionChains
-import os
-import json
 import time
 
-async def save_cookies(driver):
-    cookies = driver.get_cookies()  # Получаем все куки
-    with open("state.json", "w") as file:
-        json.dump(cookies, file)  # Сохраняем в JSON
 
 async def loggin(driver, agent) -> None:
     driver.get("https://finance.ozon.ru")
-    with open("state.json", "r") as file:
-        cookies = json.load(file)
-    for cookie in cookies:
+    for cookie in agent.state:
         driver.add_cookie(cookie)
     driver.get("https://finance.ozon.ru/lk")
     pin = agent.auth.get("code")
@@ -32,32 +24,33 @@ async def send_cred(driver, amount, payment, cred):
     pass
 
 
-
 async def main():
     _ = await init_db(PG_DSN, models, True)
     agent = await models.PmAgent.filter(pm__norm="ozon", auth__isnull=False).first()
     chrome_options = uc.ChromeOptions()
-    chrome_options.add_argument('--disable-gpu')
-    chrome_options.add_argument('--no-sandbox')
+    chrome_options.add_argument("--disable-gpu")
+    chrome_options.add_argument("--no-sandbox")
     driver = uc.Chrome(options=chrome_options)
 
     try:
-        if not os.path.exists("state.json"):
-            driver.get('https://ozon.ru/ozonid-lite')
-            time.sleep(3)
+        if agent.state:
+            await loggin(driver, agent)
+            time.sleep(1000)
+        else:
+            driver.get("https://ozon.ru/ozonid-lite")
+            driver.implicitly_wait(3)
             driver.find_element(By.NAME, "autocomplete").send_keys(agent.auth.get("phone"))
             driver.find_element(By.CLASS_NAME, "b201-a").click()
             sms_code = input("Введите 6-ти значный код: ")
+            driver.implicitly_wait(3)
             driver.find_element(By.CLASS_NAME, "d01-a.d01-a5").send_keys(sms_code)
-            time.sleep(3)
-            await save_cookies(driver)
-        else:
-            await loggin(driver, agent)
-            time.sleep(1000)
+            driver.implicitly_wait(5)
+            agent.state = driver.get_cookies()  # Получаем все куки
+            await agent.save()
 
     finally:
         driver.quit()
 
 
 if __name__ == "__main__":
-    run(main())
+    run(main())

{xync_client-0.0.50.dev1 → xync_client-0.0.50.dev2/xync_client.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xync-client
-Version: 0.0.50.dev1
+Version: 0.0.50.dev2
 Author-email: Mike Artemiev <mixartemev@gmail.com>
 Project-URL: Homepage, https://gitlab.com/XyncNet/client
 Project-URL: Repository, https://gitlab.com/XyncNet/client