126 lines
4.6 KiB
Python
126 lines
4.6 KiB
Python
import json
|
|
import openpyxl.cell
|
|
import requests
|
|
from importlib import reload
|
|
|
|
import openpyxl
|
|
from openpyxl import load_workbook
|
|
import config
|
|
|
|
def trim(string: str) -> str:
|
|
if string == '': return ''
|
|
try:
|
|
while string[0] == ' ': string = string[1: ]
|
|
while string[-1] == ' ': string = string[ :-1]
|
|
except: pass
|
|
return string
|
|
|
|
def update(group: str, course: str, debug: bool = false):
|
|
files = json.load(open('files.json', 'r', encoding='utf-8'))
|
|
url = files[group][course]
|
|
|
|
response = requests.get(url, stream=True)
|
|
|
|
print(f'\n{group} ({course})')
|
|
with open("temp/table.xlsx", "wb") as handle:
|
|
for data in response.iter_content():
|
|
handle.write(data)
|
|
|
|
result = {}
|
|
book = load_workbook('./temp/table.xlsx', data_only = True).active
|
|
for c in 'CDEFGHIJKLMNOPQRSTUVWXYZ':
|
|
usedfrom = 8
|
|
try:
|
|
typename = book[f'{c}8'].value.replace(' ','') # type: ignore
|
|
if not typename: raise Exception
|
|
|
|
except:
|
|
usedfrom = 9
|
|
try: typename = book[f'{c}9'].value.replace(' ','') # type: ignore
|
|
except: continue
|
|
if not typename: continue
|
|
# if c != 'D': continue #test
|
|
|
|
print(f'├── {typename}') if debug else ...
|
|
result[typename] = {}
|
|
|
|
last_day = ''
|
|
last_delta = ''
|
|
last_lesson = ''
|
|
|
|
this = {}
|
|
|
|
for r in range(usedfrom+1, 99):
|
|
day = book[f'A{r}'].value # type: ignore
|
|
if day: day = day.capitalize()
|
|
delta = book[f'B{r}'].value # type: ignore
|
|
lesson = book[f'{c}{r}'].value # type: ignore
|
|
|
|
print(day, delta, lesson, f'l_col:{c} row:{r}')
|
|
|
|
if day == 'Суббота': break
|
|
|
|
if delta:
|
|
SS, EE = delta.split('-')[0], delta.split('-')[1]
|
|
if len(SS) == 3: SS = f'{SS[0]}:{SS[1]}{SS[2]}'
|
|
else: SS = f'{SS[0]}{SS[1]}:{SS[2]}{SS[3]}'
|
|
|
|
if len(EE) == 3: EE = f'{EE[0]}:{EE[1]}{EE[2]}'
|
|
else: EE = f'{EE[0]}{EE[1]}:{EE[2]}{EE[3]}'
|
|
delta = f'{SS} - {EE}'
|
|
|
|
print(f'{c}{r}', day, delta, 'Lesson text...' if lesson else 'None', book[f'{c}{r}']) if debug else ... # type: ignore
|
|
|
|
if day or delta or (lesson) or (not lesson and type(book[f'{c}{r}']) != openpyxl.cell.MergedCell): # type: ignore
|
|
if day and day != last_day:
|
|
last_day = day
|
|
this[last_day] = {}
|
|
|
|
if delta and delta != last_delta:
|
|
last_delta = delta
|
|
this[last_day][last_delta] = []
|
|
|
|
if lesson and lesson != last_lesson:
|
|
last_lesson = lesson
|
|
d = filter(lambda _: _!='', lesson.replace('\n', ' ').split(' '))
|
|
lesson = []
|
|
for _ in d: lesson.append(trim(_))
|
|
lessonOBJ = {'type': '', 'name': '', 'info': [], 'raw': lesson, 'ok': True}
|
|
if lesson[0] == '*':
|
|
lessonOBJ['ok'] = False
|
|
this[last_day][last_delta].append(lessonOBJ)
|
|
continue
|
|
|
|
lessonOBJ['name'] = trim(lesson[0])
|
|
try: lessonOBJ['type'] = trim(lesson[1])
|
|
except:
|
|
lessonOBJ['ok'] = False
|
|
this[last_day][last_delta].append(lessonOBJ)
|
|
continue
|
|
|
|
addInfo = ' '.join(lesson[2:]).replace(' ', '$')
|
|
if len(lesson[1].split(' ')) > 1:
|
|
addInfo = ' '.join(lesson[1].split(' ')[1:]).replace(' ', '$') + '$$' + addInfo
|
|
for info in addInfo.split('$'):
|
|
if info == '' or info == ' ': continue
|
|
lessonOBJ['info'].append(trim(info))
|
|
this[last_day][last_delta].append(lessonOBJ)
|
|
else: this[last_day][last_delta].append(None)
|
|
|
|
|
|
|
|
result[typename] = this
|
|
print(f'└──── Downloaded \n') if debug else ...
|
|
json.dump(result, open(f'./timetables/{group} {course}.json'.replace(' ','_'), 'w', encoding='utf-8'), ensure_ascii=False, indent=4)
|
|
|
|
|
|
|
|
def updateAll(debug: bool = false):
|
|
global config
|
|
config = reload(config)
|
|
|
|
for group, courses in config.LOAD.items():
|
|
for course in courses:
|
|
update(group, course, debug)
|
|
|
|
# updateAll()
|