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

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

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

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

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

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

@@ -18,6 +18,13 @@ DECK_COLOR_DATA_TEMPLATE = (
     "{user_group_param}&start_date={start_date_str}&end_date={end_date_str}&combine_splash=true"
 )
 
+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 = {
     ColName.NAME: pl.col("name").cast(pl.String),
     ColName.COLOR: pl.col("color").cast(pl.String),
@@ -53,11 +60,16 @@ deck_color_col_defs = {
 
 def deck_color_df(
     set_code: str,
-    start_date: dt.date,
-    end_date: dt.date,
     format: str = "PremierDraft",
     player_cohort: str = "all",
+    start_date: dt.date | None = None,
+    end_date: dt.date | None = None,
 ):
+    if start_date is None:
+        start_date = START_DATE_MAP[set_code]
+    if end_date is None:
+        end_date = dt.date.today() - dt.timedelta(days=1)
+
     target_dir, filename = cache.deck_color_file_path(
         set_code,
         format,
@@ -109,12 +121,17 @@ def deck_color_df(
 
 def base_ratings_df(
     set_code: str,
-    start_date: dt.date,
-    end_date: dt.date,
     format: str = "PremierDraft",
     player_cohort: str = "all",
     deck_colors: str | list[str] = "any",
+    start_date: dt.date | None = None,
+    end_date: dt.date | None = None,
 ) -> pl.DataFrame:
+    if start_date is None:
+        start_date = START_DATE_MAP[set_code]
+    if end_date is None:
+        end_date = dt.date.today() - dt.timedelta(days=1)
+
     if isinstance(deck_colors, str):
         deck_colors = [deck_colors]
 

{spells_mtg-0.11.7 → 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