adding sun & moon series (#251)

Author: digitalghost-dev

card_data/pipelines/defs/extract/tcgcsv/extract_pricing.py

@@ -48,6 +48,22 @@
     "swsh3": "2675",
     "swsh2": "2626",
     "swsh1": "2585",
+    # Sun & Moon
+    "sm12": "2534",
+    "sm115": "2480",
+    "sm11": "2464",
+    "sm10": "2420",
+    "sm9": "2377",
+    "sm8": "2328",
+    "sm7.5": "2295",
+    "sm7": "2278",
+    "sm6": "2209",
+    "sm5": "2178",
+    "sm4": "2071",
+    "sm3.5": "2054",
+    "sm3": "1957",
+    "sm2": "1919",
+    "sm1": "1863"
 }
 
 
@@ -74,6 +90,18 @@ def get_card_number(card: dict) -> Optional[str]:
     return None
 
 
+def normalize_card_number(card_number: str) -> str:
+    """Zero-pad numeric parts of a card number to 3 digits.
+    e.g. '1/149' -> '001/149', '10/149' -> '010/149'.
+    """
+    if "/" in card_number:
+        left, right = card_number.split("/", 1)
+        left = left.zfill(3) if left.isdigit() else left
+        right = right.zfill(3) if right.isdigit() else right
+        return f"{left}/{right}"
+    return card_number
+
+
 def extract_card_name(full_name: str) -> str:
     """Extract clean card name, removing variant information after dash and parenthetical suffixes"""
 
@@ -151,6 +179,9 @@ def pull_product_information(set_number: str) -> pl.DataFrame:
         if card_number is None:
             continue
 
+        if set_number.startswith("sm"):
+            card_number = normalize_card_number(card_number)
+
         card_info = {
             "product_id": card["productId"],
             "name": extract_card_name(card_name),

card_data/pipelines/defs/extract/tcgdex/extract_series.py

@@ -31,6 +31,6 @@ def extract_series_data() -> pl.DataFrame:
         raise
 
     filtered = [
-        s.model_dump(mode="json") for s in validated if s.id in ["swsh", "sv", "me"]
+        s.model_dump(mode="json") for s in validated if s.id in ["me", "sv", "swsh", "sm"]
     ]
     return pl.DataFrame(filtered)

card_data/pipelines/defs/extract/tcgdex/extract_sets.py

@@ -24,6 +24,7 @@ def extract_sets_data() -> pl.DataFrame:
         "https://api.tcgdex.net/v2/en/series/me",
         "https://api.tcgdex.net/v2/en/series/sv",
         "https://api.tcgdex.net/v2/en/series/swsh",
+        "https://api.tcgdex.net/v2/en/series/sm",
     ]
 
     flat: list[dict] = []

card_data/pipelines/poke_cli_dbt/models/sets.sql

@@ -5,4 +5,4 @@
 
 SELECT *
 FROM {{ source('staging', 'sets') }}
-WHERE set_id NOT IN ('fut2020', 'mep', 'svp', 'swshp')
+WHERE set_id NOT IN ('fut2020', 'mep', 'svp', 'swshp', 'cel25', 'sma', 'smp', 'det1')

cmd/card/serieslist.go

@@ -9,6 +9,7 @@ var seriesIDMap = map[string]string{
 	"Mega Evolution":   "me",
 	"Scarlet & Violet": "sv",
 	"Sword & Shield":   "swsh",
+	"Sun & Moon":       "sm",
 }
 
 type SeriesModel struct {
@@ -64,6 +65,7 @@ func SeriesList() SeriesModel {
 		item("Mega Evolution"),
 		item("Scarlet & Violet"),
 		item("Sword & Shield"),
+		item("Sun & Moon"),
 	}
 
 	const listWidth = 20

cmd/card/serieslist_test.go

@@ -152,12 +152,12 @@ func TestSeriesList(t *testing.T) {
 		t.Error("SeriesList() should create a list with items")
 	}
 
-	if len(items) != 3 {
-		t.Errorf("Expected 3 items, got %d", len(items))
+	if len(items) != 4 {
+		t.Errorf("Expected 4 items, got %d", len(items))
 	}
 
-	// Verify all three series are present
-	expectedSeries := []string{"Mega Evolution", "Scarlet & Violet", "Sword & Shield"}
+	// Verify all four series are present
+	expectedSeries := []string{"Mega Evolution", "Scarlet & Violet", "Sword & Shield", "Sun & Moon"}
 	for i, expected := range expectedSeries {
 		itemStr := string(items[i].(item))
 		if itemStr != expected {