randydev commited on
Commit
ef6ddf3
·
verified ·
1 Parent(s): 60cf536

Update Akeno/utils/base_sqlite.py

Browse files
Files changed (1) hide show
  1. Akeno/utils/base_sqlite.py +5 -3
Akeno/utils/base_sqlite.py CHANGED
@@ -5,8 +5,10 @@ from config import SQLITE_CLOUD_URL
5
  conn = sqlitecloud.connect(SQLITE_CLOUD_URL)
6
  cursor = conn.cursor()
7
 
 
 
8
  cursor.execute('''
9
- CREATE TABLE IF NOT EXISTS prefixes_dev (
10
  user_id INTEGER PRIMARY KEY,
11
  prefix TEXT NOT NULL
12
  )
@@ -15,14 +17,14 @@ conn.commit()
15
 
16
  async def set_prefix_in_db(user_id: int, prefix: str):
17
  cursor.execute('''
18
- INSERT INTO prefixes_dev (user_id, prefix)
19
  VALUES (?, ?)
20
  ON CONFLICT(user_id) DO UPDATE SET prefix=excluded.prefix
21
  ''', (user_id, prefix))
22
  conn.commit()
23
 
24
  async def get_prefix(user_id: int):
25
- cursor.execute('SELECT prefix FROM prefixes_dev WHERE user_id=?', (user_id,))
26
  result = cursor.fetchone()
27
 
28
  if result:
 
5
  conn = sqlitecloud.connect(SQLITE_CLOUD_URL)
6
  cursor = conn.cursor()
7
 
8
+ cursor.execute('USE DATABASE akeno_dev')
9
+
10
  cursor.execute('''
11
+ CREATE TABLE IF NOT EXISTS prefixes (
12
  user_id INTEGER PRIMARY KEY,
13
  prefix TEXT NOT NULL
14
  )
 
17
 
18
  async def set_prefix_in_db(user_id: int, prefix: str):
19
  cursor.execute('''
20
+ INSERT INTO prefixes (user_id, prefix)
21
  VALUES (?, ?)
22
  ON CONFLICT(user_id) DO UPDATE SET prefix=excluded.prefix
23
  ''', (user_id, prefix))
24
  conn.commit()
25
 
26
  async def get_prefix(user_id: int):
27
+ cursor.execute('SELECT prefix FROM prefixes WHERE user_id=?', (user_id,))
28
  result = cursor.fetchone()
29
 
30
  if result: