spells-mtg 0.3.0__py3-none-any.whl → 0.3.1__py3-none-any.whl

spells/cards.py

@@ -76,7 +76,10 @@ def _extract_value(set_code: str, name: str, card_dict: dict, field: CardAttr):
 
 def card_df(draft_set_code: str, names: list[str]) -> pl.DataFrame:
     draft_set_json = _fetch_mtg_json(draft_set_code)
-    set_codes = draft_set_json["data"]["booster"]["play"]["sourceSetCodes"]
+    booster_info = draft_set_json["data"]["booster"]
+
+    booster_type = "play" if "play" in booster_info else "draft"
+    set_codes = booster_info[booster_type]["sourceSetCodes"]
     set_codes.reverse()
 
     card_data_map = {}

spells/columns.py

@@ -510,6 +510,20 @@ _column_specs = [
         expr=pl.col(ColName.MANA_VALUE) * pl.col(ColName.DECK),
         dependencies=[ColName.MANA_VALUE, ColName.DECK],
     ),
+    ColumnSpec(
+        name=ColName.DECK_LANDS,
+        col_type=ColType.CARD_SUM,
+        expr=pl.when(pl.col(ColName.CARD_TYPE).str.contains("Land"))
+        .then(pl.col(ColName.DECK))
+        .otherwise(0),
+        dependencies=[ColName.CARD_TYPE, ColName.DECK],
+    ),
+    ColumnSpec(
+        name=ColName.DECK_SPELLS,
+        col_type=ColType.CARD_SUM,
+        expr=pl.col(ColName.DECK) - pl.col(ColName.DECK_SPELLS),
+        dependencies=[ColName.DECK, ColName.DECK_SPELLS],
+    ),
     ColumnSpec(
         name=ColName.MANA_COST,
         col_type=ColType.CARD_ATTR,
@@ -733,6 +747,18 @@ _column_specs = [
         expr=pl.col(ColName.DECK_MANA_VALUE) / pl.col(ColName.DECK),
         dependencies=[ColName.DECK_MANA_VALUE, ColName.DECK],
     ),
+    ColumnSpec(
+        name=ColName.DECK_LANDS_AVG,
+        col_type=ColType.AGG,
+        expr=pl.col(ColName.DECK_LANDS) / pl.col(ColName.NUM_GAMES),
+        dependencies=[ColName.DECK_LANDS, ColName.NUM_GAMES],
+    ),
+    ColumnSpec(
+        name=ColName.DECK_SPELLS_AVG,
+        col_type=ColType.AGG,
+        expr=pl.col(ColName.DECK_SPELLS) / pl.col(ColName.NUM_GAMES),
+        dependencies=[ColName.DECK_SPELLS, ColName.NUM_GAMES],
+    ),
 ]
 
 col_spec_map = {col.name: col for col in _column_specs}

spells/enums.py

@@ -117,6 +117,8 @@ class ColName(StrEnum):
     SUBTYPE = "subtype"
     MANA_VALUE = "mana_value"
     DECK_MANA_VALUE = "deck_mana_value"
+    DECK_LANDS = "deck_lands"
+    DECK_SPELLS = "deck_spells"
     MANA_COST = "mana_cost"
     POWER = "power"
     TOUGHNESS = "toughness"
@@ -157,3 +159,5 @@ class ColName(StrEnum):
     GIH_WR_STDEV = "gh_wr_stdev"
     GIH_WR_Z = "gih_wr_z"
     DECK_MANA_VALUE_AVG = "deck_mana_value_avg"
+    DECK_LANDS_AVG = "deck_lands_avg"
+    DECK_SPELLS_AVG = "deck_spells_avg"

{spells_mtg-0.3.0.dist-info → spells_mtg-0.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: spells-mtg
-Version: 0.3.0
+Version: 0.3.1
 Summary: analaysis of 17Lands.com public datasets
 Author-Email: Joel Barnes <oelarnes@gmail.com>
 License: MIT
@@ -424,6 +424,7 @@ A table of all included columns. Columns can be referenced by enum or by string
 | `CARD_TYPE` | `"card_type"` | `CARD` | `CARD_ATTR` | | String |
 | `SUBTYPE` | `"subtype"` | `CARD` | `CARD_ATTR` | | String |
 | `MANA_VALUE` | `"mana_value"` | `CARD` | `CARD_ATTR` | | Float |
+| `DECK_MANA_VALUE` | `"deck_mana_value"` | | `CARD_SUM` | `DECK` * `MANA_VALUE` | Float |
 | `MANA_COST` | `"mana_cost"` | `CARD` | `CARD_ATTR` | | String |
 | `POWER` | `"power"` | `CARD` | `CARD_ATTR` | | Float |
 | `TOUGHNESS` | `"toughness"` | `CARD` | `CARD_ATTR` | | Float |
@@ -462,6 +463,7 @@ A table of all included columns. Columns can be referenced by enum or by string
 | `GIH_WR_VAR` | `"gih_wr_var"` | | `AGG` | Game-weighted Variance | Float |
 | `GIH_WR_STDEV` | `"gh_wr_stdev"` | | `AGG` | Sqrt of `GIH_WR_VAR` | Float |
 | `GIH_WR_Z` | `"gih_wr_z"` | | `AGG` |`GIH_WR_EXCESS` / `GIH_WR_STDEV` | Float |
+| `DECK_MANA_VALUE_AVG` | `"deck_mana_value_avg"` | | `AGG` | `DECK_MANA_VALUE ` / `DECK` | Float |
 
 # Roadmap to 1.0
 

{spells_mtg-0.3.0.dist-info → spells_mtg-0.3.1.dist-info}/RECORD

@@ -1,15 +1,15 @@
 spells/__init__.py,sha256=QCPWQySUK2SZtCU-mSZLsn7vrNLJMDsRwil8gmAzmdk,151
 spells/cache.py,sha256=4v7h8D3TtaT0R_EdiRNhdcQrXzdH_CukezO6oAXvNEY,2956
-spells/cards.py,sha256=6seKpgI4TlJxI20bvcgt5VpxzsbCnnjWneDWpneJJ6Y,3497
-spells/columns.py,sha256=_x7vVSh-55TJRjV7LQ29Kj_Ek__ef4pvxUsX57s9Iwo,22488
+spells/cards.py,sha256=EOXAB_F2yedjf6KquCERCIHl0TSIJIoOe1jv8g4JzOc,3601
+spells/columns.py,sha256=QIM-dV00Pe5En9EozPWpH_u5f8VSgxd0Pxd2QxYuOFk,23423
 spells/draft_data.py,sha256=ABIQoUD8MKnc2goTUGY33dOxCpH8ydXXuQnbHMNULZs,10999
-spells/enums.py,sha256=t-1PrBUqbxZRShDOgmTHkhiHIOIMHTbD7RUgKiQdx-c,4621
+spells/enums.py,sha256=_yvb9AFPofRYwf0gndFteGQS9bSzHy2_HphYYhFr9Ro,4761
 spells/external.py,sha256=qe6wOBDhPN4CZNQvYRq6G-OpIZcWTZzJjySgnf2Gu1o,10258
 spells/filter.py,sha256=J-YTOOAzOQpvIX29tviYL04RVoOUlfsbjBXoQBDCEdQ,3380
 spells/manifest.py,sha256=oGkWuBYquJYmtaIp2O-x5nFKD5TkU5lzS-vera69jyU,7244
 spells/schema.py,sha256=z8Qn2SiHG4T6YfPsz8xHLGMjU_Ofm76-Vrquh3b9B64,6422
-spells_mtg-0.3.0.dist-info/METADATA,sha256=hhHnMORK-fcalrC-Y416Eyfx-ZBaldqZYaXKpMSh7a8,42882
-spells_mtg-0.3.0.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
-spells_mtg-0.3.0.dist-info/entry_points.txt,sha256=a9Y1omdl9MdnKuIj3aOodgrp-zZII6OCdvqwgP6BFvI,63
-spells_mtg-0.3.0.dist-info/licenses/LICENSE,sha256=tS54XYbJSgmq5zuHhbsQGbNQLJPVgXqhF5nu2CSRMig,1068
-spells_mtg-0.3.0.dist-info/RECORD,,
+spells_mtg-0.3.1.dist-info/METADATA,sha256=3K1GrMa-GXcb3J-0Xg71Vqi93LxoynWwLQ8UmcjhVxM,43073
+spells_mtg-0.3.1.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
+spells_mtg-0.3.1.dist-info/entry_points.txt,sha256=a9Y1omdl9MdnKuIj3aOodgrp-zZII6OCdvqwgP6BFvI,63
+spells_mtg-0.3.1.dist-info/licenses/LICENSE,sha256=tS54XYbJSgmq5zuHhbsQGbNQLJPVgXqhF5nu2CSRMig,1068
+spells_mtg-0.3.1.dist-info/RECORD,,