From 660cb8002b071b13584d1ee70d3815c194019326 Mon Sep 17 00:00:00 2001
From: valoka <valio557782@gmail.com>
Date: Thu, 19 Sep 2024 23:00:24 +0900
Subject: [PATCH] added dc bot

---
 main.py | 78 +++++++++++++++++++++++++++++++++++++++++++++++++--------
 1 file changed, 68 insertions(+), 10 deletions(-)

diff --git a/main.py b/main.py
index 32ac6a6..53435cb 100644
--- a/main.py
+++ b/main.py
@@ -1,14 +1,72 @@
 from selenium import webdriver
-from selenium.webdriver.common.keys import Keys
+from selenium.webdriver.firefox.options import Options
+# from selenium.webdriver.common.keys import Keys
 from selenium.webdriver.common.by import By
 import time
+from bs4 import BeautifulSoup
+import lxml
 
-driver = webdriver.Firefox()
-driver.get("https://amursu.ru/obrazovanie/timetable/timetable-group/?id_group=1891")
-assert "Расписание" in driver.title
-# /html/body/main/div[2]/div/div/div[3]/div[4]/div[3]
-infa = driver.find_element(By.XPATH, '/html/body/main/div[2]/div/div/div[3]/div[4]/div[7]/table/tbody/tr[1]/td[3]/div/div[2]')
-print(infa)
-print(infa.text)
-# time.sleep(10)
-driver.close()
\ No newline at end of file
+import discord
+from discord.ext import commands
+
+#objs
+options = Options()
+intents = discord.Intents.default()
+prikols = commands.Bot(intents=intents, command_prefix = '!')
+
+# consts
+options.headless = True
+intents.message_content = True
+intents.messages = True
+
+url = "https://amursu.ru/obrazovanie/timetable/timetable-group/?id_group=1891"
+
+# гайд на завтра:
+# 1. собираем все хпатх | ок
+# 2. расчекрыживаем при помощи bs4 | ок
+# 3. приводим в читаемый вид | fail
+# 4. отправляем в дс | ок
+
+def get_day_info(day : int, url : str) -> list:
+    '''Get's day of th week number
+        posible numbers: 3-9 (included)'''
+    driver = webdriver.Firefox()
+    driver.get(url)
+    assert "Расписание" in driver.title
+
+    elem = []
+    arr = ''
+    elem.append(driver.find_element(By.XPATH, f'/html/body/main/div[2]/div/div/div[3]/div[4]/div[{day}]').get_attribute('innerHTML'))
+    driver.close()
+
+    text = str(elem)
+
+    soup = BeautifulSoup(str(text), 'lxml')
+    soupa = soup.findAll("a", attrs = {'class':'ui link blue speak'})
+    soup = BeautifulSoup(str(soupa)).find_all('a')
+    for data in soup:
+        arr += data.get('data-speech') + '\n'
+    
+    print(arr)
+    return arr
+
+def get_week_info():
+    week_info = []
+    for i in range(3, 10):
+        week_info.append(get_day_info())
+    return week_info
+
+@prikols.command()
+async def day(ctx, arg):
+    await ctx.send(get_day_info(arg, url=url))
+
+@prikols.command()
+async def week(ctx, arg):
+    await ctx.send(get_day_info(arg, url=url))
+
+@prikols.event
+async def on_ready():
+    print('yolo') 
+
+# steal my token lol
+prikols.run('MTE5ODIwMTEwMTU2OTI0OTM4MA.G7h2hW.ei4EZVmva7SymBijBO9tcx8xhNBtQMMSu4AsE0')
\ No newline at end of file