浏览代码

callback fixes

Hal De 4 年之前
父节点
当前提交
e7551be8b0
共有 1 个文件被更改,包括 3 次插入3 次删除
  1. 3 3
      app/app.py

+ 3 - 3
app/app.py

@@ -40,9 +40,6 @@ class PintDB:
   def __init__(self, app: Optional[Pint] = None) -> None:
     self.init_app(app)
     self._db = Database(app.config["DB_URI"])
-    self._db.connect()
-    self._db.execute(query='CREATE TABLE IF NOT EXISTS callback_urls (device VARCHAR(16) PRIMARY KEY, url VARCHAR(255))')
-    self._db.disconnect()
 
   def init_app(self, app: Pint) -> None:
     app.before_serving(self._before_serving)
@@ -141,6 +138,9 @@ async def reloadCallback(mngr: Manager, msg: Message):
   await refreshStatesCache()
   await refreshQueuesCache()
   await rebindLostDevices()
+  await db.connect()
+  await db.execute(query='CREATE TABLE IF NOT EXISTS callback_urls (device VARCHAR(16) PRIMARY KEY, url VARCHAR(255))')
+  await db.disconnect()
 
 @manager.register_event('ExtensionStatus')
 async def extensionStatusCallback(mngr: Manager, msg: Message):