97 lines
2.2 KiB
Python
97 lines
2.2 KiB
Python
|
import json
|
|||
|
|
import sys
|
|||
|
|
|
|||
|
|
if len(sys.argv) != 2:
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class Lesson:
|
|||
|
|
def __init__(self):
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
def read_from_schedule_pair(self, p):
|
|||
|
|
self.time = p["time"].rsplit(':', 1)[0]
|
|||
|
|
self.name = p["name"]
|
|||
|
|
self.teacher = p["teacherName"]
|
|||
|
|
self.type = p["tag"]
|
|||
|
|
|
|||
|
|
self.selectable = False
|
|||
|
|
self.link = ""
|
|||
|
|
self.nolink = True
|
|||
|
|
|
|||
|
|
def get_packable_object(self):
|
|||
|
|
return {
|
|||
|
|
"name": self.name,
|
|||
|
|
"teacher": self.teacher,
|
|||
|
|
"link": self.link,
|
|||
|
|
"type": self.type,
|
|||
|
|
"selectable": self.selectable,
|
|||
|
|
"nolink": self.nolink
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
def day_to_lesson_pool(pairs):
|
|||
|
|
pool = []
|
|||
|
|
|
|||
|
|
for p in pairs:
|
|||
|
|
l = Lesson()
|
|||
|
|
l.read_from_schedule_pair(p)
|
|||
|
|
pool.append(l)
|
|||
|
|
|
|||
|
|
return pool
|
|||
|
|
|
|||
|
|
def lesson_pool_to_objs(pool):
|
|||
|
|
times = []
|
|||
|
|
|
|||
|
|
for i in (list(map(int, (i.time.split(':')))) for i in pool):
|
|||
|
|
if i not in times:
|
|||
|
|
times.append(i)
|
|||
|
|
|
|||
|
|
times.sort(key = lambda x: x[0] * 60 + x[1])
|
|||
|
|
|
|||
|
|
|
|||
|
|
result = {}
|
|||
|
|
|
|||
|
|
for i in times:
|
|||
|
|
lesson_timestamp = f"{i[0]:02}:{i[1]:02}"
|
|||
|
|
required_timestamp = f"{i[0]}:{i[1]}"
|
|||
|
|
|
|||
|
|
objs_dup = (j.get_packable_object() for j in pool if j.time == lesson_timestamp)
|
|||
|
|
objs = []
|
|||
|
|
|
|||
|
|
for i in objs_dup:
|
|||
|
|
if i not in objs:
|
|||
|
|
objs.append(i)
|
|||
|
|
|
|||
|
|
if len(objs) == 0:
|
|||
|
|
continue
|
|||
|
|
elif len(objs) == 1:
|
|||
|
|
result[required_timestamp] = list(objs)[0]
|
|||
|
|
else:
|
|||
|
|
result[required_timestamp] = list(objs)
|
|||
|
|
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
|
|||
|
|
def process_week(week):
|
|||
|
|
result = []
|
|||
|
|
|
|||
|
|
for weekday in ["Пн", "Вв", "Ср", "Чт", "Пт"]:
|
|||
|
|
day_pool = day_to_lesson_pool(next(filter(lambda l: l['day'] == weekday, week))['pairs'])
|
|||
|
|
result.append(lesson_pool_to_objs(day_pool))
|
|||
|
|
|
|||
|
|
result.append({})
|
|||
|
|
result.append({})
|
|||
|
|
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
|
|||
|
|
ext_schedule = json.load(open(sys.argv[1], 'r'))
|
|||
|
|
|
|||
|
|
int_schedule = []
|
|||
|
|
|
|||
|
|
int_schedule += process_week(ext_schedule["scheduleFirstWeek"])
|
|||
|
|
int_schedule += process_week(ext_schedule["scheduleSecondWeek"])
|
|||
|
|
|
|||
|
|
print(json.dumps(int_schedule, ensure_ascii = False, indent = 4))
|