Fix time not change

byeice · Aug 31, 2022 · e72329a · e72329a
1 parent 42c76a5
commit e72329a
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/db/task.py b/db/task.py
@@ -87,7 +87,7 @@ async def get(self, id, fields=None, one_or_none=False, first=True, to_dict=True
             return self.to_dict(result,fields)
         return result
 
-    async def list(self, userid=None, fields=None, limit=1000, to_dict=True, scan=False, scan_time=time.time(), sql_session=None, **kwargs):
+    async def list(self, userid=None, fields=None, limit=1000, to_dict=True, scan=False, scan_time=None, sql_session=None, **kwargs):
         if fields is None:
             _fields = Task
         else:
@@ -97,7 +97,7 @@ async def list(self, userid=None, fields=None, limit=1000, to_dict=True, scan=Fa
         if userid is not None:
             smtm = smtm.where(Task.userid == userid)
 
-        if scan:
+        if scan and scan_time is not None:
             smtm = smtm.where(Task.next <= scan_time)
 
         for key, value in kwargs.items():
@@ -114,5 +114,7 @@ async def list(self, userid=None, fields=None, limit=1000, to_dict=True, scan=Fa
     def delete(self, id, sql_session=None):
         return self._delete(delete(Task).where(Task.id == id), sql_session=sql_session)
 
-    async def scan(self, now=time.time(), fields=None, sql_session=None):
+    async def scan(self, now=None, fields=None, sql_session=None):
+        if now is None:
+            now = time.time()
         return await self.list(fields=fields, scan=True, scan_time=now, limit=None, sql_session=sql_session)