resourcecap 0.2.0__tar.gz → 0.2.1__tar.gz

{resourcecap-0.2.0 → resourcecap-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: resourcecap
-Version: 0.2.0
+Version: 0.2.1
 Summary: A python package to limit the resource usage of any program
 Author: purushottam-dafure
 Author-email: purushottam-dafure <purushottamdafure@gmail.com>

{resourcecap-0.2.0 → resourcecap-0.2.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "resourcecap"
-version = "0.2.0"
+version = "0.2.1"
 description = "A python package to limit the resource usage of any program"
 readme = "README.md"
 license = "MIT"
@@ -17,9 +17,6 @@ classifiers = [
 name = "purushottam-dafure"
 email = "purushottamdafure@gmail.com"
 
-[project.scripts]
-resourcecap = "resourcecap:main"
-
 [build-system]
 requires = ["uv_build>=0.12.17,<0.13.0"]
 build-backend = "uv_build"

{resourcecap-0.2.0 → resourcecap-0.2.1}/pyproject.toml.orig

@@ -1,6 +1,6 @@
 [project]
 name = "resourcecap"
-version = "0.2.0"
+version = "0.2.1"
 description = "A python package to limit the resource usage of any program"
 readme = "README.md"
 license = "MIT"
@@ -16,9 +16,6 @@ classifiers = [
     "Typing :: Typed",
 ]
 
-[project.scripts]
-resourcecap = "resourcecap:main"
-
 [build-system]
 requires = ["uv_build>=0.12.17,<0.13.0"]
 build-backend = "uv_build"

{resourcecap-0.2.0 → resourcecap-0.2.1}/src/resourcecap/__init__.py

@@ -2,8 +2,4 @@ from ._budget import Budget, BudgetExhaustedError
 from ._costs import costs
 from ._resources import Spendable
 
-__all__ = ["Budget", "BudgetExhaustedError", "Spendable", "costs", "main"]
-
-
-def main() -> None:
-    print("Hello from resourcecap!")
+__all__ = ["Budget", "BudgetExhaustedError", "Spendable", "costs"]

{resourcecap-0.2.0 → resourcecap-0.2.1}/src/resourcecap/_budget.py

@@ -1,5 +1,6 @@
 import contextvars
 import logging
+import threading
 from collections.abc import Hashable, Mapping
 from types import TracebackType
 
@@ -39,6 +40,7 @@ class Budget:
     ) -> None:
         self.limits = merge_resources(amount, limits)
         self.spent: dict[Hashable, float] = {}
+        self._lock = threading.Lock()
         self._token: contextvars.Token[tuple[_tracking._Chargeable, ...]] | None = None
 
     def exhausts_at_start(self, key: Hashable) -> bool:
@@ -53,14 +55,15 @@ class Budget:
         (spend charged against it so far), and `remaining` (`allocated
         - used`, which can go negative for a `warn_only` resource).
         """
-        return {
-            key: {
-                "allocated": limit.amount,
-                "used": self.spent.get(key, 0.0),
-                "remaining": limit.amount - self.spent.get(key, 0.0),
+        with self._lock:
+            return {
+                key: {
+                    "allocated": limit.amount,
+                    "used": self.spent.get(key, 0.0),
+                    "remaining": limit.amount - self.spent.get(key, 0.0),
+                }
+                for key, limit in self.limits.items()
             }
-            for key, limit in self.limits.items()
-        }
 
     def charge(self, key: Hashable, amount: float) -> None:
         """Record spend against this budget's limit for `key`.
@@ -73,8 +76,10 @@ class Budget:
         if limit is None:
             return
 
-        spent = self.spent.get(key, 0.0) + amount
-        self.spent[key] = spent
+        with self._lock:
+            spent = self.spent.get(key, 0.0) + amount
+            self.spent[key] = spent
+
         if spent > limit.amount:
             if limit.warn_only:
                 logger.warning(
@@ -86,7 +91,8 @@ class Budget:
                 )
 
     def __enter__(self) -> "Budget":
-        self.spent = {}
+        with self._lock:
+            self.spent = {}
         self._token = _tracking.push(self)
         return self