FabioSantos commited on
Commit
e99aac7
·
verified ·
1 Parent(s): e93807d

Update utils/db.py

Browse files
Files changed (1) hide show
  1. 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