-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathextract_data.py
More file actions
32 lines (29 loc) · 1.17 KB
/
extract_data.py
File metadata and controls
32 lines (29 loc) · 1.17 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
from datetime import datetime
import time
def extract_data_lingoace(data: dict) -> list:
result = []
for lesson in data['tutorTimetableList']:
start_utc = lesson['startTime']
end_utc = lesson['endTime']
dt_start = datetime.strptime(start_utc, '%Y-%m-%dT%H:%M:00.000Z')
dt_end = datetime.strptime(end_utc, '%Y-%m-%dT%H:%M:00.000Z')
unix_start = int(time.mktime(dt_start.timetuple()) + 10800) # +3 hours
unix_end = int(time.mktime(dt_end.timetuple()) + 10800) # +3 hours
name = lesson['studentUsername']
data = {'start':unix_start, 'end':unix_end, 'name': name}
result.append(data)
return result
def extract_data_qkid(data: list) -> list:
result = []
for lesson in data:
if lesson['status'] == 0:
start = lesson['beginAt']
end = lesson['endAt']
level = lesson['params'].get('abbreviation')
unit = lesson['params'].get('lessonName')
if level:
name = f'{level} - {unit}'
else:
name = 'Reading class'
result.append({'start':start, 'end':end, 'name': name})
return result