126 lines
3.6 KiB
Python
126 lines
3.6 KiB
Python
# schedule.py
|
|
#
|
|
# experimenting with manipulating and querying the schedule of courses
|
|
|
|
|
|
import json, re
|
|
from datetime import datetime
|
|
|
|
course_types = {'in-person':'IP','hybrid':'H','online':'O','online live':'OL'}
|
|
|
|
def course_to_string(crs):
|
|
# crn type days start end cred code name teacher
|
|
lengths = [5, 3, 5, 6, 6, 4, 4, 13, 35, 25]
|
|
items = [ crs[x] for x in 'crn,type,days,time_start,time_end,cred,act,code,name,teacher'.split(',')]
|
|
items[1] = course_types[ items[1] ]
|
|
items[5] = items[5][0:3]
|
|
result = " ".join(f"{str(val):{width}}" for val, width in zip(items, lengths))
|
|
return result
|
|
|
|
|
|
def parse_days(str_days):
|
|
# return a list with actual day of week names
|
|
days = []
|
|
if 'M' in str_days:
|
|
days.append('monday')
|
|
if 'T' in str_days:
|
|
days.append('tuesday')
|
|
if 'W' in str_days:
|
|
days.append('wednesday')
|
|
if 'R' in str_days:
|
|
days.append('thursday')
|
|
if 'F' in str_days:
|
|
days.append('friday')
|
|
if 'S' in str_days:
|
|
days.append('saturday')
|
|
return days
|
|
|
|
def parse_courses(filename):
|
|
with open(filename) as f:
|
|
courses = json.load(f)
|
|
|
|
locations, teachers, days, now = {}, {}, {}, datetime.now()
|
|
weekdays = ['monday', 'tuesday', 'wednesday', 'thursday', 'friday', 'saturday', 'sunday']
|
|
|
|
for course in courses:
|
|
#print(course)
|
|
|
|
# Add course to teachers dict
|
|
course['teacher'] = re.sub(r'\s+', ' ', course['teacher'])
|
|
teachers.setdefault(course['teacher'], []).append(course)
|
|
|
|
if course['type'] != 'in-person': continue
|
|
# Add course to locations dict
|
|
locations.setdefault(course['loc'], []).append(course)
|
|
|
|
# Add course to days dict
|
|
for day in parse_days(course['days']):
|
|
days.setdefault(day, []).append(course)
|
|
|
|
# Check if course is happening now
|
|
if course['time_start'] == '': continue
|
|
if course['time_end'] == '': continue
|
|
start_time = datetime.strptime(course['time_start'], '%H:%M')
|
|
end_time = datetime.strptime(course['time_end'], '%H:%M')
|
|
#if start_time.time() <= now.time() <= end_time.time():
|
|
# print(f"{course['code']} is happening now in {course['loc']}")
|
|
|
|
return locations, teachers, days
|
|
|
|
# Example usage
|
|
locations, teachers, days = parse_courses('cache/sample_semester.json')
|
|
|
|
#print(json.dumps(locations,indent=2))
|
|
#print(json.dumps(teachers,indent=2))
|
|
#print(json.dumps(days,indent=2))
|
|
|
|
dump = 0
|
|
|
|
if dump:
|
|
for L,C in locations.items():
|
|
print(f"\n{L}")
|
|
for crs in C:
|
|
print(" " + course_to_string(crs))
|
|
|
|
for T,C in teachers.items():
|
|
print(f"\n{T}")
|
|
for crs in C:
|
|
print(" " + course_to_string(crs))
|
|
|
|
# EXAMPLES
|
|
#
|
|
print("\n\n")
|
|
|
|
# Get schedule for a specific room on a specific day
|
|
room = 'HU 104'
|
|
day = 'monday'
|
|
print(f"\nSchedule for {room} on {day}:")
|
|
for course in locations[room]:
|
|
if day in parse_days(course['days']):
|
|
print(" " + course_to_string(course))
|
|
|
|
# Get weekly schedule for a specific teacher
|
|
teacher = 'Kimberly J Smith'
|
|
print(f"\nWeekly schedule for {teacher}:")
|
|
for course in teachers[teacher]:
|
|
print(" " + course_to_string(course))
|
|
|
|
import sys
|
|
import os
|
|
|
|
columns, rows = os.get_terminal_size()
|
|
print(f"Terminal size: {columns} columns x {rows} rows")
|
|
|
|
|
|
def write_at(row, col, text):
|
|
sys.stdout.write(f"\033[{row};{col}H{text}")
|
|
sys.stdout.flush()
|
|
|
|
#
|
|
lines = [1,2,3,4,5]
|
|
for L in lines:
|
|
myline = rows - L
|
|
write_at(myline, 1, f"This is line {myline}")
|
|
|
|
write_at(rows,0,'')
|