1
Fork 0
mirror of https://github.com/wlinator/luminara.git synced 2024-10-03 00:03:13 +00:00
Lumi/db/database.py

50 lines
996 B
Python

import sqlite3
from sqlite3 import Error
def create_connection():
try:
conn = sqlite3.connect("db/rcu.db")
except Error as e:
print("'create_connection()' Error occurred: {}".format(e))
return
return conn
def execute_query(query, values=None):
conn = create_connection()
cursor = conn.cursor()
if values:
cursor.execute(query, values)
else:
cursor.execute(query)
conn.commit()
return cursor
def select_query(query, values=None):
conn = create_connection()
cursor = conn.cursor()
if values:
return cursor.execute(query, values).fetchall()
else:
return cursor.execute(query).fetchall()
def select_query_one(query, values=None):
conn = create_connection()
cursor = conn.cursor()
if values:
output = cursor.execute(query, values).fetchone()
else:
output = cursor.execute(query).fetchone()
if output:
return output[0]
return None