ask input for filenames

Author: ROODAY

generate_quote_audio.py

@@ -4,7 +4,8 @@
 from pydub import AudioSegment
 from io import BytesIO
 
-with open("data/final schedule with uuids.csv", encoding="utf-8") as scheduleFile:
+filename = input("Enter data filename (default: contacts.csv): ") or "contacts.csv"
+with open("data/{}".format(filename), encoding="utf-8") as scheduleFile:
     graduates = [
         {k: v for k, v in row.items()}
         for row in csv.DictReader(scheduleFile, skipinitialspace=True)

generate_schedule.py

@@ -36,7 +36,8 @@ def comparer(left, right):
 
 
 # load data then sort by descending timezone and ascending school within timezone
-with open("data/school timezones.csv", encoding="utf-8") as schoolTimezonesFile:
+filename = input("Enter timezones filename (default: school timezones.csv): ") or "school timezones.csv"
+with open("data/{}".format(filename), encoding="utf-8") as schoolTimezonesFile:
     data = [
         {k: v if k == "School" else float(v) for k, v in row.items()}
         for row in csv.DictReader(schoolTimezonesFile, skipinitialspace=True)
@@ -124,7 +125,8 @@ def __eq__(self, other):
     print(dupes)
 
 # now to order students
-with open("data/HS - Master RSVPs.csv", encoding="utf-8") as rsvpFile:
+filename = input("Enter contacts filename (default: contacts.csv): ") or "contacts.csv"
+with open("data/{}".format(filename), encoding="utf-8") as rsvpFile:
     rsvps = [
         {k: v for k, v in row.items()}
         for row in csv.DictReader(rsvpFile, skipinitialspace=True)
@@ -220,8 +222,9 @@ def __eq__(self, other):
     print(tabulate(print_order, headers="keys"))
 
 # save utc student schedule
+filename = input("Enter student schedule filename (default: student schedule utc.csv): ") or "student schedule utc.csv"
 with open(
-    "data/student schedule utc.csv", "w", encoding="utf-8", newline=""
+    "data/{}".format(filename), "w", encoding="utf-8", newline=""
 ) as studentScheduleFile:
     csvFields = list(pretty_order[0].keys())
     writer = csv.DictWriter(studentScheduleFile, fieldnames=csvFields)

import_sql.py

@@ -18,7 +18,8 @@
     cursorclass=pymysql.cursors.DictCursor,
 )
 
-with open("data/final schedule with uuids.csv", encoding="utf-8") as scheduleFile:
+filename = input("Enter data filename (default: contacts.csv): ") or "contacts.csv"
+with open("data/{}".format(filename), encoding="utf-8") as scheduleFile:
     graduates = [
         {k: v for k, v in row.items()}
         for row in csv.DictReader(scheduleFile, skipinitialspace=True)

merge_schedule_with_contacts.py

@@ -6,9 +6,12 @@
 
 PRINT_TABULATE = False
 SHOW_VIS = False
+SAVE_SCHOOL_CSV = True
+SAVE_SCHOOL_HTML = True
 
 # load clustered schedule (make sure you go make that first in google sheets)
-with open("data/student schedule utc clustered.csv", encoding="utf-8") as file:
+filename = input("Enter clustered schedule filename (default: student schedule utc clustered.csv): ") or "student schedule utc clustered.csv"
+with open("data/{}".format(filename), encoding="utf-8") as file:
     data = [
         {k: v for k, v in row.items()}
         for row in csv.DictReader(file, skipinitialspace=True)
@@ -28,7 +31,8 @@
     plt.show()
 
 # load original RSVPs and make map keyed by email
-with open("data/HS - Master RSVPs.csv", encoding="utf-8") as rsvpFile:
+filename = input("Enter contacts filename (default: contacts.csv): ") or "contacts.csv"
+with open("data/{}".format(filename), encoding="utf-8") as rsvpFile:
     rsvps = [
         {k: v for k, v in row.items()}
         for row in csv.DictReader(rsvpFile, skipinitialspace=True)
@@ -103,21 +107,25 @@
     for s in schoolOrder
 ]
 
-"""# save utc school schedule
-with open(
-    "data/FINAL school schedule utc.csv", "w", encoding="utf-8", newline=""
-) as schoolScheduleCSVFile:
-    csvFields = ["School", "Start Time"]
-    writer = csv.DictWriter(schoolScheduleCSVFile, fieldnames=csvFields)
-    writer.writeheader()
-    writer.writerows(schoolOrder)
-    print("\n--> Saved FINAL school schedule utc CSV!")"""
+if SAVE_SCHOOL_CSV:
+    # save utc school schedule
+    filename = input("Enter school schedule CSV filename (default: school schedule utc.csv): ") or "school schedule utc.csv"
+    with open(
+        "data/{}".format(filename), "w", encoding="utf-8", newline=""
+    ) as schoolScheduleCSVFile:
+        csvFields = ["School", "Start Time"]
+        writer = csv.DictWriter(schoolScheduleCSVFile, fieldnames=csvFields)
+        writer.writeheader()
+        writer.writerows(schoolOrder)
+        print("\n--> Saved FINAL school schedule utc CSV!")
 
-# use tabulate to save html school schedule
-with open("data/school-schedule.html", "w", encoding="utf-8") as schoolScheduleHTMLFile:
-    # schoolScheduleHTMLFile.write(tabulate(schoolOrder, headers="keys", tablefmt="html"))
-    print(schoolOrder)
-    print("\n--> Saved FINAL school schedule utc HTML!")
+if SAVE_SCHOOL_HTML:
+    # use tabulate to save html school schedule
+    filename = input("Enter school schedule HTML filename (default: school-schedule.html): ") or "school-schedule.html"
+    with open("data/{}".format(filename), "w", encoding="utf-8") as schoolScheduleHTMLFile:
+        schoolScheduleHTMLFile.write(tabulate(schoolOrder, headers="keys", tablefmt="html"))
+        print(schoolOrder)
+        print("\n--> Saved FINAL school schedule utc HTML!")
 
 # save final dataset
 finalData = []
@@ -130,8 +138,9 @@
     student["Approx Time Zone"] = item["Time Zone"]
     finalData.append(student)
 
+filename = input("Enter student schedule filename (default: student schedule utc.csv): ") or "student schedule utc.csv"
 with open(
-    "data/FINAL student schedule utc.csv", "w", encoding="utf-8", newline=""
+    "data/{}".format(filename), "w", encoding="utf-8", newline=""
 ) as studentScheduleFile:
     csvFields = list(finalData[0].keys())
     writer = csv.DictWriter(studentScheduleFile, fieldnames=csvFields)

play_ceremony_audio.py

@@ -30,8 +30,8 @@ def playAudio(filePath, name):
     audio = AudioSegment.from_mp3(filePath)
     play(audio)
 
-
-with open("data/final schedule with uuids.csv", encoding="utf-8") as scheduleFile:
+filename = input("Enter data filename (default: contacts.csv): ") or "contacts.csv"
+with open("data/{}".format(filename), encoding="utf-8") as scheduleFile:
     graduates = [
         {k: v for k, v in row.items()}
         for row in csv.DictReader(scheduleFile, skipinitialspace=True)

reset_timeslots.py

@@ -1,5 +1,32 @@
+import csv
+import pymysql.cursors
+import os
+from datetime import datetime
+from dotenv import load_dotenv
 
-```sql
-SET @row_number = 0; 
-UPDATE graduates SET timeslot = FROM_UNIXTIME(30 * (@row_number:=@row_number + 1) + UNIX_TIMESTAMP());
-```
+load_dotenv()
+
+# Connect to the database
+connection = pymysql.connect(
+    host=os.getenv("SQL_HOST"),
+    user=os.getenv("SQL_USER"),
+    password=os.getenv("SQL_PASSWORD"),
+    db=os.getenv("SQL_DB"),
+    charset="utf8mb4",
+    autocommit=True,
+    init_command="SET SESSION time_zone='+00:00'",
+    cursorclass=pymysql.cursors.DictCursor,
+)
+
+try:
+    with connection.cursor() as cursor:
+        # init temp variable
+        sql = "SET @row_number = 0"
+        cursor.execute(sql)
+
+        # use variable to update timeslots
+        sql = "UPDATE `graduates` SET `timeslot` = FROM_UNIXTIME(30 * (@row_number:=@row_number + 1) + UNIX_TIMESTAMP()) WHERE `graduated` = 0"
+        cursor.execute(sql)
+        print("> Reset timeslots")
+finally:
+    connection.close()