randydev commited on
Commit
a81cdcd
·
verified ·
1 Parent(s): d8b2b9e

Update Akeno/utils/base_sqlite.py

Browse files
Files changed (1) hide show
  1. Akeno/utils/base_sqlite.py +7 -10
Akeno/utils/base_sqlite.py CHANGED
@@ -24,15 +24,12 @@ async def set_prefix_in_db(user_id: int, prefix: str):
24
  conn.commit()
25
 
26
  async def get_prefix(user_id: int):
27
- loop = asyncio.get_running_loop()
28
- result = await loop.run_in_executor(None, fetch_prefix_from_db, user_id)
29
- return result
30
 
31
- def fetch_prefix_from_db(user_id: int):
32
- with conn.cursor() as cursor:
33
- cursor.execute('SELECT prefix FROM prefixes WHERE user_id=?', (user_id,))
34
- result = cursor.fetchone()
35
- return result[0] if result else None
36
 
37
- async def close_connection():
38
- conn.close()
 
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:
31
+ return result[0]
32
+ else:
33
+ return None
 
34
 
35
+ conn.close()