1
Fork 0
mirror of https://github.com/wlinator/luminara.git synced 2024-10-02 20:23:12 +00:00
Lumi/db/database.py
2023-07-02 08:20:16 -04:00

53 lines
1 KiB
Python

import logging
import sqlite3
from sqlite3 import Error
racu_logs = logging.getLogger('Racu.Core')
def create_connection():
try:
conn = sqlite3.connect("db/rcu.db")
except Error as e:
racu_logs.error("'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