mirror of
https://github.com/SpudGunMan/meshing-around.git
synced 2026-03-28 17:32:36 +01:00
107 lines
4.0 KiB
Python
107 lines
4.0 KiB
Python
# Checkin Checkout database module for the bot
|
|
# K7MHI Kelly Keeton 2024
|
|
|
|
import sqlite3
|
|
from modules.log import *
|
|
import time
|
|
|
|
trap_list_checklist = ("checkin", "checkout", "checklist", "purgein", "purgeout")
|
|
|
|
def initialize_checklist_database():
|
|
# create the database
|
|
conn = sqlite3.connect(checklist_db)
|
|
c = conn.cursor()
|
|
# Check if the checkin table exists, and create it if it doesn't
|
|
c.execute('''CREATE TABLE IF NOT EXISTS checkin
|
|
(checkin_id INTEGER PRIMARY KEY, checkin_name TEXT, checkin_date TEXT, checkin_time TEXT, checkin_notes TEXT)''')
|
|
# Check if the checkout table exists, and create it if it doesn't
|
|
c.execute('''CREATE TABLE IF NOT EXISTS checkout
|
|
(checkout_id INTEGER PRIMARY KEY, checkout_name TEXT, checkout_date TEXT, checkout_time TEXT, checkout_notes TEXT)''')
|
|
conn.commit()
|
|
conn.close()
|
|
logger.debug("System: Ensured data/checklist.db exists with required tables")
|
|
|
|
def checkin(name, date, time, notes):
|
|
# checkin a user
|
|
conn = sqlite3.connect(checklist_db)
|
|
c = conn.cursor()
|
|
try:
|
|
c.execute("INSERT INTO checkin (checkin_name, checkin_date, checkin_time, checkin_notes) VALUES (?, ?, ?, ?)", (name, date, time, notes))
|
|
except sqlite3.OperationalError as e:
|
|
if "no such table" in str(e):
|
|
initialize_checklist_database()
|
|
c.execute("INSERT INTO checkin (checkin_name, checkin_date, checkin_time, checkin_notes) VALUES (?, ?, ?, ?)", (name, date, time, notes))
|
|
else:
|
|
raise
|
|
conn.commit()
|
|
conn.close()
|
|
return "Checked in: "
|
|
|
|
def delete_checkin(checkin_id):
|
|
# delete a checkin
|
|
conn = sqlite3.connect(checklist_db)
|
|
c = conn.cursor()
|
|
c.execute("DELETE FROM checkin WHERE checkin_id = ?", (checkin_id,))
|
|
conn.commit()
|
|
conn.close()
|
|
return "Checkin deleted."
|
|
|
|
def checkout(name, date, time, notes):
|
|
# checkout a user
|
|
conn = sqlite3.connect(checklist_db)
|
|
c = conn.cursor()
|
|
try:
|
|
c.execute("INSERT INTO checkout (checkout_name, checkout_date, checkout_time, checkout_notes) VALUES (?, ?, ?, ?)", (name, date, time, notes))
|
|
except sqlite3.OperationalError as e:
|
|
if "no such table" in str(e):
|
|
initialize_checklist_database()
|
|
c.execute("INSERT INTO checkout (checkout_name, checkout_date, checkout_time, checkout_notes) VALUES (?, ?, ?, ?)", (name, date, time, notes))
|
|
else:
|
|
raise
|
|
conn.commit()
|
|
conn.close()
|
|
return "Checked out: "
|
|
|
|
def delete_checkout(checkout_id):
|
|
# delete a checkout
|
|
conn = sqlite3.connect(checklist_db)
|
|
c = conn.cursor()
|
|
c.execute("DELETE FROM checkout WHERE checkout_id = ?", (checkout_id,))
|
|
conn.commit()
|
|
conn.close()
|
|
return "Checkout deleted."
|
|
|
|
def list_checkin():
|
|
# list checkins
|
|
conn = sqlite3.connect(checklist_db)
|
|
c = conn.cursor()
|
|
c.execute("""
|
|
SELECT * FROM checkin
|
|
WHERE checkin_id NOT IN (
|
|
SELECT checkin_id FROM checkout
|
|
WHERE checkout_date > checkin_date OR (checkout_date = checkin_date AND checkout_time > checkin_time)
|
|
)
|
|
""")
|
|
rows = c.fetchall()
|
|
conn.close()
|
|
checkin_list = ""
|
|
for row in rows:
|
|
checkin_list += "Checkin ID: " + str(row[0]) + " Name: " + row[1] + " Date: " + row[2] + " Time: " + row[3] + " Notes: " + row[4] + "\n"
|
|
return checkin_list
|
|
|
|
def handle_checklist(nodeID, message):
|
|
current_date = time.strftime("%Y-%m-%d")
|
|
current_time = time.strftime("%H:%M:%S")
|
|
# handle checklist commands
|
|
if "checkin" in message.lower():
|
|
return checkin(nodeID, current_date, current_time, message.split(" ", 1)[1])
|
|
elif "checkout" in message.lower():
|
|
return checkout(nodeID, current_date, current_time, message.split(" ", 1)[1])
|
|
elif "purgein" in message.lower():
|
|
return delete_checkin(nodeID)
|
|
elif "purgeout" in message.lower():
|
|
return delete_checkout(nodeID)
|
|
elif "checklist" in message.lower():
|
|
return list_checkin()
|
|
else:
|
|
return "Invalid command." |