async-task-kit 0.1.1__tar.gz → 0.1.2__tar.gz

{async_task_kit-0.1.1 → async_task_kit-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: async-task-kit
-Version: 0.1.1
+Version: 0.1.2
 Summary: A powerful async task processing kit based on RabbitMQ with Coroutine, Thread, and Process support.
 Author-email: realwrtoff <realwrtoff@gmail.com>
 Classifier: Programming Language :: Python :: 3
@@ -10,6 +10,7 @@ Requires-Python: >=3.12
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: aio-pika>=9.4.0
+Requires-Dist: asyncpg>=0.31.0
 Requires-Dist: python-dotenv>=1.0.0
 Dynamic: license-file
 

async_task_kit-0.1.2/async_task_kit/utils/env_loader.py

@@ -0,0 +1,55 @@
+from __future__ import annotations
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
+
+
+class EnvLoader:
+    def __init__(self, task_id: str | None = None):
+        self.prefix = f"{task_id.upper()}_" if task_id else ""
+
+    def _candidate_keys(self, key: str) -> list[str]:
+        uk = key.upper()
+        if self.prefix:
+            # 有任务前缀：优先前缀变量，其次全局公共变量
+            return [self.prefix + uk, uk]
+        # 无task_id，只查全局公共变量
+        return [uk]
+
+    def get(self, key: str, default: str | None = None) -> str | None:
+        for k in self._candidate_keys(key):
+            val = os.getenv(k)
+            if val is not None:
+                return val.strip()
+        return default
+
+    def get_int(self, key: str, default: int | None = None) -> int | None:
+        raw_val = None
+        for k in self._candidate_keys(key):
+            raw_val = os.getenv(k)
+            if raw_val is not None:
+                break
+        if raw_val is None:
+            return default
+        try:
+            return int(raw_val.strip())
+        except (ValueError, TypeError):
+            return default
+
+    def get_bool(self, key: str, default: bool = False) -> bool:
+        raw_val = None
+        for k in self._candidate_keys(key):
+            raw_val = os.getenv(k)
+            if raw_val is not None:
+                break
+        if raw_val is None:
+            return default
+        v = raw_val.strip().lower()
+        true_set = {"1", "true", "yes", "on", "y"}
+        false_set = {"0", "false", "no", "off", "n"}
+        if v in true_set:
+            return True
+        if v in false_set:
+            return False
+        return default

{async_task_kit-0.1.1 → async_task_kit-0.1.2}/async_task_kit.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: async-task-kit
-Version: 0.1.1
+Version: 0.1.2
 Summary: A powerful async task processing kit based on RabbitMQ with Coroutine, Thread, and Process support.
 Author-email: realwrtoff <realwrtoff@gmail.com>
 Classifier: Programming Language :: Python :: 3
@@ -10,6 +10,7 @@ Requires-Python: >=3.12
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: aio-pika>=9.4.0
+Requires-Dist: asyncpg>=0.31.0
 Requires-Dist: python-dotenv>=1.0.0
 Dynamic: license-file
 

{async_task_kit-0.1.1 → async_task_kit-0.1.2}/async_task_kit.egg-info/requires.txt

@@ -1,2 +1,3 @@
 aio-pika>=9.4.0
+asyncpg>=0.31.0
 python-dotenv>=1.0.0

{async_task_kit-0.1.1 → async_task_kit-0.1.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "async-task-kit"
-version = "0.1.1"
+version = "0.1.2"
 description = "A powerful async task processing kit based on RabbitMQ with Coroutine, Thread, and Process support."
 readme = "README.md"
 requires-python = ">=3.12"
@@ -13,7 +13,8 @@ authors = [
 ]
 dependencies = [
   "aio-pika>=9.4.0",
-  "python-dotenv>=1.0.0"
+  "asyncpg>=0.31.0",
+  "python-dotenv>=1.0.0",
 ]
 classifiers = [
     "Programming Language :: Python :: 3",
@@ -21,6 +22,9 @@ classifiers = [
     "Operating System :: OS Independent",
 ]
 
+[dependency-groups]
+dev = ["build", "twine"]
+
 [tool.setuptools.packages.find]
 where = ["."]
-include = ["async_task_kit*"]
+include = ["async_task_kit*"]

async_task_kit-0.1.1/async_task_kit/utils/env_loader.py

@@ -1,16 +0,0 @@
-import os
-from dotenv import load_dotenv
-load_dotenv()
-
-class EnvLoader:
-    def __init__(self, task_id=None):
-        self.prefix = f"{task_id}_".upper() if task_id else ""
-
-    def get(self, key, default=None):
-        return os.getenv(f"{self.prefix}{key}", default)
-
-    def get_int(self, key, default=1):
-        try:
-            return int(os.getenv(f"{self.prefix}{key}", default))
-        except:
-            return default