ib-clearance/IA.py

#!/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()