AIL/ail-framework
7
0
Fork 0
mirror of https://github.com/ail-project/ail-framework.git synced 2024-11-30 01:37:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: [crawler] fix undefined capture status

Browse source
This commit is contained in:
Terrtia 2023-05-25 16:26:48 +02:00
parent 9a4feb93a0
commit 405d097024
No known key found for this signature in database
GPG key ID: 1E1B1F50D84613D0
1 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
bin/lib/crawlers.py
View file

@ -1091,7 +1091,10 @@ class CrawlerCapture:
return self.get_task().get_start_time() return self.get_task().get_start_time()
def get_status(self): def get_status(self):
return r_cache.hget(f'crawler:capture:{self.uuid}', 'status') status = r_cache.hget(f'crawler:capture:{self.uuid}', 'status')
if not status:
status = -1
return status
def is_ongoing(self): def is_ongoing(self):
return self.get_status() == CaptureStatus.ONGOING return self.get_status() == CaptureStatus.ONGOING
@ -1109,7 +1112,7 @@ class CrawlerCapture:
def update(self, status): def update(self, status):
# Error or Reload # Error or Reload
if not status: if not status:
r_cache.hset(f'crawler:capture:{self.uuid}', 'status', CaptureStatus.UNKNOWN) r_cache.hset(f'crawler:capture:{self.uuid}', 'status', CaptureStatus.UNKNOWN.value)
r_cache.zadd('crawler:captures', {self.uuid: 0}) r_cache.zadd('crawler:captures', {self.uuid: 0})
else: else:
last_check = int(time.time()) last_check = int(time.time())