diff --git a/data.db b/data.db index f1b57cf..a7cb204 100644 Binary files a/data.db and b/data.db differ diff --git a/database.py b/database.py index 30867c2..daaac12 100644 --- a/database.py +++ b/database.py @@ -10,16 +10,6 @@ cur = con.cursor() cur.execute("CREATE TABLE IF NOT EXISTS TIMESTAMPS (session_id INTEGER PRIMARY KEY, time_in STRING, time_out STRING, name STRING, purpose STRING);") cur.execute("CREATE TABLE IF NOT EXISTS USERS (id INTEGER PRIMARY KEY, name STRING, cumulative_hours INTEGER);") -# declare and insert some initial values -timestamp_list = [ - ('Mikayla', 'work on project'), - ('not Mikayla', 'work on project'), - ('someone else', 'debugging') -] - -cur.executemany("INSERT INTO TIMESTAMPS (name, purpose) VALUES (?, ?)", timestamp_list) -con.commit() - # Functions to define: # 1) Select all timestamps # 2) Find a timestamp by date range @@ -31,6 +21,14 @@ def create_new_stamp(timestamp, name, purpose): cur.execute("INSERT INTO TIMESTAMPS (time_in, name, purpose) VALUES (?, ?, ?)", to_insert) con.commit() +def punch_out(timestamp, id): + data = (timestamp, id) + cur.execute("""UPDATE TIMESTAMPS + SET time_out = ? + WHERE session_id = ? + """, data) + con.commit() + def get_all_stamps(): return cur.execute("SELECT * FROM TIMESTAMPS;") diff --git a/user_input.py b/user_input.py index 15488d8..df43be4 100644 --- a/user_input.py +++ b/user_input.py @@ -1,4 +1,5 @@ from database import * +from datetime import * user_prompt = """ Welcome to the personal time stamp program. @@ -15,10 +16,10 @@ Please choose from the following options: # Inner functions detailed below: def handle_first_option(): - time_of_stamp = datetime.now() in_out = input("Punching in or out? i/o \n") if in_out == 'i': + time_of_stamp = datetime.now() print("Creating new timestamp and punching in!") print("Please enter the following details:") name = input("Your name: ") @@ -43,7 +44,12 @@ def handle_first_option(): print("Invalid input.") handle_first_option() elif in_out == 'o': - print("Punching out.") + time_of_stamp = datetime.now() + print("Preparing to punch out.") + session = input("Please provide your session ID." ) + punch_out(time_of_stamp, session) + print("Punch out successful. Returning...") + parse_input() else: print("Invalid input. Please try again:") handle_first_option()