58 lines
1.5 KiB
Python
58 lines
1.5 KiB
Python
import sqlite3
|
|
from user_input import *
|
|
from datetime import datetime
|
|
|
|
# establish a connection to a .db file and create a cursor object
|
|
con = sqlite3.connect('data.db')
|
|
cur = con.cursor()
|
|
|
|
# create initial tables
|
|
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 = [
|
|
(1, '1pm', 'null', 'Mikayla', 'work on project'),
|
|
(2, '9pm', '11pm', 'not Mikayla', 'work on project'),
|
|
(3, '8am', '5pm', 'someone else', 'debugging')
|
|
]
|
|
|
|
cur.executemany("INSERT INTO TIMESTAMPS VALUES (?, ?, ?, ?, ?)", timestamp_list)
|
|
|
|
# Functions to define:
|
|
# 1) Select all timestamps
|
|
# 2) Find a timestamp by date range
|
|
# 3) Calculate total hours for the week
|
|
# 4) Calculate complete sum of hours
|
|
|
|
def create_new_stamp():
|
|
current_time = ''
|
|
|
|
def get_all_stamps():
|
|
return cur.execute("SELECT * FROM TIMESTAMPS;")
|
|
|
|
|
|
def get_number_of_stamps(input):
|
|
table_rows = []
|
|
for row in cur.execute("SELECT * FROM TIMESTAMPS;"):
|
|
table_rows.append(row)
|
|
|
|
return table_rows[:input]
|
|
|
|
|
|
def get_table_length():
|
|
table_rows = []
|
|
for row in cur.execute("SELECT * FROM TIMESTAMPS;"):
|
|
table_rows.append(row)
|
|
return len(table_rows)
|
|
|
|
|
|
def get_stamp_by_date():
|
|
pass
|
|
|
|
def get_weekly_hours():
|
|
pass
|
|
|
|
def get_all_hours():
|
|
pass
|