Selaa lähdekoodia

newContentIdTask.py
add info: 每次任务获取任务数量详情

罗俊辉 6 kuukautta sitten
vanhempi
commit
0ab5c09429
2 muutettua tiedostoa jossa 8 lisäystä ja 1 poistoa
  1. 7 0
      applications/db/__init__.py
  2. 1 1
      newContentIdTask.py

+ 7 - 0
applications/db/__init__.py

@@ -69,6 +69,13 @@ class AsyncMySQLClient(object):
                     await coon.rollback()
                     raise
 
+    async def __aenter__(self):
+        await self.init_pool()
+        return self
+
+    async def __aexit__(self, exc_type, exc_val, exc_tb):
+        await self.close_pool()
+
 
 class TaskMySQLClient(object):
     """

+ 1 - 1
newContentIdTask.py

@@ -18,7 +18,7 @@ async def main_job():
     # new_content_id_task = NewContentIdTask(async_mysql_pool)
     # await new_content_id_task.deal()
     async with AsyncMySQLClient() as async_mysql_pool:
-        await async_mysql_pool.init_pool()
+        # await async_mysql_pool.init_pool()
         new_content_id_task = NewContentIdTask(async_mysql_pool)
         await new_content_id_task.deal()