2025-08-20 10:46:23 -07:00
|
|
|
|
from __future__ import annotations
|
|
|
|
|
|
|
|
|
|
|
|
from typing import Dict, List
|
|
|
|
|
|
import csv
|
|
|
|
|
|
import os
|
|
|
|
|
|
import datetime as _dt
|
|
|
|
|
|
import re as _re
|
|
|
|
|
|
import logging_util
|
|
|
|
|
|
|
|
|
|
|
|
logger = logging_util.logging.getLogger(__name__)
|
|
|
|
|
|
|
|
|
|
|
|
try:
|
|
|
|
|
|
from prettytable import PrettyTable # type: ignore
|
|
|
|
|
|
except Exception: # pragma: no cover
|
|
|
|
|
|
PrettyTable = None # type: ignore
|
|
|
|
|
|
|
|
|
|
|
|
class ReportingMixin:
|
|
|
|
|
|
"""Phase 6: Reporting, summaries, and export helpers."""
|
|
|
|
|
|
|
|
|
|
|
|
def _wrap_cell(self, text: str, width: int = 28) -> str:
|
|
|
|
|
|
words = text.split()
|
|
|
|
|
|
lines: List[str] = []
|
|
|
|
|
|
current_line = []
|
|
|
|
|
|
current_len = 0
|
|
|
|
|
|
for w in words:
|
|
|
|
|
|
if current_len + len(w) + (1 if current_line else 0) > width:
|
|
|
|
|
|
lines.append(' '.join(current_line))
|
|
|
|
|
|
current_line = [w]
|
|
|
|
|
|
current_len = len(w)
|
|
|
|
|
|
else:
|
|
|
|
|
|
current_line.append(w)
|
|
|
|
|
|
current_len += len(w) + (1 if len(current_line) > 1 else 0)
|
|
|
|
|
|
if current_line:
|
|
|
|
|
|
lines.append(' '.join(current_line))
|
|
|
|
|
|
return '\n'.join(lines)
|
|
|
|
|
|
|
|
|
|
|
|
def print_type_summary(self):
|
|
|
|
|
|
type_counts: Dict[str,int] = {}
|
|
|
|
|
|
for name, info in self.card_library.items():
|
|
|
|
|
|
ctype = info.get('Type', 'Unknown')
|
|
|
|
|
|
cnt = info.get('Count',1)
|
|
|
|
|
|
type_counts[ctype] = type_counts.get(ctype,0) + cnt
|
|
|
|
|
|
total_cards = sum(type_counts.values())
|
|
|
|
|
|
self.output_func("\nType Summary:")
|
|
|
|
|
|
for t, c in sorted(type_counts.items(), key=lambda kv: (-kv[1], kv[0])):
|
|
|
|
|
|
self.output_func(f" {t:<15} {c:>3} ({(c/total_cards*100 if total_cards else 0):5.1f}%)")
|
|
|
|
|
|
def export_decklist_csv(self, directory: str = 'deck_files', filename: str | None = None) -> str:
|
|
|
|
|
|
"""Export current decklist to CSV (enriched).
|
|
|
|
|
|
|
|
|
|
|
|
Filename pattern (default): commanderFirstWord_firstTheme_YYYYMMDD.csv
|
|
|
|
|
|
Included columns (enriched when possible):
|
|
|
|
|
|
Name, Count, Type, ManaCost, ManaValue, Colors, Power, Toughness, Role, Tags, Text
|
|
|
|
|
|
Falls back gracefully if snapshot rows missing.
|
|
|
|
|
|
"""
|
|
|
|
|
|
os.makedirs(directory, exist_ok=True)
|
|
|
|
|
|
if filename is None:
|
|
|
|
|
|
cmdr = getattr(self, 'commander_name', '') or getattr(self, 'commander', '') or ''
|
|
|
|
|
|
cmdr_first = cmdr.split()[0] if cmdr else 'deck'
|
|
|
|
|
|
theme = getattr(self, 'primary_tag', None) or (self.selected_tags[0] if getattr(self, 'selected_tags', []) else None)
|
|
|
|
|
|
theme_first = str(theme).split()[0] if theme else 'notheme'
|
|
|
|
|
|
def _slug(s: str) -> str:
|
|
|
|
|
|
s2 = _re.sub(r'[^A-Za-z0-9_]+', '', s)
|
|
|
|
|
|
return s2 or 'x'
|
|
|
|
|
|
cmdr_slug = _slug(cmdr_first)
|
|
|
|
|
|
theme_slug = _slug(theme_first)
|
|
|
|
|
|
date_part = _dt.date.today().strftime('%Y%m%d')
|
|
|
|
|
|
filename = f"{cmdr_slug}_{theme_slug}_{date_part}.csv"
|
|
|
|
|
|
fname = os.path.join(directory, filename)
|
|
|
|
|
|
|
|
|
|
|
|
full_df = getattr(self, '_full_cards_df', None)
|
|
|
|
|
|
combined_df = getattr(self, '_combined_cards_df', None)
|
|
|
|
|
|
snapshot = full_df if full_df is not None else combined_df
|
|
|
|
|
|
row_lookup: Dict[str, any] = {}
|
|
|
|
|
|
if snapshot is not None and not snapshot.empty and 'name' in snapshot.columns:
|
|
|
|
|
|
for _, r in snapshot.iterrows():
|
|
|
|
|
|
nm = str(r.get('name'))
|
|
|
|
|
|
if nm not in row_lookup:
|
|
|
|
|
|
row_lookup[nm] = r
|
|
|
|
|
|
|
|
|
|
|
|
headers = [
|
|
|
|
|
|
"Name","Count","Type","ManaCost","ManaValue","Colors","Power","Toughness",
|
|
|
|
|
|
"Role","SubRole","AddedBy","TriggerTag","Synergy","Tags","Text"
|
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
|
|
# Precedence list for sorting
|
|
|
|
|
|
precedence_order = [
|
|
|
|
|
|
'Commander', 'Battle', 'Planeswalker', 'Creature', 'Instant', 'Sorcery', 'Artifact', 'Enchantment', 'Land'
|
|
|
|
|
|
]
|
|
|
|
|
|
precedence_index = {k: i for i, k in enumerate(precedence_order)}
|
|
|
|
|
|
commander_name = getattr(self, 'commander_name', '') or getattr(self, 'commander', '') or ''
|
|
|
|
|
|
|
|
|
|
|
|
def classify(primary_type_line: str, card_name: str) -> str:
|
|
|
|
|
|
if commander_name and card_name == commander_name:
|
|
|
|
|
|
return 'Commander'
|
|
|
|
|
|
tl = (primary_type_line or '').lower()
|
|
|
|
|
|
if 'battle' in tl:
|
|
|
|
|
|
return 'Battle'
|
|
|
|
|
|
if 'planeswalker' in tl:
|
|
|
|
|
|
return 'Planeswalker'
|
|
|
|
|
|
if 'creature' in tl:
|
|
|
|
|
|
return 'Creature'
|
|
|
|
|
|
if 'instant' in tl:
|
|
|
|
|
|
return 'Instant'
|
|
|
|
|
|
if 'sorcery' in tl:
|
|
|
|
|
|
return 'Sorcery'
|
|
|
|
|
|
if 'artifact' in tl:
|
|
|
|
|
|
return 'Artifact'
|
|
|
|
|
|
if 'enchantment' in tl:
|
|
|
|
|
|
return 'Enchantment'
|
|
|
|
|
|
if 'land' in tl:
|
|
|
|
|
|
return 'Land'
|
|
|
|
|
|
return 'ZZZ'
|
|
|
|
|
|
|
|
|
|
|
|
rows: List[tuple] = [] # (sort_key, row_data)
|
|
|
|
|
|
|
|
|
|
|
|
for name, info in self.card_library.items():
|
|
|
|
|
|
base_type = info.get('Card Type') or info.get('Type','')
|
|
|
|
|
|
base_mc = info.get('Mana Cost','')
|
|
|
|
|
|
base_mv = info.get('Mana Value', info.get('CMC',''))
|
|
|
|
|
|
role = info.get('Role','') or ''
|
|
|
|
|
|
tags = info.get('Tags',[]) or []
|
|
|
|
|
|
tags_join = '; '.join(tags)
|
|
|
|
|
|
text_field = ''
|
|
|
|
|
|
colors = ''
|
|
|
|
|
|
power = ''
|
|
|
|
|
|
toughness = ''
|
|
|
|
|
|
row = row_lookup.get(name)
|
|
|
|
|
|
if row is not None:
|
|
|
|
|
|
row_type = row.get('type', row.get('type_line', ''))
|
|
|
|
|
|
if row_type:
|
|
|
|
|
|
base_type = row_type
|
|
|
|
|
|
mc = row.get('manaCost', '')
|
|
|
|
|
|
if mc:
|
|
|
|
|
|
base_mc = mc
|
|
|
|
|
|
mv = row.get('manaValue', row.get('cmc', ''))
|
|
|
|
|
|
if mv not in (None, ''):
|
|
|
|
|
|
base_mv = mv
|
|
|
|
|
|
colors_raw = row.get('colorIdentity', row.get('colors', []))
|
|
|
|
|
|
if isinstance(colors_raw, list):
|
|
|
|
|
|
colors = ''.join(colors_raw)
|
|
|
|
|
|
elif colors_raw not in (None, ''):
|
|
|
|
|
|
colors = str(colors_raw)
|
|
|
|
|
|
power = row.get('power', '') or ''
|
|
|
|
|
|
toughness = row.get('toughness', '') or ''
|
|
|
|
|
|
text_field = row.get('text', row.get('oracleText', '')) or ''
|
|
|
|
|
|
# Normalize and coerce text
|
|
|
|
|
|
if isinstance(text_field, str):
|
|
|
|
|
|
cleaned = text_field
|
|
|
|
|
|
else:
|
|
|
|
|
|
try:
|
|
|
|
|
|
import math as _math
|
|
|
|
|
|
if isinstance(text_field, float) and (_math.isnan(text_field)):
|
|
|
|
|
|
cleaned = ''
|
|
|
|
|
|
else:
|
|
|
|
|
|
cleaned = str(text_field) if text_field is not None else ''
|
|
|
|
|
|
except Exception:
|
|
|
|
|
|
cleaned = str(text_field) if text_field is not None else ''
|
|
|
|
|
|
cleaned = cleaned.replace('\n', ' ').replace('\r', ' ')
|
|
|
|
|
|
while ' ' in cleaned:
|
|
|
|
|
|
cleaned = cleaned.replace(' ', ' ')
|
|
|
|
|
|
text_field = cleaned
|
|
|
|
|
|
|
|
|
|
|
|
cat = classify(base_type, name)
|
|
|
|
|
|
prec = precedence_index.get(cat, 999)
|
|
|
|
|
|
# Alphabetical within category (no mana value sorting)
|
|
|
|
|
|
rows.append(((prec, name.lower()), [
|
|
|
|
|
|
name,
|
|
|
|
|
|
info.get('Count',1),
|
|
|
|
|
|
base_type,
|
|
|
|
|
|
base_mc,
|
|
|
|
|
|
base_mv,
|
|
|
|
|
|
colors,
|
|
|
|
|
|
power,
|
|
|
|
|
|
toughness,
|
|
|
|
|
|
info.get('Role') or role,
|
|
|
|
|
|
info.get('SubRole') or '',
|
|
|
|
|
|
info.get('AddedBy') or '',
|
|
|
|
|
|
info.get('TriggerTag') or '',
|
|
|
|
|
|
info.get('Synergy') if info.get('Synergy') is not None else '',
|
|
|
|
|
|
tags_join,
|
|
|
|
|
|
text_field[:800] if isinstance(text_field, str) else str(text_field)[:800]
|
|
|
|
|
|
]))
|
|
|
|
|
|
# Now sort (category precedence, then alphabetical name)
|
|
|
|
|
|
rows.sort(key=lambda x: x[0])
|
|
|
|
|
|
|
|
|
|
|
|
with open(fname, 'w', newline='', encoding='utf-8') as f:
|
|
|
|
|
|
w = csv.writer(f)
|
|
|
|
|
|
w.writerow(headers)
|
|
|
|
|
|
for _, data_row in rows:
|
|
|
|
|
|
w.writerow(data_row)
|
|
|
|
|
|
|
|
|
|
|
|
self.output_func(f"Deck exported to {fname}")
|
2025-08-20 11:17:51 -07:00
|
|
|
|
# Auto-generate matching plaintext list (best-effort; ignore failures)
|
|
|
|
|
|
try: # pragma: no cover - sidecar convenience
|
|
|
|
|
|
stem = os.path.splitext(os.path.basename(fname))[0]
|
|
|
|
|
|
# Always overwrite sidecar to reflect latest deck state
|
|
|
|
|
|
self.export_decklist_text(directory=directory, filename=stem + '.txt') # type: ignore[attr-defined]
|
|
|
|
|
|
except Exception:
|
|
|
|
|
|
logger.warning("Plaintext sidecar export failed (non-fatal)")
|
2025-08-20 10:46:23 -07:00
|
|
|
|
return fname
|
|
|
|
|
|
|
2025-08-20 11:17:51 -07:00
|
|
|
|
def export_decklist_text(self, directory: str = 'deck_files', filename: str | None = None) -> str:
|
|
|
|
|
|
"""Export a simple plaintext list: one line per unique card -> "[Count] [Card Name]".
|
|
|
|
|
|
|
|
|
|
|
|
Naming mirrors CSV export (same stem, .txt extension). Sorting follows same
|
|
|
|
|
|
category precedence then alphabetical within category for consistency.
|
|
|
|
|
|
"""
|
|
|
|
|
|
os.makedirs(directory, exist_ok=True)
|
|
|
|
|
|
# Derive base filename logic (shared with CSV exporter) – intentionally duplicated to avoid refactor risk.
|
|
|
|
|
|
if filename is None:
|
|
|
|
|
|
cmdr = getattr(self, 'commander_name', '') or getattr(self, 'commander', '') or ''
|
|
|
|
|
|
cmdr_first = cmdr.split()[0] if cmdr else 'deck'
|
|
|
|
|
|
theme = getattr(self, 'primary_tag', None) or (self.selected_tags[0] if getattr(self, 'selected_tags', []) else None)
|
|
|
|
|
|
theme_first = str(theme).split()[0] if theme else 'notheme'
|
|
|
|
|
|
def _slug(s: str) -> str:
|
|
|
|
|
|
s2 = _re.sub(r'[^A-Za-z0-9_]+', '', s)
|
|
|
|
|
|
return s2 or 'x'
|
|
|
|
|
|
cmdr_slug = _slug(cmdr_first)
|
|
|
|
|
|
theme_slug = _slug(theme_first)
|
|
|
|
|
|
date_part = _dt.date.today().strftime('%Y%m%d')
|
|
|
|
|
|
filename = f"{cmdr_slug}_{theme_slug}_{date_part}.txt"
|
|
|
|
|
|
if not filename.lower().endswith('.txt'):
|
|
|
|
|
|
filename = filename + '.txt'
|
|
|
|
|
|
path = os.path.join(directory, filename)
|
|
|
|
|
|
|
|
|
|
|
|
# Sorting reproduction
|
|
|
|
|
|
precedence_order = [
|
|
|
|
|
|
'Commander', 'Battle', 'Planeswalker', 'Creature', 'Instant', 'Sorcery', 'Artifact', 'Enchantment', 'Land'
|
|
|
|
|
|
]
|
|
|
|
|
|
precedence_index = {k: i for i, k in enumerate(precedence_order)}
|
|
|
|
|
|
commander_name = getattr(self, 'commander_name', '') or getattr(self, 'commander', '') or ''
|
|
|
|
|
|
def classify(primary_type_line: str, card_name: str) -> str:
|
|
|
|
|
|
if commander_name and card_name == commander_name:
|
|
|
|
|
|
return 'Commander'
|
|
|
|
|
|
tl = (primary_type_line or '').lower()
|
|
|
|
|
|
if 'battle' in tl:
|
|
|
|
|
|
return 'Battle'
|
|
|
|
|
|
if 'planeswalker' in tl:
|
|
|
|
|
|
return 'Planeswalker'
|
|
|
|
|
|
if 'creature' in tl:
|
|
|
|
|
|
return 'Creature'
|
|
|
|
|
|
if 'instant' in tl:
|
|
|
|
|
|
return 'Instant'
|
|
|
|
|
|
if 'sorcery' in tl:
|
|
|
|
|
|
return 'Sorcery'
|
|
|
|
|
|
if 'artifact' in tl:
|
|
|
|
|
|
return 'Artifact'
|
|
|
|
|
|
if 'enchantment' in tl:
|
|
|
|
|
|
return 'Enchantment'
|
|
|
|
|
|
if 'land' in tl:
|
|
|
|
|
|
return 'Land'
|
|
|
|
|
|
return 'ZZZ'
|
|
|
|
|
|
|
|
|
|
|
|
# We may want enriched type lines from snapshot; build quick lookup
|
|
|
|
|
|
full_df = getattr(self, '_full_cards_df', None)
|
|
|
|
|
|
combined_df = getattr(self, '_combined_cards_df', None)
|
|
|
|
|
|
snapshot = full_df if full_df is not None else combined_df
|
|
|
|
|
|
row_lookup: Dict[str, any] = {}
|
|
|
|
|
|
if snapshot is not None and not snapshot.empty and 'name' in snapshot.columns:
|
|
|
|
|
|
for _, r in snapshot.iterrows():
|
|
|
|
|
|
nm = str(r.get('name'))
|
|
|
|
|
|
if nm not in row_lookup:
|
|
|
|
|
|
row_lookup[nm] = r
|
|
|
|
|
|
|
|
|
|
|
|
sortable: List[tuple] = []
|
|
|
|
|
|
for name, info in self.card_library.items():
|
|
|
|
|
|
base_type = info.get('Card Type') or info.get('Type','')
|
|
|
|
|
|
row = row_lookup.get(name)
|
|
|
|
|
|
if row is not None:
|
|
|
|
|
|
row_type = row.get('type', row.get('type_line', ''))
|
|
|
|
|
|
if row_type:
|
|
|
|
|
|
base_type = row_type
|
|
|
|
|
|
cat = classify(base_type, name)
|
|
|
|
|
|
prec = precedence_index.get(cat, 999)
|
|
|
|
|
|
sortable.append(((prec, name.lower()), name, info.get('Count',1)))
|
|
|
|
|
|
sortable.sort(key=lambda x: x[0])
|
|
|
|
|
|
|
|
|
|
|
|
with open(path, 'w', encoding='utf-8') as f:
|
|
|
|
|
|
for _, name, count in sortable:
|
|
|
|
|
|
f.write(f"{count} {name}\n")
|
|
|
|
|
|
self.output_func(f"Plaintext deck list exported to {path}")
|
|
|
|
|
|
return path
|
|
|
|
|
|
|
2025-08-20 10:46:23 -07:00
|
|
|
|
def print_card_library(self, table: bool = True): # noqa: C901
|
|
|
|
|
|
if table and PrettyTable is None:
|
|
|
|
|
|
table = False
|
|
|
|
|
|
if not table:
|
|
|
|
|
|
self.output_func("\nCard Library:")
|
|
|
|
|
|
for name, info in sorted(self.card_library.items()):
|
|
|
|
|
|
self.output_func(f" {info.get('Count',1)}x {name} [{info.get('Type','')}] ({info.get('Role','')})")
|
|
|
|
|
|
return
|
|
|
|
|
|
# PrettyTable mode
|
|
|
|
|
|
pt = PrettyTable()
|
|
|
|
|
|
pt.field_names = ["Name","Count","Type","CMC","Role","Tags","Notes"]
|
|
|
|
|
|
pt.align = 'l'
|
|
|
|
|
|
for name, info in sorted(self.card_library.items()):
|
|
|
|
|
|
pt.add_row([
|
|
|
|
|
|
self._wrap_cell(name),
|
|
|
|
|
|
info.get('Count',1),
|
|
|
|
|
|
info.get('Type',''),
|
|
|
|
|
|
info.get('CMC',''),
|
|
|
|
|
|
self._wrap_cell(info.get('Role','')),
|
|
|
|
|
|
self._wrap_cell(','.join(info.get('Tags',[]) or [])),
|
|
|
|
|
|
self._wrap_cell(info.get('SourceNotes',''))
|
|
|
|
|
|
])
|
|
|
|
|
|
self.output_func("\nCard Library (tabular):")
|
|
|
|
|
|
self.output_func(pt.get_string())
|
