tubearchivist
/
tubearchivist
mirror of https://github.com/tubearchivist/tubearchivist.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

improve comment notify timeouts

This commit is contained in:
simon 2023-03-04 17:37:58 +07:00
parent c204c6ca7f
commit ee568534f6
No known key found for this signature in database
GPG Key ID: 2C15AA5E89985DD4
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tubearchivist/home/src/index/comments.py
View File

@ -210,6 +210,9 @@ class CommentList:
return
total_videos = len(self.video_ids)
if notify:
self._notify(f"add comments for {total_videos} videos", False)
for idx, video_id in enumerate(self.video_ids):
comment = Comments(video_id, config=self.config)
if notify:
@ -219,16 +222,16 @@ class CommentList:
comment.upload_comments()
if notify:
self.notify_final(total_videos)
self._notify(f"added comments for {total_videos} videos", 5)
@staticmethod
def notify_final(total_videos):
"""send final notification"""
def _notify(message, expire):
"""send notification"""
key = "message:download"
message = {
"status": key,
"level": "info",
"title": "Download and index comments finished",
"message": f"added comments for {total_videos} videos",
"message": message,
}
RedisArchivist().set_message(key, message, expire=4)
RedisArchivist().set_message(key, message, expire=expire)