Spaces:
Sleeping
Sleeping
Update utils/db.py
Browse files- utils/db.py +20 -20
utils/db.py
CHANGED
|
@@ -1,20 +1,20 @@
|
|
| 1 |
-
import sqlite3
|
| 2 |
-
import pandas as pd
|
| 3 |
-
|
| 4 |
-
def connect():
|
| 5 |
-
return sqlite3.connect("db/aba_data.db")
|
| 6 |
-
|
| 7 |
-
def query_sessions(user_id, start=None, end=None):
|
| 8 |
-
conn = connect()
|
| 9 |
-
if start and end:
|
| 10 |
-
query = """
|
| 11 |
-
SELECT * FROM sessions WHERE user_id = ? AND date BETWEEN ? AND ? ORDER BY date
|
| 12 |
-
"""
|
| 13 |
-
df = pd.read_sql_query(query, conn, params=(user_id, start, end))
|
| 14 |
-
else:
|
| 15 |
-
query = """
|
| 16 |
-
SELECT * FROM sessions WHERE user_id = ? ORDER BY date
|
| 17 |
-
"""
|
| 18 |
-
df = pd.read_sql_query(query, conn, params=(user_id,))
|
| 19 |
-
conn.close()
|
| 20 |
-
return df
|
|
|
|
| 1 |
+
import sqlite3
|
| 2 |
+
import pandas as pd
|
| 3 |
+
|
| 4 |
+
def connect():
|
| 5 |
+
return sqlite3.connect("https://fabiosantos-teama.hf.space/db/aba_data.db")
|
| 6 |
+
|
| 7 |
+
def query_sessions(user_id, start=None, end=None):
|
| 8 |
+
conn = connect()
|
| 9 |
+
if start and end:
|
| 10 |
+
query = """
|
| 11 |
+
SELECT * FROM sessions WHERE user_id = ? AND date BETWEEN ? AND ? ORDER BY date
|
| 12 |
+
"""
|
| 13 |
+
df = pd.read_sql_query(query, conn, params=(user_id, start, end))
|
| 14 |
+
else:
|
| 15 |
+
query = """
|
| 16 |
+
SELECT * FROM sessions WHERE user_id = ? ORDER BY date
|
| 17 |
+
"""
|
| 18 |
+
df = pd.read_sql_query(query, conn, params=(user_id,))
|
| 19 |
+
conn.close()
|
| 20 |
+
return df
|