Add files via upload

parse.py

@@ -0,0 +1,110 @@
+import json
+import openpyxl.cell
+import requests
+from importlib import reload
+
+import openpyxl
+from openpyxl import load_workbook
+import config
+
+
+def update(group: str, course: str):
+    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':
+        try: typename = book[f'{c}8'].value.replace(' ','')
+        except: continue
+        if not typename: continue
+        # if c != 'D': continue #test
+        
+        print(f'├── {typename}')
+        result[typename] = {}
+
+        last_day = ''
+        last_delta = ''
+        last_lesson = ''
+
+        this = {}
+
+        for r in range(9, 99):
+            day = book[f'A{r}'].value
+            if day: day = day.capitalize()
+            delta = book[f'B{r}'].value
+            lesson = book[f'{c}{r}'].value
+            
+            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 day or delta or (lesson) or (not lesson and type(book[f'{c}{r}']) != openpyxl.cell.MergedCell):
+                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
+                    lessonOBJ = {'type': '', 'name': '', 'info': [], 'raw': lesson, 'ok': True}
+                    if lesson[0] == '*': 
+                        lessonOBJ['ok'] = False
+                        this[last_day][last_delta].append(lessonOBJ)
+                        continue
+                    lessonOBJ['name'] = lesson.split('\n')[0]
+                    try: lessonOBJ['type'] = lesson.split('\n')[1].split('  ')[0]
+                    except:
+                        lessonOBJ['ok'] = False
+                        this[last_day][last_delta].append(lessonOBJ)
+                        continue
+
+                    addInfo = '  '.join(lesson.split('\n')[2:]).replace('  ', '$')
+                    if len(lesson.split('\n')[1].split('  ')) > 1:
+                        addInfo = '  '.join(lesson.split('\n')[1].split('  ')[1:]).replace('  ', '$') + '$$' + addInfo 
+                    for info in addInfo.split('$'):
+                        if info == '' or info == ' ': continue
+                        while (info[0] == ' ' or info[-1] == ' '):
+                            if info[0] == ' ': info = info[1:]
+                            elif info[-1] == ' ': info = info[:-1]
+                        lessonOBJ['info'].append(info)
+                    this[last_day][last_delta].append(lessonOBJ)
+                else: this[last_day][last_delta].append(None)
+
+
+
+        result[typename] = this
+    print(f'└──── Downloaded \n')
+    json.dump(result, open(f'./timetables/{group} {course}.json'.replace(' ','_'), 'w', encoding='utf-8'), ensure_ascii=False, indent=4)
+
+
+
+def updateAll(): 
+    global config
+    config = reload(config)
+
+    for group, courses in config.LOAD.items():
+        for course in courses:
+            update(group, course)
+
+updateAll()