доделал работу с api yandex lyceum теперь без костылей

functions/yandex.py

@@ -0,0 +1,51 @@
+import sqlite3
+import time
+import requests
+from aiogram import types
+from selenium import webdriver
+from selenium.webdriver.chrome.service import Service
+from selenium.webdriver.common.by import By
+from webdriver_manager.chrome import ChromeDriverManager
+
+
+async def create_yandex(message: types.Message):
+    conn = sqlite3.connect('db.sql')
+    cur = conn.cursor()
+    response = cur.execute("SELECT * FROM yandex WHERE user_id=?", (message.from_user.id, )).fetchall()
+    conn.close()
+    if response:
+        await message.reply("Ваш аккаунт уже добавлен")
+    options = webdriver.ChromeOptions()
+    options.add_argument("--headless")
+    driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()), options=options)
+    driver.get('https://passport.yandex.ru/auth/')
+    time.sleep(1)
+    driver.find_element(By.CLASS_NAME, "Button2").click()
+    time.sleep(1)
+    driver.find_element(By.XPATH, '//*[@id="passp-field-login"]').send_keys(message.text.split()[1])
+    driver.find_element(By.XPATH, '//*[@id="passp:sign-in"]').click()
+    time.sleep(1)
+    driver.find_element(By.XPATH, '//*[@id="passp-field-passwd"]').send_keys(message.text.split()[2])
+    driver.find_element(By.XPATH, '//*[@id="passp:sign-in"]').click()
+    time.sleep(5)
+    try:
+        cookies = [i["value"] for i in driver.get_cookies() if i["name"] == "Session_id"][0]
+        conn = sqlite3.connect('db.sql')
+        cur = conn.cursor()
+        cur.execute("INSERT INTO yandex (user_id, Session_id) VALUES (?, ?)", (message.from_user.id, str(cookies)))
+        conn.commit()
+        conn.close()
+        await message.reply("Ваш аккаунт добавлен")
+    except IndexError:
+        await message.reply("Ошибка пароля или логина")
+
+async def statistic(message: types.Message):
+    conn = sqlite3.connect('db.sql')
+    cur = conn.cursor()
+    response = cur.execute("SELECT * FROM yandex WHERE user_id=?", (message.from_user.id,)).fetchall()
+    conn.close()
+    if not response:
+        await message.reply("Сначала используйте /create_yandex с логином и паролем для добавление аккаунта")
+    cookies = {"Session_id": response[0][2]}
+    response = requests.get("https://lyceum.yandex.ru/api/notifications?isRead=false&limit=20", cookies=cookies)
+    await message.reply(f'У вас {response.json()["unreadCount"]} уведомлений. Прочитайте хоть')