qa1145 commited on
Commit
c896e8b
·
verified ·
1 Parent(s): cce8e49

Upload sqlite.py

Browse files
Files changed (1) hide show
  1. astrbot/core/db/sqlite.py +5 -3
astrbot/core/db/sqlite.py CHANGED
@@ -65,9 +65,11 @@ class SQLiteDatabase(BaseDatabase):
65
 
66
  async def _create_default_api_key(self) -> None:
67
  """Create a default developer API key if none exists."""
68
- keys = await self.list_api_keys()
69
- if keys:
70
- return
 
 
71
 
72
  raw_key = "abk_astrbot"
73
  key_hash = hashlib.pbkdf2_hmac(
 
65
 
66
  async def _create_default_api_key(self) -> None:
67
  """Create a default developer API key if none exists."""
68
+ async with self.engine.begin() as conn:
69
+ result = await conn.execute(text("SELECT COUNT(*) FROM api_keys"))
70
+ count = result.scalar()
71
+ if count > 0:
72
+ return
73
 
74
  raw_key = "abk_astrbot"
75
  key_hash = hashlib.pbkdf2_hmac(