108 lines
2.6 KiB
Python
108 lines
2.6 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import sqlite3
|
||
|
|
import time
|
||
|
|
|
||
|
|
|
||
|
|
def connect_database(file):
|
||
|
|
"""
|
||
|
|
Create and return Connection object
|
||
|
|
:param file: database file
|
||
|
|
"""
|
||
|
|
conn = sqlite3.connect(file)
|
||
|
|
return conn
|
||
|
|
|
||
|
|
|
||
|
|
def get_lower_weekday():
|
||
|
|
"""
|
||
|
|
Return current weekday's name as lowercase string
|
||
|
|
"""
|
||
|
|
day = time.strftime("%A")
|
||
|
|
lower_day = day.lower()
|
||
|
|
return lower_day
|
||
|
|
|
||
|
|
|
||
|
|
def check_id(conn, id):
|
||
|
|
"""
|
||
|
|
Return True if existing ID, False is not.
|
||
|
|
:param conn: Connection object
|
||
|
|
:param id: ID of student, int
|
||
|
|
"""
|
||
|
|
cur = conn.cursor()
|
||
|
|
cur.execute("SELECT id FROM students")
|
||
|
|
ids = cur.fetchall()
|
||
|
|
|
||
|
|
for x in ids:
|
||
|
|
if id == x[0]:
|
||
|
|
return True
|
||
|
|
else:
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
def select_final_time(conn, day, id):
|
||
|
|
"""
|
||
|
|
Return ending time of final lesson student must attend, or None.
|
||
|
|
:param conn: Connection object
|
||
|
|
:param day: current weekday, lowercase str
|
||
|
|
:param id: ID of student, int
|
||
|
|
"""
|
||
|
|
cur = conn.cursor()
|
||
|
|
cur.execute("""
|
||
|
|
SELECT end_time
|
||
|
|
FROM timetable
|
||
|
|
INNER JOIN students ON g1 = lesson_id
|
||
|
|
OR g2 = lesson_id
|
||
|
|
OR g3 = lesson_id
|
||
|
|
OR g4 = lesson_id
|
||
|
|
OR g5 = lesson_id
|
||
|
|
OR g6 = lesson_id
|
||
|
|
WHERE id = ? AND day = ?
|
||
|
|
ORDER BY end_time DESC
|
||
|
|
LIMIT 1
|
||
|
|
""", (id, day))
|
||
|
|
finish = cur.fetchone()
|
||
|
|
if str(type(finish)) == "<class 'tuple'>":
|
||
|
|
return finish[0]
|
||
|
|
else:
|
||
|
|
return finish
|
||
|
|
|
||
|
|
|
||
|
|
def parse_time_string(timestring):
|
||
|
|
"""
|
||
|
|
Parse given 24h time string of format 'HH:MM' into time_struct
|
||
|
|
:param timestring: time string 'HH:MM'
|
||
|
|
"""
|
||
|
|
timestruct = time.strptime(timestring, "%H:%M")
|
||
|
|
return timestruct
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
db = input(
|
||
|
|
"Please enter name of database (located in the same folder as the program): ")
|
||
|
|
|
||
|
|
y = "y"
|
||
|
|
while y == "y":
|
||
|
|
conn = connect_database(db)
|
||
|
|
day = get_lower_weekday()
|
||
|
|
id = int(input("Please input the student ID number: "))
|
||
|
|
|
||
|
|
if check_id(conn, id):
|
||
|
|
raw = select_final_time(conn, day, id)
|
||
|
|
final = parse_time_string(raw) if raw != None else None
|
||
|
|
|
||
|
|
if final != None:
|
||
|
|
current_time = parse_time_string(time.strftime("%H:%M"))
|
||
|
|
|
||
|
|
if current_time > final:
|
||
|
|
print("Clear to leave.")
|
||
|
|
else:
|
||
|
|
print("Student does not have clearance.")
|
||
|
|
elif final == None:
|
||
|
|
print("Clear to leave.")
|
||
|
|
|
||
|
|
y = input("Press y and enter to run again.")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|