update for punch out works

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;")
 

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()