minor changes

Author: offish

docs/usage/currency.py

@@ -1,6 +1,5 @@
 from tf2_utils import CurrencyExchange, Inventory, map_inventory
 
-
 inventory_provider = Inventory("steamcommunity")
 
 # Get our inventory of a user
@@ -26,7 +25,7 @@
 
 currency.calculate()
 
-if not currency.is_possible():
+if not currency.is_possible:
     print("Trade is not possible")
     # either no combination worked, or someone did not have enough pure
 

src/tf2_utils/inventory.py

@@ -8,7 +8,9 @@
 from .sku import get_sku
 
 
-def map_inventory(inventory: dict, add_skus: bool = False) -> list[dict]:
+def map_inventory(
+    inventory: dict, add_skus: bool = False, skip_untradable: bool = False
+) -> list[dict]:
     """Matches classids and instanceids, merges these and
     adds `sku` to each item entry if `add_skus` is enabled"""
     mapped_inventory = []
@@ -18,6 +20,9 @@ def map_inventory(inventory: dict, add_skus: bool = False) -> list[dict]:
 
     for asset in inventory["assets"]:
         for desc in inventory["descriptions"]:
+            if skip_untradable and not desc["tradable"]:
+                continue
+
             if (
                 asset["classid"] != desc["classid"]
                 or asset["instanceid"] != desc["instanceid"]

src/tf2_utils/item.py

@@ -9,7 +9,7 @@ def __init__(self, item: dict) -> None:
         self.tags = item.get("tags", [])
 
     def is_tf2(self) -> bool:
-        return self.item["appid"] == 440
+        return int(self.item["appid"]) == 440
 
     def is_tradable(self) -> bool:
         return self.item.get("tradable", 1) == 1

src/tf2_utils/sku.py

@@ -8,8 +8,9 @@
 __all__ = [
     "get_sku_properties",
     "is_sku",
-    "is_pure",
+    "is_key",
     "is_metal",
+    "is_pure",
     "get_metal",
     "get_properties",
     "get_property",
@@ -64,14 +65,18 @@ def is_sku(item: str) -> bool:
     return item.find(";") != -1
 
 
-def is_pure(sku: str) -> bool:
-    return sku in ["5000;6", "5001;6", "5002;6", "5021;6"]
+def is_key(sku: str) -> bool:
+    return sku == "5021;6"
 
 
 def is_metal(sku: str) -> bool:
     return sku in ["5000;6", "5001;6", "5002;6"]
 
 
+def is_pure(sku: str) -> bool:
+    return is_metal(sku) or is_key(sku)
+
+
 def get_metal(sku: str) -> int:
     assert is_metal(sku), f"sku {sku} is not metal"