Add project files: decks, scripts, and collection data
This commit is contained in:
119
scripts/find_synergies.py
Normal file
119
scripts/find_synergies.py
Normal file
@@ -0,0 +1,119 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
Find cards with specific synergies from a hydrated collection.
|
||||
Usage:
|
||||
python find_synergies.py --collection collection_hydrated/deck.json --colors U R --keywords "landfall" "enter the battlefield"
|
||||
python find_synergies.py --collection collection_hydrated/deck.json --colors U R --creature-type Bird
|
||||
python find_synergies.py --collection collection_hydrated/deck.json --colors U R --cmc-min 4 --type instant
|
||||
"""
|
||||
|
||||
import argparse
|
||||
import json
|
||||
import re
|
||||
import sys
|
||||
from typing import Optional
|
||||
|
||||
|
||||
def load_collection(path: str) -> dict:
|
||||
"""Load hydrated collection JSON."""
|
||||
with open(path, 'r') as f:
|
||||
cards = json.load(f)
|
||||
return {c['name']: c for c in cards}
|
||||
|
||||
|
||||
def matches_colors(card_identity: set, allowed_colors: set) -> bool:
|
||||
"""Check if card identity fits within allowed colors."""
|
||||
if not card_identity:
|
||||
return True
|
||||
return card_identity <= allowed_colors
|
||||
|
||||
|
||||
def extract_creature_types(type_line: str) -> list:
|
||||
"""Extract creature types from type line."""
|
||||
if not type_line:
|
||||
return []
|
||||
parts = type_line.split('—')
|
||||
if len(parts) < 2:
|
||||
return []
|
||||
types_part = parts[1].strip()
|
||||
return [t.strip().lower() for t in types_part.split()]
|
||||
|
||||
|
||||
def main():
|
||||
parser = argparse.ArgumentParser(description='Find synergistic cards in collection')
|
||||
parser.add_argument('--collection', '-c', required=True, help='Hydrated collection JSON')
|
||||
parser.add_argument('--colors', nargs='+', default=[], help='Allowed colors (e.g., U R for Izzet)')
|
||||
parser.add_argument('--exclude', nargs='+', default=[], help='Cards to exclude (already in deck)')
|
||||
parser.add_argument('--keywords', '-k', nargs='+', help='Keywords to search in oracle text')
|
||||
parser.add_argument('--creature-type', '-t', help='Creature type to search for')
|
||||
parser.add_argument('--type', help='Card type to filter (creature, instant, sorcery, etc.)')
|
||||
parser.add_argument('--cmc-min', type=int, help='Minimum CMC')
|
||||
parser.add_argument('--cmc-max', type=int, help='Maximum CMC')
|
||||
parser.add_argument('--output', '-o', help='Output JSON file')
|
||||
parser.add_argument('--format', choices=['json', 'text'], default='text', help='Output format')
|
||||
args = parser.parse_args()
|
||||
|
||||
collection = load_collection(args.collection)
|
||||
allowed_colors = set(c.upper() for c in args.colors)
|
||||
exclude = set(args.exclude)
|
||||
|
||||
matches = []
|
||||
for name, card in collection.items():
|
||||
if name in exclude:
|
||||
continue
|
||||
|
||||
ci = set(card.get('color_identity', []))
|
||||
if not matches_colors(ci, allowed_colors):
|
||||
continue
|
||||
|
||||
type_line = card.get('type_line', '').lower() if card.get('type_line') else ''
|
||||
|
||||
if args.type and args.type.lower() not in type_line:
|
||||
continue
|
||||
|
||||
cmc = card.get('cmc', 0) or 0
|
||||
if args.cmc_min is not None and cmc < args.cmc_min:
|
||||
continue
|
||||
if args.cmc_max is not None and cmc > args.cmc_max:
|
||||
continue
|
||||
|
||||
if args.creature_type:
|
||||
creature_types = extract_creature_types(type_line)
|
||||
if args.creature_type.lower() not in creature_types:
|
||||
continue
|
||||
|
||||
oracle = card.get('oracle_text', '').lower() if card.get('oracle_text') else ''
|
||||
|
||||
if args.keywords:
|
||||
if not any(kw.lower() in oracle for kw in args.keywords):
|
||||
continue
|
||||
|
||||
matches.append({
|
||||
'name': name,
|
||||
'mana_cost': card.get('mana_cost', ''),
|
||||
'cmc': cmc,
|
||||
'type_line': card.get('type_line', ''),
|
||||
'colors': card.get('colors', []),
|
||||
'color_identity': list(ci),
|
||||
'oracle_text': card.get('oracle_text', ''),
|
||||
'count': card.get('count', 1)
|
||||
})
|
||||
|
||||
matches.sort(key=lambda x: (x['cmc'], x['name']))
|
||||
|
||||
if args.format == 'json':
|
||||
output = json.dumps(matches, indent=2)
|
||||
if args.output:
|
||||
with open(args.output, 'w') as f:
|
||||
f.write(output)
|
||||
else:
|
||||
print(output)
|
||||
else:
|
||||
for card in matches:
|
||||
print(f"{card['mana_cost']} {card['name']} ({int(card['cmc'])}cmc) x{card['count']}")
|
||||
if card['oracle_text']:
|
||||
print(f" {card['oracle_text'][:100]}...")
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
Reference in New Issue
Block a user