spells-mtg 0.7.4__py3-none-any.whl → 0.8.1__py3-none-any.whl

spells/__init__.py

@@ -1,5 +1,5 @@
 from spells.columns import ColSpec
 from spells.enums import ColType, ColName
-from spells.draft_data import summon
+from spells.draft_data import summon, view_select, get_names
 
-__all__ = ["summon", "ColSpec", "ColType", "ColName"]
+__all__ = ["summon", "view_select", "get_names", "ColSpec", "ColType", "ColName"]

spells/columns.py

@@ -25,6 +25,8 @@ class ColDef:
 
 
 default_columns = [
+    ColName.COLOR,
+    ColName.RARITY,
     ColName.NUM_SEEN,
     ColName.ALSA,
     ColName.NUM_TAKEN,

spells/draft_data.py

@@ -35,7 +35,7 @@ def _cache_key(args) -> str:
 
 
 @functools.lru_cache(maxsize=None)
-def _get_names(set_code: str) -> list[str]:
+def get_names(set_code: str) -> list[str]:
     card_fp = 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())
@@ -86,7 +86,7 @@ def _get_card_context(
 
         loaded_context = {row[ColName.NAME]: row for row in select_rows}
     else:
-        names = _get_names(set_code)
+        names = get_names(set_code)
         loaded_context = {name: {} for name in names}
 
     if card_context is not None:
@@ -265,7 +265,7 @@ def _hydrate_col_defs(
     set_context: pl.DataFrame | dict[str, Any] | None = None,
     card_only: bool = False,
 ):
-    names = _get_names(set_code)
+    names = get_names(set_code)
 
     set_context = _get_set_context(set_code, set_context)
 
@@ -538,7 +538,14 @@ def summon(
     ), "What happened? We mean to use one of the sets manifest, it shouldn't matter which."
 
     if m.group_by:
-        full_agg_df = full_agg_df.group_by(m.group_by).sum()
+        gb = m.group_by
+        # an agg may depend on some card column that hasn't been explicitly requested, but can be safely
+        # depended on if we are grouping by name
+        if ColName.NAME in m.group_by and View.CARD in m.view_cols:
+            for col in m.view_cols[View.CARD]:
+                if col not in m.group_by:
+                    gb = tuple([*gb, col])
+        full_agg_df = full_agg_df.group_by(gb).sum()
     else:
         full_agg_df = full_agg_df.sum()
 
@@ -550,3 +557,46 @@ def summon(
     )
 
     return ret_df
+
+
+def view_select(
+    set_code: str,
+    view: View,
+    columns: list[str],
+    filter_spec: dict | None = None,
+    extensions: dict[str, ColSpec] | list[dict[str, ColSpec]] | None = None,
+    card_context: dict | pl.DataFrame | None = None,
+    set_context: dict | pl.DataFrame | None = None,
+) -> pl.LazyFrame:
+    specs = get_specs()
+
+    if extensions is not None:
+        if not isinstance(extensions, list):
+            extensions = [extensions]
+        for ext in extensions:
+            specs.update(ext)
+
+    col_def_map = _hydrate_col_defs(set_code, specs, card_context, set_context)
+
+    df_path = data_file_path(set_code, view)
+    base_view_df = pl.scan_parquet(df_path)
+
+    select_cols = frozenset(columns)
+
+    filter_ = spells.filter.from_spec(filter_spec)
+    if filter_ is not None:
+        select_cols = select_cols.union(filter_.lhs)
+
+    base_df_prefilter = _view_select(
+        base_view_df,
+        select_cols,
+        col_def_map,
+        is_agg_view=False,
+    )
+
+    if filter_ is not None:
+        base_df = base_df_prefilter.filter(filter_.expr)
+    else:
+        base_df = base_df_prefilter
+
+    return base_df.select(columns)

spells/manifest.py

@@ -163,14 +163,14 @@ def create(
     group_by: list[str] | None = None,
     filter_spec: dict | None = None,
 ):
-    gbs = (
-        (ColName.NAME, ColName.COLOR, ColName.RARITY)
-        if group_by is None
-        else tuple(group_by)
-    )
+    gbs = (ColName.NAME,) if group_by is None else tuple(group_by)
 
     if columns is None:
         cols = tuple(spells.columns.default_columns)
+        if ColName.NAME not in gbs:
+            cols = tuple(
+                col for col in cols if col not in (ColName.COLOR, ColName.RARITY)
+            )
     else:
         cols = tuple(columns)
 

{spells_mtg-0.7.4.dist-info → spells_mtg-0.8.1.dist-info}/METADATA

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

spells_mtg-0.8.1.dist-info/RECORD

@@ -0,0 +1,16 @@
+spells/__init__.py,sha256=dWkRW298GOYHFiG84S3cUrp4R-vydgA-v95CFPaWu2M,221
+spells/cache.py,sha256=YpTmlW-U2EosvGCH9gRQdQc8gRjjcGbgQovn5gfA_WE,3165
+spells/cards.py,sha256=EOXAB_F2yedjf6KquCERCIHl0TSIJIoOe1jv8g4JzOc,3601
+spells/columns.py,sha256=SdNw51lA1ox5rD2M7F8M2yqo-m0Dal5sKzS4Rp4q9MQ,18092
+spells/draft_data.py,sha256=R44E8eb7bMOT0AxnkWcOAyuVCLDUxbvYmlYCLEtqi-s,18800
+spells/enums.py,sha256=DL7e1xDEvrsTMbA7vJB_Et1DaYkyO4rIEzvIQDz3MZk,4871
+spells/extension.py,sha256=RFLnNdT_4fIibgw_lAsAjQsFzWmNjJp8KRXJvagdz3Y,6580
+spells/external.py,sha256=PN0PkwdKnvx-4HkApFhI1_ZwRvBUDMUd3Vfbky0xSJA,11786
+spells/filter.py,sha256=J-YTOOAzOQpvIX29tviYL04RVoOUlfsbjBXoQBDCEdQ,3380
+spells/manifest.py,sha256=dOUmj2uZZ17vCWpFwv7B5F6wOIWnoQdZkEB9SDKdx9M,8310
+spells/schema.py,sha256=DbMvV8PIThJTp0Xzp_XIorlW6JhE1ud1kWRGf5SQ4_c,6406
+spells_mtg-0.8.1.dist-info/METADATA,sha256=fltb_cVkAsksPVHjJ77GT4C6QJ1MnjJ3CQyHBC5IXjQ,46731
+spells_mtg-0.8.1.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
+spells_mtg-0.8.1.dist-info/entry_points.txt,sha256=a9Y1omdl9MdnKuIj3aOodgrp-zZII6OCdvqwgP6BFvI,63
+spells_mtg-0.8.1.dist-info/licenses/LICENSE,sha256=tS54XYbJSgmq5zuHhbsQGbNQLJPVgXqhF5nu2CSRMig,1068
+spells_mtg-0.8.1.dist-info/RECORD,,

spells_mtg-0.7.4.dist-info/RECORD

@@ -1,16 +0,0 @@
-spells/__init__.py,sha256=iyX74sLFJIQ_5ShZ7KDOeBlbYLm21ibWxwyzCGnKdg8,169
-spells/cache.py,sha256=YpTmlW-U2EosvGCH9gRQdQc8gRjjcGbgQovn5gfA_WE,3165
-spells/cards.py,sha256=EOXAB_F2yedjf6KquCERCIHl0TSIJIoOe1jv8g4JzOc,3601
-spells/columns.py,sha256=kWeDofGL0bYcxq4OOfUwkU4Ty3fm0xux6R9774v9HVQ,18053
-spells/draft_data.py,sha256=YeixEMn9OZu9fAFH-VMC6urkbybySqlHITBhfZLFigc,17271
-spells/enums.py,sha256=DL7e1xDEvrsTMbA7vJB_Et1DaYkyO4rIEzvIQDz3MZk,4871
-spells/extension.py,sha256=RFLnNdT_4fIibgw_lAsAjQsFzWmNjJp8KRXJvagdz3Y,6580
-spells/external.py,sha256=PN0PkwdKnvx-4HkApFhI1_ZwRvBUDMUd3Vfbky0xSJA,11786
-spells/filter.py,sha256=J-YTOOAzOQpvIX29tviYL04RVoOUlfsbjBXoQBDCEdQ,3380
-spells/manifest.py,sha256=83VCuS5AuVjvOltdNscOOwekAfeIEHSg9a2O_aKAyA0,8214
-spells/schema.py,sha256=DbMvV8PIThJTp0Xzp_XIorlW6JhE1ud1kWRGf5SQ4_c,6406
-spells_mtg-0.7.4.dist-info/METADATA,sha256=imq85pXWkIfPCz3xg7g5oi3vpmdLlNriPP6dy-7D8Z4,46731
-spells_mtg-0.7.4.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
-spells_mtg-0.7.4.dist-info/entry_points.txt,sha256=a9Y1omdl9MdnKuIj3aOodgrp-zZII6OCdvqwgP6BFvI,63
-spells_mtg-0.7.4.dist-info/licenses/LICENSE,sha256=tS54XYbJSgmq5zuHhbsQGbNQLJPVgXqhF5nu2CSRMig,1068
-spells_mtg-0.7.4.dist-info/RECORD,,