spells-mtg 0.11.5__py3-none-any.whl → 0.11.7__py3-none-any.whl

spells/card_data_files.py

@@ -18,12 +18,6 @@ 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),
-}
-
 ratings_col_defs = {
     ColName.NAME: pl.col("name").cast(pl.String),
     ColName.COLOR: pl.col("color").cast(pl.String),
@@ -40,7 +34,8 @@ ratings_col_defs = {
     ColName.WON_OPENING_HAND: pl.col("opening_hand_game_count")
     * pl.col("opening_hand_win_rate").cast(pl.Float64),
     ColName.DRAWN: pl.col("drawn_game_count").cast(pl.Int64),
-    ColName.WON_DRAWN: pl.col("drawn_win_rate") * pl.col("drawn_game_count").cast(pl.Float64),
+    ColName.WON_DRAWN: pl.col("drawn_win_rate")
+    * pl.col("drawn_game_count").cast(pl.Float64),
     ColName.NUM_GIH: pl.col("ever_drawn_game_count").cast(pl.Int64),
     ColName.NUM_GIH_WON: pl.col("ever_drawn_game_count")
     * pl.col("ever_drawn_win_rate").cast(pl.Float64),
@@ -58,16 +53,11 @@ 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,
@@ -106,9 +96,9 @@ def deck_color_df(
             [
                 pl.lit(set_code).alias(ColName.EXPANSION),
                 pl.lit(format).alias(ColName.EVENT_TYPE),
-                (pl.lit("Top") if player_cohort == "top" else pl.lit(None)).alias(
-                    ColName.PLAYER_COHORT
-                ).cast(pl.String),
+                (pl.lit("Top") if player_cohort == "top" else pl.lit(None))
+                .alias(ColName.PLAYER_COHORT)
+                .cast(pl.String),
                 *[val.alias(key) for key, val in deck_color_col_defs.items()],
             ]
         )
@@ -119,17 +109,12 @@ 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]
 
@@ -171,20 +156,24 @@ def base_ratings_df(
                 out=ratings_file_path,
             )
 
-        concat_list.append(pl.read_json(ratings_file_path).with_columns(
-            (pl.lit(deck_color) if deck_color != "any" else pl.lit(None)).alias(
-                ColName.MAIN_COLORS
-            ).cast(pl.String)
-        ).select(
-            [
-                pl.lit(set_code).alias(ColName.EXPANSION),
-                pl.lit(format).alias(ColName.EVENT_TYPE),
-                (pl.lit("Top") if player_cohort == "top" else pl.lit(None)).alias(
-                    ColName.PLAYER_COHORT
-                ).cast(pl.String),
-                ColName.MAIN_COLORS,
-                *[val.alias(key) for key, val in ratings_col_defs.items()],
-            ]
-       ))
+        concat_list.append(
+            pl.read_json(ratings_file_path, infer_schema_length=500)
+            .with_columns(
+                (pl.lit(deck_color) if deck_color != "any" else pl.lit(None))
+                .alias(ColName.MAIN_COLORS)
+                .cast(pl.String)
+            )
+            .select(
+                [
+                    pl.lit(set_code).alias(ColName.EXPANSION),
+                    pl.lit(format).alias(ColName.EVENT_TYPE),
+                    (pl.lit("Top") if player_cohort == "top" else pl.lit(None))
+                    .alias(ColName.PLAYER_COHORT)
+                    .cast(pl.String),
+                    ColName.MAIN_COLORS,
+                    *[val.alias(key) for key, val in ratings_col_defs.items()],
+                ]
+            )
+        )
 
     return pl.concat(concat_list)

spells/config.py

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

{spells_mtg-0.11.5.dist-info → spells_mtg-0.11.7.dist-info}/METADATA

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

{spells_mtg-0.11.5.dist-info → spells_mtg-0.11.7.dist-info}/RECORD

@@ -3,10 +3,10 @@ spells/.ruff_cache/0.8.6/17785301476771359756,sha256=gPYLG8psuOSo37IefLzTu5wgRrb
 spells/.ruff_cache/CACHEDIR.TAG,sha256=WVMVbX4MVkpCclExbq8m-IcOZIOuIZf5FrYw5Pk-Ma4,43
 spells/__init__.py,sha256=0pnh2NLn9FrNKncE9-tBsighp3e8YAsN--_ovUpgWGs,333
 spells/cache.py,sha256=6cl0q62erR3LCANPSfxG5-J7JQfLwNdWjzlBpfiL4IE,7174
-spells/card_data_files.py,sha256=mqPLJD-wBx_4rn-jyVoa2i-f8-uwyM6R7JACvBVu3Gg,6404
+spells/card_data_files.py,sha256=uEYVmyqIrMwVKTX2HAr8rEZi3WpW3uPzv_6TcZzPQDQ,6036
 spells/cards.py,sha256=6stFPhJOzHqvQnkSv9cDeylRa_7L9Y8sOaowiZhzz6I,4174
 spells/columns.py,sha256=s_PYyg2QaRL6kLWFNKCBEfbMF0x7O7-Yd9SeG1ttYL4,18206
-spells/config.py,sha256=zpRUZ-6JKALE149L0yfeD1AgqKPaVBGHfD_TdHi-jqE,235
+spells/config.py,sha256=CVE2O-ZPgthbSqAB7Pul2yeFcyLHpL52OiXH7s9yl48,246
 spells/draft_data.py,sha256=EC7TCAlrWWhDFQtGx6ie-SBTNvdsoPNef7oZ0w9TSW0,21012
 spells/enums.py,sha256=gbwfon6tQCoKDb-m4hSaHWi9slj82yqaH3qhYMVrsck,4991
 spells/extension.py,sha256=LBqGbJbe7iSRQkxJK7npkADCfzhdnIwwVvlmTn8xvjQ,8454
@@ -16,8 +16,8 @@ spells/log.py,sha256=3avmg65hru8K9npKLvPp1wWWxq-hoEYDUCbxqhPkKUw,2175
 spells/manifest.py,sha256=ExWVk17BRw615UmvrV817xwz457yfTNdNMNE_M00aEg,8338
 spells/schema.py,sha256=DbMvV8PIThJTp0Xzp_XIorlW6JhE1ud1kWRGf5SQ4_c,6406
 spells/utils.py,sha256=IO3brrXVvZla0LRTEB5v6NgGqZb_rYA46XtKBURGMNk,1944
-spells_mtg-0.11.5.dist-info/METADATA,sha256=A-aXPXsZKoJHsDzge8xRtvskSzBx_aepgiE7L6-Zqyw,47369
-spells_mtg-0.11.5.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
-spells_mtg-0.11.5.dist-info/entry_points.txt,sha256=a9Y1omdl9MdnKuIj3aOodgrp-zZII6OCdvqwgP6BFvI,63
-spells_mtg-0.11.5.dist-info/licenses/LICENSE,sha256=tS54XYbJSgmq5zuHhbsQGbNQLJPVgXqhF5nu2CSRMig,1068
-spells_mtg-0.11.5.dist-info/RECORD,,
+spells_mtg-0.11.7.dist-info/METADATA,sha256=odFdI4bYRUnk_aiFFnl0nEE5XNTOGwq-scbQ2dCFbjU,47369
+spells_mtg-0.11.7.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
+spells_mtg-0.11.7.dist-info/entry_points.txt,sha256=a9Y1omdl9MdnKuIj3aOodgrp-zZII6OCdvqwgP6BFvI,63
+spells_mtg-0.11.7.dist-info/licenses/LICENSE,sha256=tS54XYbJSgmq5zuHhbsQGbNQLJPVgXqhF5nu2CSRMig,1068
+spells_mtg-0.11.7.dist-info/RECORD,,