FAST_TimeTable/main.py at main · Hashimk101/FAST_TimeTable · GitHub

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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
from databaseHandler import (
    days_of_week,
    make_database,
    insert_timetable,
    read_and_clean_classroom_df,
    read_and_clean_lab_df,
    fetch_timetable_for_section,
    get_sheets_service,
    get_day_sheet_mapping
)

from main_menu import (
    main_menu
)

SPREADSHEET_ID = "1vlTuotLw34fedME3gNQj09cZw-todVomxAiu5P1wZ6Q"

COURSE_DATABASE = "uni_timetable.db"
LAB_DATABASE = "uni_timetable_lab.db"

def main():
    # 1. Create databases and get timetable data from google sheets
    make_database(COURSE_DATABASE, 'CLASSROOM')
    make_database(LAB_DATABASE, 'LAB')

    service = get_sheets_service()
    sheet_mapping = get_day_sheet_mapping(service, SPREADSHEET_ID, refresh=True)
    print(f"Discovered day sheet mapping (hidden sheets skipped): {sheet_mapping}")

    for day in days_of_week:
        if day not in sheet_mapping:
            print(f"Skipping {day}: No visible/active sheet found.")
            continue

        actual_sheet = sheet_mapping[day]
        try:
            # classroom setup
            clean_df = read_and_clean_classroom_df(service, SPREADSHEET_ID, actual_sheet)
            insert_timetable(clean_df, day, COURSE_DATABASE, 'Room', 'CLASSROOM')
            print(f"Inserted classroom timetable for {day} (from sheet '{actual_sheet}')")

            # lab setup
            clean_dflab = read_and_clean_lab_df(service, SPREADSHEET_ID, actual_sheet)
            insert_timetable(clean_dflab, day, LAB_DATABASE, 'Lab', 'LAB')
            print(f"Inserted lab timetable for {day} (from sheet '{actual_sheet}')")

        except Exception as e:
            print(f"Error processing {day} (sheet '{actual_sheet}'): {e}")

    print("\nDatabase creation complete!")
    print("- Classroom timetable: uni_timetable.db")
    print("- Lab timetable: uni_timetable_lab.db")


if __name__ == "__main__":
    # 1. build the database
    main()