why-tools 0.2.1__tar.gz → 0.2.2__tar.gz

{why_tools-0.2.1 → why_tools-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: why-tools
-Version: 0.2.1
+Version: 0.2.2
 Summary: 为什么还要使用别的工具呢？
 Home-page: https://github.com/yintian710/whytools
 Author: yintian

{why_tools-0.2.1 → why_tools-0.2.2}/why_tools.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: why-tools
-Version: 0.2.1
+Version: 0.2.2
 Summary: 为什么还要使用别的工具呢？
 Home-page: https://github.com/yintian710/whytools
 Author: yintian

{why_tools-0.2.1 → why_tools-0.2.2}/why_tools.egg-info/SOURCES.txt

@@ -36,6 +36,7 @@ ytools/utils/date.py
 ytools/utils/encrypt.py
 ytools/utils/extractors.py
 ytools/utils/file.py
+ytools/utils/host_ip.py
 ytools/utils/magic.py
 ytools/utils/magic_for_class.py
 ytools/utils/nodes.py

why_tools-0.2.2/ytools/VERSION

@@ -0,0 +1 @@
+0.2.2

{why_tools-0.2.1 → why_tools-0.2.2}/ytools/arq/client/base.py

@@ -6,6 +6,10 @@
 @Software: PyCharm
 @Desc    : 
 """
+import asyncio
+import json
+
+from ytools.utils.host_ip import get_local_ip
 from ytools.utils.magic import require
 
 require("redis")
@@ -27,11 +31,13 @@ class BaseClient:
             queue_name=None,
             redis=None
     ):
+        self.info = {}
         self.set_queue(queue_name)
         if isinstance(redis, dict):
             self.redis = self.make_redis(**redis)
         elif redis:
             self.redis = redis
+        asyncio.create_task(self.heartbeat())
 
     def set_queue(self, queue_name):
         self.queue_name = queue_name or setting.DEFAULT_QUEUE_NAME
@@ -74,6 +80,13 @@ class BaseClient:
         status_queue = self.get_queue(status_id, base=self.status_queue)
         return await self.redis.delete(status_queue)
 
+    async def heartbeat(self):
+        h_key = self.get_queue(self.__class__.__name__.lower(), get_local_ip(), base=self.queue_name)
+        interval = 5
+        while True:
+            await self.redis.set(h_key, value=json.dumps(self.info), ex=interval + 1)
+            await asyncio.sleep(interval)
+
 
 if __name__ == '__main__':
     pass

why_tools-0.2.2/ytools/utils/host_ip.py

@@ -0,0 +1,28 @@
+# -*- coding: utf-8 -*-
+"""
+@File    : host_ip.py
+@Author  : yintian
+@Date    : 2025/11/5 14:21
+@Software: PyCharm
+@Desc    : 
+"""
+import socket
+from functools import lru_cache
+
+
+@lru_cache()
+def get_local_ip():
+    try:
+        # 创建一个socket对象并连接到外部地址
+        s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+        # 使用Google的DNS服务器地址，无需实际发送数据
+        s.connect(("8.8.8.8", 80))
+        local_ip = s.getsockname()[0]
+        s.close()
+        return local_ip
+    except Exception as e:
+        raise RuntimeError(f"无法获取内网IP: {str(e)}")
+
+
+if __name__ == '__main__':
+    pass

why_tools-0.2.1/ytools/VERSION

@@ -1 +0,0 @@
-0.2.1