fix: include paintKit in gen code for sticker slab keychains

Author: vlydev

cs2_inspect/__init__.py

@@ -15,5 +15,10 @@
 
 from .inspect_link import deserialize, is_classic, is_masked, serialize
 from .models import ItemPreviewData, Sticker
+from .gen_codes import generate, gen_code_from_link, to_gen_code, parse_gen_code
 
-__all__ = ["serialize", "deserialize", "is_masked", "is_classic", "ItemPreviewData", "Sticker"]
+__all__ = [
+    "serialize", "deserialize", "is_masked", "is_classic",
+    "ItemPreviewData", "Sticker",
+    "generate", "gen_code_from_link", "to_gen_code", "parse_gen_code",
+]

cs2_inspect/gen_codes.py

@@ -41,6 +41,8 @@ def _serialize_sticker_pairs(stickers: list[Sticker], pad_to: Optional[int] = No
         for s in sorted(filtered, key=lambda x: x.slot):
             result.append(str(s.sticker_id))
             result.append(_format_float(float(s.wear) if s.wear is not None else 0.0))
+            if s.paint_kit is not None:
+                result.append(str(s.paint_kit))
 
     return result
 

tests/test_gen_codes.py

@@ -44,6 +44,27 @@ def test_zero_wear_float_format(self):
         code = to_gen_code(item)
         assert code == "!gen 7 474 306 0"
 
+    def test_keychain_with_paint_kit_appends_paint_kit(self):
+        item = ItemPreviewData(
+            defindex=1355, paintindex=0, paintseed=0, paintwear=0.0,
+            keychains=[Sticker(slot=0, sticker_id=37, wear=0.0, paint_kit=929)],
+        )
+        code = to_gen_code(item, prefix="")
+        tokens = code.split()
+        assert tokens[-3] == "37"
+        assert tokens[-2] == "0"
+        assert tokens[-1] == "929"
+
+    def test_keychain_without_paint_kit_no_extra_token(self):
+        item = ItemPreviewData(
+            defindex=7, paintindex=0, paintseed=0, paintwear=0.0,
+            keychains=[Sticker(slot=0, sticker_id=36, wear=0.0)],
+        )
+        code = to_gen_code(item, prefix="")
+        tokens = code.split()
+        assert tokens[-2] == "36"
+        assert tokens[-1] == "0"
+
 
 class TestParseGenCode:
     def test_basic_parse(self):
@@ -71,6 +92,17 @@ def test_parse_with_sticker_and_keychain(self):
         assert item.keychains[0].sticker_id == 36
 
 
+class TestGenCodeFromLinkSlab:
+    def test_mousesports_slab_url_ends_with_paint_kit(self):
+        from cs2_inspect import gen_code_from_link
+        slab_url = "steam://run/730//+csgo_econ_action_preview%20819181994A8BA181A982B189E981F181238086898191A4E1208698F309C9"
+        code = gen_code_from_link(slab_url, prefix="")
+        tokens = code.split()
+        assert tokens[-3] == "37"
+        assert tokens[-2] == "0"
+        assert tokens[-1] == "929"
+
+
 class TestGenCodeFromLink:
     def test_from_hex(self):
         from cs2_inspect import generate, gen_code_from_link