spells-mtg 0.11.6__tar.gz → 0.11.8__tar.gz

{spells_mtg-0.11.6 → spells_mtg-0.11.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: spells-mtg
-Version: 0.11.6
+Version: 0.11.8
 Summary: analaysis of 17Lands.com public datasets
 Author-Email: Joel Barnes <oelarnes@gmail.com>
 License: MIT

{spells_mtg-0.11.6 → spells_mtg-0.11.8}/pyproject.toml

@@ -11,7 +11,7 @@ dependencies = [
 ]
 requires-python = ">=3.11"
 readme = "README.md"
-version = "0.11.6"
+version = "0.11.8"
 
 [project.license]
 text = "MIT"

{spells_mtg-0.11.6 → spells_mtg-0.11.8}/spells/card_data_files.py

@@ -22,6 +22,7 @@ START_DATE_MAP = {
     "DFT": dt.date(2025, 2, 11),
     "TDM": dt.date(2025, 4, 8),
     "FIN": dt.date(2025, 6, 10),
+    "EOE": dt.date(2025, 7, 29),
 }
 
 ratings_col_defs = {

{spells_mtg-0.11.6 → spells_mtg-0.11.8}/spells/config.py

@@ -1,4 +1,5 @@
 all_sets = [
+    "FIN",
     "TDM",
     "DFT",
     "PIO",

{spells_mtg-0.11.6 → spells_mtg-0.11.8}/spells/draft_data.py

@@ -49,20 +49,25 @@ def _cache_key(args) -> str:
 @functools.lru_cache(maxsize=None)
 def get_names(set_code: str) -> list[str]:
     card_fp = cache.data_file_path(set_code, View.CARD)
-    card_view = pl.read_parquet(card_fp)
-    card_names_set = frozenset(card_view.get_column("name").to_list())
+    try:
+        card_view = pl.read_parquet(card_fp)
+        card_names_set = frozenset(card_view.get_column("name").to_list())
 
-    draft_fp = cache.data_file_path(set_code, View.DRAFT)
-    draft_view = pl.scan_parquet(draft_fp)
-    cols = draft_view.collect_schema().names()
+        draft_fp = cache.data_file_path(set_code, View.DRAFT)
+        draft_view = pl.scan_parquet(draft_fp)
+        cols = draft_view.collect_schema().names()
 
-    prefix = "pack_card_"
-    names = [col[len(prefix) :] for col in cols if col.startswith(prefix)]
-    draft_names_set = frozenset(names)
+        prefix = "pack_card_"
+        names = [col[len(prefix) :] for col in cols if col.startswith(prefix)]
+        draft_names_set = frozenset(names)
+
+        assert (
+            draft_names_set == card_names_set
+        ), "names mismatch between card and draft file"
+    except FileNotFoundError:
+        ratings_data = base_ratings_df(set_code)
+        names = list(ratings_data['name'])
 
-    assert (
-        draft_names_set == card_names_set
-    ), "names mismatch between card and draft file"
     return names