ping with revision

ytdlbot/constant.py

@@ -128,14 +128,22 @@ def get_receive_link_text():
 
     @staticmethod
     def ping_worker():
+        from tasks import app as celery_app
         workers = InfluxDB().extract_dashboard_data()
+        # [{'celery@BennyのMBP': 'abc'}, {'celery@BennyのMBP': 'abc'}]
+        response = celery_app.control.broadcast("ping_revision", reply=True)
+        revision = {}
+        for item in response:
+            revision.update(item)
+
         text = ""
         for worker in workers:
             fields = worker["fields"]
             hostname = worker["tags"]["hostname"]
             status = {True: "✅"}.get(fields["status"], "❌")
             active = fields["active"]
             load = "{},{},{}".format(fields["load1"], fields["load5"], fields["load15"])
-            text += f"{status}{hostname} **{active}** {load}\n"
-        return text
+            rev = revision.get(hostname, "")
+            text += f"{status}{hostname} **{active}** {load} {rev}\n"
 
+        return text

ytdlbot/tasks.py

@@ -35,7 +35,7 @@
                         ytdl_download)
 from limit import VIP
 from utils import (apply_log_formatter, auto_restart, customize_logger,
-                   get_metadata, get_user_settings)
+                   get_metadata, get_revision, get_user_settings)
 
 customize_logger(["pyrogram.client", "pyrogram.session.session", "pyrogram.connection.connection"])
 apply_log_formatter()
@@ -321,6 +321,11 @@ def ytdl_normal_download(bot_msg, client, url):
     temp_dir.cleanup()
 
 
+@Panel.register
+def ping_revision(*args):
+    return get_revision()
+
+
 @Panel.register
 def hot_patch(*args):
     app_path = pathlib.Path().cwd().parent