spells-mtg 0.10.9__tar.gz → 0.10.11__tar.gz

{spells_mtg-0.10.9 → spells_mtg-0.10.11}/PKG-INFO

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

{spells_mtg-0.10.9 → spells_mtg-0.10.11}/pyproject.toml

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

{spells_mtg-0.10.9 → spells_mtg-0.10.11}/spells/cache.py

@@ -9,7 +9,9 @@ Caches are cleared per-set when new files are downloaded.
 
 from enum import StrEnum
 import os
+from pathlib import Path
 import sys
+
 import polars as pl
 
 
@@ -70,6 +72,25 @@ def data_home() -> str:
     )
 
 
+def ad_hoc_dir():
+    ad_hoc_dir = Path(data_home()) / "ad_hoc"
+    if not os.path.isdir(ad_hoc_dir):
+        os.makedirs(ad_hoc_dir)
+    return ad_hoc_dir
+
+
+def save_ad_hoc_dataset(df: pl.DataFrame, key: str):
+    df.write_parquet(ad_hoc_dir() / f"{key}.parquet")
+
+
+def read_ad_hoc_dataset(key: str):
+    path = ad_hoc_dir() / f"{key}.parquet"
+    if os.path.exists(path):
+        return pl.read_parquet(ad_hoc_dir() / f"{key}.parquet")
+    else:
+        return None
+
+
 def create_test_data(set_code: str, test_num_drafts: int = 100):
     """
     run from prod environment to write test data for `set_code` into

{spells_mtg-0.10.9 → spells_mtg-0.10.11}/spells/config.py

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

{spells_mtg-0.10.9 → spells_mtg-0.10.11}/spells/draft_data.py

@@ -19,7 +19,7 @@ from polars.exceptions import ColumnNotFoundError
 
 from spells import cache
 import spells.filter
-import spells.manifest
+from spells import manifest
 from spells.columns import ColDef, ColSpec, get_specs
 from spells.enums import View, ColName, ColType
 from spells.log import make_verbose
@@ -379,7 +379,7 @@ def _fetch_or_cache(
 
 def _base_agg_df(
     set_code: str,
-    m: spells.manifest.Manifest,
+    m: manifest.Manifest,
     use_streaming: bool = True,
 ) -> pl.DataFrame:
     join_dfs = []
@@ -516,7 +516,7 @@ def summon(
             this_set_context = None
 
         col_def_map = _hydrate_col_defs(code, specs, set_card_context, this_set_context)
-        m = spells.manifest.create(col_def_map, columns, group_by, filter_spec)
+        m = manifest.create(col_def_map, columns, group_by, filter_spec)
 
         calc_fn = functools.partial(_base_agg_df, code, m, use_streaming=use_streaming)
         agg_df = _fetch_or_cache(

{spells_mtg-0.10.9 → spells_mtg-0.10.11}/spells/manifest.py

@@ -1,9 +1,8 @@
 from dataclasses import dataclass
 
-import spells.columns
-import spells.filter
+import spells.filter as spells_filter
 from spells.enums import View, ColName, ColType
-from spells.columns import ColDef
+from spells.columns import ColDef, default_columns
 
 
 @dataclass(frozen=True)
@@ -13,7 +12,7 @@ class Manifest:
     base_view_group_by: frozenset[str]
     view_cols: dict[View, frozenset[str]]
     group_by: tuple[str, ...]
-    filter: spells.filter.Filter | None
+    filter: spells_filter.Filter | None
 
     def __post_init__(self):
         # No name filter check
@@ -166,7 +165,7 @@ def create(
     gbs = (ColName.NAME,) if group_by is None else tuple(group_by)
 
     if columns is None:
-        cols = tuple(spells.columns.default_columns)
+        cols = tuple(default_columns)
         if ColName.NAME not in gbs:
             cols = tuple(
                 col for col in cols if col not in (ColName.COLOR, ColName.RARITY)
@@ -174,7 +173,7 @@ def create(
     else:
         cols = tuple(columns)
 
-    m_filter = spells.filter.from_spec(filter_spec)
+    m_filter = spells_filter.from_spec(filter_spec)
 
     col_set = frozenset(cols)
     col_set = col_set.union(frozenset(gbs) - {ColName.NAME})