-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdb.py
49 lines (44 loc) · 1.25 KB
/
db.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
import sqlite3
def get_db_connection():
conn = sqlite3.connect('standupbot.db')
conn.execute("PRAGMA foreign_keys = ON")
return conn
# Function to Insert/Update/Delete
def execute_query(query, params=None):
if params is None:
params = ()
try:
conn = get_db_connection()
cursor = conn.cursor()
cursor.execute(query, params)
conn.commit()
except sqlite3.DatabaseError as e:
print(f"DatabaseError: {e}")
finally:
conn.close()
# Function to fetch one record from the database
def fetch_one(query, params=None):
if params is None:
params = ()
try:
conn = get_db_connection()
cursor = conn.cursor()
cursor.execute(query, params)
return cursor.fetchone()
except sqlite3.DatabaseError as e:
print(f"DatabaseError: {e}")
finally:
conn.close()
# Function to fetch multiple records from the database
def fetch_all(query, params=None):
if params is None:
params = ()
try:
conn = get_db_connection()
cursor = conn.cursor()
cursor.execute(query, params)
return cursor.fetchall()
except sqlite3.DatabaseError as e:
print(f"DatabaseError: {e}")
finally:
conn.close()