Andreas/mtg_python_deckbuilder
1
0
Fork 0
mirror of https://github.com/mwisnowski/mtg_python_deckbuilder.git synced 2025-09-22 04:50:46 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Added card info logic, split things into multiple files

Browse source
This commit is contained in:
mwisnowski 2024-12-05 12:03:06 -08:00
parent 00ae707d16
commit 3efd1cd9a9
5 changed files with 718 additions and 568 deletions
Download patch file Download diff file Expand all files Collapse all files

BIN
__pycache__/card_info.cpython-313.pyc Normal file
View file

Binary file not shown.

BIN
__pycache__/setup.cpython-313.pyc Normal file
View file

Binary file not shown.

38
card_info.py Normal file
View file

@ -0,0 +1,38 @@
from __future__ import annotations
import inquirer.prompt # type: ignore
import pandas as pd # type: ignore
from fuzzywuzzy import fuzz, process # type: ignore
from IPython.display import display
from tabulate import tabulate
pd.set_option('display.max_colwidth', None)
pd.set_option('display.expand_frame_repr', True)
pd.options.mode.chained_assignment = None
def get_card_info():
question = [
inquirer.Text(
'card_prompt',
message='Enter a card name:'
)
]
answer = inquirer.prompt(question)
card_choice = answer['card_prompt']
df = pd.read_csv('csv_files/cards.csv', low_memory=False)
fuzzy_card_choice = process.extractOne(card_choice, df['name'], scorer=fuzz.ratio)
fuzzy_card_choice = fuzzy_card_choice[0]
filtered_df = df[df['name'] == fuzzy_card_choice]
columns_to_keep = ['name', 'colorIdentity', 'colors', 'manaCost', 'manaValue', 'type', 'keywords', 'power', 'toughness', 'text']
filtered_df = filtered_df[filtered_df['layout'].str.contains('reversible_card') == False]
filtered_df = filtered_df[filtered_df['availability'].str.contains('arena') == False]
filtered_df.drop_duplicates(subset='name', keep='first', inplace=True)
filtered_df = filtered_df[columns_to_keep].astype('string')
columns_to_keep = ['name', 'colorIdentity', 'colors', 'manaCost', 'manaValue', 'type', 'keywords', 'power', 'toughness']
filtered_df_no_text = filtered_df[columns_to_keep]
filtered_df_no_text.dropna(how='all', axis=1, inplace=True)
display(filtered_df_no_text.to_string())
display(filtered_df['text'].to_string())

625
main.py
View file

@ -5,579 +5,68 @@ import inquirer.prompt # type: ignore
import pandas as pd # type: ignore
import requests # type: ignore
#import scrython # type: ignore
import sys
banned_cards = ['Ancestral Recall', 'Balance', 'Biorhythm', 'Black Lotus',
'Braids, Cabal Minion', 'Chaos Orb', 'Coalition Victory',
'Channel', 'Dockside Extortionist', 'Emrakul, the Aeons Torn',
'Erayo, Soratami Ascendant', 'Falling Star', 'Fastbond',
'Flash', 'Gifts Ungiven', 'Golos, Tireless Pilgrim',
'Griselbrand', 'Hullbreacher', 'Iona, Shield of Emeria',
'Karakas', 'Jeweled Lotus', 'Leovold, Emissary of Trest',
'Library of Alexandria', 'Limited Resources', 'Lutri, the Spellchaser',
'Mana Crypt', 'Mox Emerald', 'Mox Jet', 'Mox Pearl', 'Mox Ruby',
'Mox Sapphire', 'Nadu, Winged Wisdom', 'Panoptic Mirror',
'Paradox Engine', 'Primeval Titan', 'Prophet of Kruphix',
'Recurring Nightmare', 'Rofellos, Llanowar Emissary', 'Shahrazad',
'Sundering Titan', 'Sway of the Stars', 'Sylvan Primordial',
'Time Vault', 'Time Walk', 'Tinker', 'Tolarian Academy',
'Trade Secrets', 'Upheaval', 'Yawgmoth\'s Bargain']
from pathlib import Path
def filter_by_color(df, column_name, value, new_csv_name):
# Filter dataframe
filtered_df = df[df[column_name] == value]
# Save the filtered dataframe to a new csv file
columns_to_keep = ['name', 'edhrecRank','colorIdentity', 'colors', 'manaCost', 'manaValue', 'type', 'keywords', 'text', 'power', 'toughness', 'printings']
filtered_df = filtered_df[columns_to_keep]
filtered_df.drop_duplicates(subset='name', inplace=True)
filtered_df.to_csv(new_csv_name, index=False)
import setup
import card_info
def initial_setup():
# Check if the overall cards.csv file exists
print('Checking if setup was previously finished.\n')
while True:
try:
with open('setup_done.txt', 'r'):
print('Setup is done.\n')
break
except FileNotFoundError:
print('Checking for cards.csv file.\n')
while True:
try:
with open('csv_files/cards.csv', 'r', encoding='utf-8'):
print('cards.csv exists.')
break
except FileNotFoundError:
print('cards.csv not found, downloading from mtgjson')
url = 'https://mtgjson.com/api/v5/csv/cards.csv'
r = requests.get(url)
with open('csv_files/cards.csv', 'wb') as outputfile:
outputfile.write(r.content)
df = pd.read_csv('csv_files/cards.csv')
df['colorIdentity'] = df['colorIdentity'].fillna('None')
Path('csv_files').mkdir(parents=True, exist_ok=True)
Path('staples').mkdir(parents=True, exist_ok=True)
print('Checking for color identity sorted files.\n')
print('Checking for colorless_cards.csv.\n')
while True:
try:
with open('csv_files/colorless_cards.csv', 'r', encoding='utf-8'):
print('colorless_cards.csv exists.\n')
break
except FileNotFoundError:
print('colorless_cards.csv not found, creating it.\n')
filter_by_color(df, 'colorIdentity', 'None', 'csv_files/colorless_cards.csv')
print('Checking for mono-color card lists.\n')
while True:
print('Checking for white_cards.csv.')
while True:
try:
with open('csv_files/white_cards.csv', 'r', encoding='utf-8'):
print('white_cards.csv exists.')
break
except FileNotFoundError:
print('white_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'W', 'csv_files/white_cards.csv')
print('Checking for blue_cards.csv.')
while True:
try:
with open('csv_files/blue_cards.csv', 'r', encoding='utf-8'):
print('blue_cards.csv exists.')
break
except FileNotFoundError:
print('blue_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'U', 'csv_files/blue_cards.csv')
print('Checking for black_cards.csv.')
while True:
try:
with open('csv_files/black_cards.csv', 'r', encoding='utf-8'):
print('black_cards.csv exists.')
break
except FileNotFoundError:
print('black_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B', 'csv_files/black_cards.csv')
print('Checking for red_cards.csv.')
while True:
try:
with open('csv_files/red_cards.csv', 'r', encoding='utf-8'):
print('red_cards.csv exists.')
break
except FileNotFoundError:
print('red_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'R', 'csv_files/red_cards.csv')
print('Checking for green_cards.csv.')
while True:
try:
with open('csv_files/green_cards.csv', 'r', encoding='utf-8'):
print('green_cards.csv exists.')
break
except FileNotFoundError:
print('green_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'G', 'csv_files/green_cards.csv')
break
print('Checking for color-pair lists.\n')
while True:
print('Checking for azorius_cards.csv.')
while True:
try:
with open('csv_files/azorius_cards.csv', 'r', encoding='utf-8'):
print('azorius_cards.csv exists.')
break
except FileNotFoundError:
print('azorius_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'U, W', 'csv_files/azorius_cards.csv')
print('Checking for orzhov_cards.csv.')
while True:
try:
with open('csv_files/orzhov_cards.csv', 'r', encoding='utf-8'):
print('orzhov_cards.csv exists.')
break
except FileNotFoundError:
print('orzhov_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B, W', 'csv_files/orzhov_cards.csv')
print('Checking for boros_cards.csv.')
while True:
try:
with open('csv_files/boros_cards.csv', 'r', encoding='utf-8'):
print('boros_cards.csv exists.')
break
except FileNotFoundError:
print('boros_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'R, W', 'csv_files/boros_cards.csv')
print('Checking for selesnya_cards.csv.')
while True:
try:
with open('csv_files/selesnya_cards.csv', 'r', encoding='utf-8'):
print('selesnya_cards.csv exists.')
break
except FileNotFoundError:
print('selesnya_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'G, W', 'csv_files/selesnya_cards.csv')
print('Checking for dimir_cards.csv.')
while True:
try:
with open('csv_files/dimir_cards.csv', 'r', encoding='utf-8'):
print('dimir_cards.csv exists.')
break
except FileNotFoundError:
print('dimir_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B, U', 'csv_files/dimir_cards.csv')
print('Checking for izzet_cards.csv.')
while True:
try:
with open('csv_files/izzet_cards.csv', 'r', encoding='utf-8'):
print('izzet_cards.csv exists.')
break
except FileNotFoundError:
print('izzet_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'R, U', 'csv_files/izzet_cards.csv')
print('Checking for simic_cards.csv.')
while True:
try:
with open('csv_files/simic_cards.csv', 'r', encoding='utf-8'):
print('simic_cards.csv exists.')
break
except FileNotFoundError:
print('simic_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'G, U', 'csv_files/simic_cards.csv')
print('Checking for rakdos_cards.csv.')
while True:
try:
with open('csv_files/rakdos_cards.csv', 'r', encoding='utf-8'):
print('rakdos_cards.csv exists.')
break
except FileNotFoundError:
print('rakdos_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B, R', 'csv_files/rakdos_cards.csv')
print('Checking for golgari_cards.csv.')
while True:
try:
with open('csv_files/golgari_cards.csv', 'r', encoding='utf-8'):
print('golgari_cards.csv exists.')
break
except FileNotFoundError:
print('golgari_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B, G', 'csv_files/golgari_cards.csv')
print('Checking for gruul_cards.csv.')
while True:
try:
with open('csv_files/gruul_cards.csv', 'r', encoding='utf-8'):
print('gruul_cards.csv exists.')
break
except FileNotFoundError:
print('gruul_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'G, R', 'csv_files/gruul_cards.csv')
break
print('Checking for three-color sets.\n')
while True:
print('Checking for bant_cards.csv.')
while True:
try:
with open('csv_files/bant_cards.csv', 'r', encoding='utf-8'):
print('bant_cards.csv exists.')
break
except FileNotFoundError:
print('bant_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'G, U, W', 'csv_files/bant_cards.csv')
print('Checking for esper_cards.csv.')
while True:
try:
with open('csv_files/esper_cards.csv', 'r', encoding='utf-8'):
print('esper_cards.csv exists.')
break
except FileNotFoundError:
print('esper_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B, U, W', 'csv_files/esper_cards.csv')
print('Checking for grixis_cards.csv.')
while True:
try:
with open('csv_files/grixis_cards.csv', 'r', encoding='utf-8'):
print('grixis_cards.csv exists.')
break
except FileNotFoundError:
print('grixis_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B, R, U', 'csv_files/grixis_cards.csv')
print('Checking for jund_cards.csv.')
while True:
try:
with open('csv_files/jund_cards.csv', 'r', encoding='utf-8'):
print('jund_cards.csv exists.')
break
except FileNotFoundError:
print('jund_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B, G, R', 'csv_files/jund_cards.csv')
print('Checking for naya_cards.csv.')
while True:
try:
with open('csv_files/naya_cards.csv', 'r', encoding='utf-8'):
print('naya_cards.csv exists.')
break
except FileNotFoundError:
print('naya_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'G, R, W', 'csv_files/naya_cards.csv')
print('Checking for abzan_cards.csv.')
while True:
try:
with open('csv_files/abzan_cards.csv', 'r', encoding='utf-8'):
print('abzan_cards.csv exists.')
break
except FileNotFoundError:
print('abzan_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B, G, W', 'csv_files/abzan_cards.csv')
print('Checking for jeskai_cards.csv.')
while True:
try:
with open('csv_files/jeskai_cards.csv', 'r', encoding='utf-8'):
print('jeskai_cards.csv exists.')
break
except FileNotFoundError:
print('jeskai_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'R, U, W', 'csv_files/jeskai_cards.csv')
print('Checking for sultai_cards.csv.')
while True:
try:
with open('csv_files/sultai_cards.csv', 'r', encoding='utf-8'):
print('sultai_cards.csv exists.')
break
except FileNotFoundError:
print('sultai_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B, U, G', 'csv_files/sultai_cards.csv')
print('Checking for temur_cards.csv.')
while True:
try:
with open('csv_files/temur_cards.csv', 'r', encoding='utf-8'):
print('temur_cards.csv exists.')
break
except FileNotFoundError:
print('temur_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'G, R, U', 'csv_files/temur_cards.csv')
break
print('Checking for four color sets.\n')
while True:
print('Checking for dune_cards.csv.')
while True:
try:
with open('csv_files/dune_cards.csv', 'r', encoding='utf-8'):
print('dune_cards.csv exists.')
break
except FileNotFoundError:
print('dune_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B, G, R, W', 'csv_files/dune_cards.csv')
print('Checking for glint_cards.csv.')
while True:
try:
with open('csv_files/glint_cards.csv', 'r', encoding='utf-8'):
print('glint_cards.csv exists.')
break
except FileNotFoundError:
print('glint_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B, G, R, U', 'csv_files/glint_cards.csv')
print('Checking for ink_cards.csv.')
while True:
try:
with open('csv_files/ink_cards.csv', 'r', encoding='utf-8'):
print('ink_cards.csv exists.')
break
except FileNotFoundError:
print('ink_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'G, R, U, W', 'csv_files/ink_cards.csv')
print('Checking for witch_cards.csv.')
while True:
try:
with open('csv_files/witch_cards.csv', 'r', encoding='utf-8'):
print('witch_cards.csv exists.')
break
except FileNotFoundError:
print('witch_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B, G, U, W', 'csv_files/witch_cards.csv')
print('Checking for yore_cards.csv.')
while True:
try:
with open('csv_files/yore_cards.csv', 'r', encoding='utf-8'):
print('yore_cards.csv exists.')
break
except FileNotFoundError:
print('yore_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B, R, U, W', 'csv_files/yore_cards.csv')
break
print('Checking for wubrg_cards.csv.\n')
while True:
try:
with open('csv_files/wubrg_cards.csv', 'r', encoding='utf-8'):
print('wubrg_cards.csv exists.')
break
except FileNotFoundError:
print('wubrg_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B, G, R, U, W', 'csv_files/wubrg_cards.csv')
print('Creating setup_done.txt file.\n')
with open('setup_done.txt', 'w') as f:
f.write('Setup is done.')
f.close()
staple_lists = ['Colorless', 'White', 'Blue', 'Black']
colorless_staples = [] # type: ignore
white_staples = [] # type: ignore
blue_staples = [] # type: ignore
black_staples = [] # type: ignore
def make_staples_list():
# Generate list of staples by color. This is subjective and can be modified as seen fit.
# At present it's based on top cards from edhrec, using the first 15 or so of each color
print('Making staple lists.\n')
print('Colorless staples:')
global colorless_staples
colorless_staples = [
'Sol Ring',
'Arcane Signet',
'Fellwar Stone',
'Skullclamp',
'Lightning Greaves',
'Swiftfoot Boots',
'Thought Vessel',
'Solemn Simulacrum',
'Mind Stone',
'Wayfarer\'s Bauble',
'Commander\'s Sphere'
]
colorless_staples.sort()
print(colorless_staples)
print('White staples:')
global white_staples
white_staples = [
'Swords to Plowshares',
'Path to Exile',
'Generous Gift',
'Smothering Tithe',
'Esper Sentinel',
'Teferi\'s Protection',
'Enlightened Tutor',
'Farewell',
'Austere Command',
'Ghostly Prison',
'Sun Titan',
'Flawless Maneuver',
'Akroma\'s Will',
'Annointed Procession',
'Wrath of God'
]
print(white_staples)
print('Blue staples:')
global blue_staples
blue_staples = [
'Counterspell',
'Cyclonic Rift',
'Rhystic Study',
'Negate',
'Brainstorm',
'An Offer You Can\'t Refuse',
'Arcane Denial',
'Mystic Remora',
'Fierce Guardianship',
'Swan Song',
'Mana Drain',
'Mystical Tutor',
'Windfall',
'Ponder',
'Frantic Search',
'Pongify',
'Propoganda',
'Preordain',
'Opt',
'Rapid Hybridization'
]
print(blue_staples)
print('Black staples:')
global black_staples
black_staples = [
'Demonic Tutor',
'Dark Ritual',
'Vampiric Tutor',
'Feed the Swarm',
'Toxic Deluge',
'Reanimate',
'Phyrexian Arena',
'Blood Artist',
'Victimize',
'Deadly Dispute',
'Deadly Rollick',
'Black Market Connections',
'Zulaport Cutthroat',
'Village Rites',
'Sign in Blood',
'Animate Dead',
'Bolas\'s Citadel',
'Gray Merchant of Asphodel',
'Syr Konrad, the Grim',
'Diabolic Intent'
]
print(black_staples)
return colorless_staples, white_staples, blue_staples, black_staples
def generate_staple_lists():
# Colorless staples
global colorless_staples
print('Colorless staples:')
while True:
try:
with open('staples/colorless.txt', 'r') as file:
colorless_staples = file.read().split('\n')
del colorless_staples[-1]
print('\n'.join(colorless_staples), '\n')
break
except FileNotFoundError:
print('Generating colorless staples list.')
df = pd.read_csv('csv_files/colorless_cards.csv')
df['edhrecRank'] = pd.to_numeric(df['edhrecRank'], downcast='integer', errors='coerce')
df = df.dropna(subset=['edhrecRank'])
df['edhrecRank'] = df['edhrecRank'].astype(int)
columns_to_keep = ['name', 'edhrecRank', 'type']
df = df[columns_to_keep]
i = 1
while len(colorless_staples) < 20:
for index, row in df.iterrows():
if row['edhrecRank'] == i:
if 'Land' not in row['type'] and row['name'] not in banned_cards:
colorless_staples.append(row['name'])
i += 1
#print(colorless_staples)
with open('staples/colorless.txt', 'w') as f:
for items in colorless_staples:
f.write('%s\n' %items)
# White staples
print('White staples:')
global white_staples
while True:
try:
with open('staples/white.txt', 'r') as file:
white_staples = file.read().split('\n')
del white_staples[-1]
print('\n'.join(white_staples), '\n')
break
except FileNotFoundError:
print('Generating white staples list.')
df = pd.read_csv('csv_files/white_cards.csv')
df['edhrecRank'] = pd.to_numeric(df['edhrecRank'], downcast='integer', errors='coerce')
df = df.dropna(subset=['edhrecRank'])
df['edhrecRank'] = df['edhrecRank'].astype(int)
columns_to_keep = ['name', 'edhrecRank', 'type']
df = df[columns_to_keep]
i = 1
while len(white_staples) < 20:
for index, row in df.iterrows():
if row['edhrecRank'] == i:
if row['name'] not in banned_cards:
white_staples.append(row['name'])
i += 1
#print(white_staples)
with open('staples/white.txt', 'w') as f:
for items in white_staples:
f.write('%s\n' %items)
def get_card_info():
question = [
inquirer.List(
'staple_list',
message='Choose a staple list to check through',
choices=staple_lists,
carousel=True
)
]
answer = inquirer.prompt(question)
staple_list_choice = answer['staple_list']
if staple_list_choice == 'Colorless':
staple = 'colorless'
while True:
choice = 'Menu'
while choice == 'Menu':
question = [
inquirer.List(
'card_list',
message='Choose a card from the list.',
choices=colorless_staples,
carousel=True
)
inquirer.List('menu',
message='What would you like to do?',
choices=['Initial Setup', 'Regenerate Card CSVs', 'Setup Staples', 'Build a Deck', 'Get Card Info', 'Quit'],
carousel=True)
]
answer = inquirer.prompt(question)
card_choice = answer['card_list']
elif staple_list_choice == 'White':
staple = 'white'
question = [
inquirer.List(
'card_list',
message='Choose a card from the list.',
choices=white_staples,
carousel=True
)
]
answer = inquirer.prompt(question)
card_choice = answer['card_list']
elif staple_list_choice == 'Blue':
staple = 'blue'
question = [
inquirer.List(
'card_list',
message='Choose a card from the list.',
choices=blue_staples,
carousel=True
)
]
answer = inquirer.prompt(question)
card_choice = answer['card_list']
elif staple_list_choice == 'Black':
staple = 'black'
question = [
inquirer.List(
'card_list',
message='Choose a card from the list.',
choices=black_staples,
carousel=True
)
]
answer = inquirer.prompt(question)
card_choice = answer['card_list']
df = pd.read_csv(f'csv_files/{staple}_cards.csv')
filtered_df = df[df['name'] == card_choice]
columns_to_keep = ['name', 'colorIdentity', 'colors', 'manaCost', 'manaValue', 'type', 'keywords', 'power', 'toughness', 'text']
filtered_df = filtered_df[columns_to_keep]
print(filtered_df)
choice = answer['menu']
initial_setup()
#make_staples_list()
generate_staple_lists()
#get_card_info()
# Run through initial setup
while choice == 'Initial Setup':
setup.initial_setup()
choice = 'Menu'
break
# Run through csv regeneration
while choice == 'Regenerate Card CSVs':
setup.regenerate_csvs()
choice = 'Menu'
break
# Setup staples
while choice == 'Setup Staples':
setup.generate_staple_lists()
choice = 'Menu'
break
# Make a new deck
while choice == 'Build a Deck':
print('Deck building not yet implemented')
choice = 'Menu'
break
# Get a cards info
while choice == 'Get Card Info':
card_info.get_card_info()
question = [
inquirer.Confirm('continue',
message='Would you like to look up another card?'
)
]
answer = inquirer.prompt(question)
new_card = answer['continue']
if new_card:
choice == 'Get Card Info'
else:
choice = 'Menu'
break
# Quit
while choice == 'Quit':
sys.exit()

623
setup.py Normal file
View file

@ -0,0 +1,623 @@
from __future__ import annotations
import pandas as pd # type: ignore
import requests # type: ignore
staple_lists = ['Colorless', 'White', 'Blue', 'Black']
colorless_staples = [] # type: ignore
white_staples = [] # type: ignore
blue_staples = [] # type: ignore
black_staples = [] # type: ignore
banned_cards = ['Ancestral Recall', 'Balance', 'Biorhythm', 'Black Lotus',
'Braids, Cabal Minion', 'Chaos Orb', 'Coalition Victory',
'Channel', 'Dockside Extortionist', 'Emrakul, the Aeons Torn',
'Erayo, Soratami Ascendant', 'Falling Star', 'Fastbond',
'Flash', 'Gifts Ungiven', 'Golos, Tireless Pilgrim',
'Griselbrand', 'Hullbreacher', 'Iona, Shield of Emeria',
'Karakas', 'Jeweled Lotus', 'Leovold, Emissary of Trest',
'Library of Alexandria', 'Limited Resources', 'Lutri, the Spellchaser',
'Mana Crypt', 'Mox Emerald', 'Mox Jet', 'Mox Pearl', 'Mox Ruby',
'Mox Sapphire', 'Nadu, Winged Wisdom', 'Panoptic Mirror',
'Paradox Engine', 'Primeval Titan', 'Prophet of Kruphix',
'Recurring Nightmare', 'Rofellos, Llanowar Emissary', 'Shahrazad',
'Sundering Titan', 'Sway of the Stars', 'Sylvan Primordial',
'Time Vault', 'Time Walk', 'Tinker', 'Tolarian Academy',
'Trade Secrets', 'Upheaval', 'Yawgmoth\'s Bargain']
def filter_by_color(df, column_name, value, new_csv_name):
# Filter dataframe
filtered_df = df[df[column_name] == value]
"""
Save the filtered dataframe to a new csv file, and narrow down/rearranges the columns it
keeps to increase readability/trim some extra data.
Additionally attempts to remove as many duplicates (including cards with reversible prints,
as well as taking out Arena-only cards.
"""
filtered_df.sort_values('name')
filtered_df = filtered_df[filtered_df['layout'].str.contains('reversible_card') == False]
filtered_df = filtered_df[filtered_df['availability'].str.contains('arena') == False]
filtered_df.drop_duplicates(subset='name', keep='first', inplace=True)
columns_to_keep = ['name', 'edhrecRank','colorIdentity', 'colors', 'manaCost', 'manaValue', 'type', 'keywords', 'text', 'power', 'toughness']
filtered_df = filtered_df[columns_to_keep]
filtered_df.sort_values(by='name', key=lambda col: col.str.lower(), inplace=True)
filtered_df.to_csv(new_csv_name, index=False)
def initial_setup():
# Check if the overall cards.csv file exists
print('Checking for cards.csv file.\n')
while True:
try:
with open('csv_files/cards.csv', 'r', encoding='utf-8'):
print('cards.csv exists.')
break
except FileNotFoundError:
# If the cards.csv file does not exist or can't be found, pull it from mtgjson.com
print('cards.csv not found, downloading from mtgjson')
url = 'https://mtgjson.com/api/v5/csv/cards.csv'
r = requests.get(url)
with open('csv_files/cards.csv', 'wb') as outputfile:
outputfile.write(r.content)
df = pd.read_csv('csv_files/cards.csv', low_memory=False)
df['colorIdentity'] = df['colorIdentity'].fillna('None')
# Checking for and creating individual color identity sorted csvs
print('Checking for color identity sorted files.\n')
# Colorless
print('Checking for colorless_cards.csv.\n')
while True:
try:
with open('csv_files/colorless_cards.csv', 'r', encoding='utf-8'):
print('colorless_cards.csv exists.\n')
break
except FileNotFoundError:
print('colorless_cards.csv not found, creating it.\n')
filter_by_color(df, 'colorIdentity', 'None', 'csv_files/colorless_cards.csv')
print('Checking for mono-color card lists.\n')
while True:
print('Checking for white_cards.csv.')
while True:
try:
with open('csv_files/white_cards.csv', 'r', encoding='utf-8'):
print('white_cards.csv exists.\n')
break
except FileNotFoundError:
print('white_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'W', 'csv_files/white_cards.csv')
print('Checking for blue_cards.csv.')
while True:
try:
with open('csv_files/blue_cards.csv', 'r', encoding='utf-8'):
print('blue_cards.csv exists.\n')
break
except FileNotFoundError:
print('blue_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'U', 'csv_files/blue_cards.csv')
print('Checking for black_cards.csv.')
while True:
try:
with open('csv_files/black_cards.csv', 'r', encoding='utf-8'):
print('black_cards.csv exists.\n')
break
except FileNotFoundError:
print('black_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B', 'csv_files/black_cards.csv')
print('Checking for red_cards.csv.')
while True:
try:
with open('csv_files/red_cards.csv', 'r', encoding='utf-8'):
print('red_cards.csv exists.\n')
break
except FileNotFoundError:
print('red_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'R', 'csv_files/red_cards.csv')
print('Checking for green_cards.csv.')
while True:
try:
with open('csv_files/green_cards.csv', 'r', encoding='utf-8'):
print('green_cards.csv exists.\n')
break
except FileNotFoundError:
print('green_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'G', 'csv_files/green_cards.csv')
break
print('Checking for color-pair lists.\n')
while True:
print('Checking for azorius_cards.csv.')
while True:
try:
with open('csv_files/azorius_cards.csv', 'r', encoding='utf-8'):
print('azorius_cards.csv exists.\n')
break
except FileNotFoundError:
print('azorius_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'U, W', 'csv_files/azorius_cards.csv')
print('Checking for orzhov_cards.csv.')
while True:
try:
with open('csv_files/orzhov_cards.csv', 'r', encoding='utf-8'):
print('orzhov_cards.csv exists.\n')
break
except FileNotFoundError:
print('orzhov_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B, W', 'csv_files/orzhov_cards.csv')
print('Checking for boros_cards.csv.')
while True:
try:
with open('csv_files/boros_cards.csv', 'r', encoding='utf-8'):
print('boros_cards.csv exists.\n')
break
except FileNotFoundError:
print('boros_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'R, W', 'csv_files/boros_cards.csv')
print('Checking for selesnya_cards.csv.')
while True:
try:
with open('csv_files/selesnya_cards.csv', 'r', encoding='utf-8'):
print('selesnya_cards.csv exists.\n')
break
except FileNotFoundError:
print('selesnya_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'G, W', 'csv_files/selesnya_cards.csv')
print('Checking for dimir_cards.csv.')
while True:
try:
with open('csv_files/dimir_cards.csv', 'r', encoding='utf-8'):
print('dimir_cards.csv exists.\n')
break
except FileNotFoundError:
print('dimir_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B, U', 'csv_files/dimir_cards.csv')
print('Checking for izzet_cards.csv.')
while True:
try:
with open('csv_files/izzet_cards.csv', 'r', encoding='utf-8'):
print('izzet_cards.csv exists.\n')
break
except FileNotFoundError:
print('izzet_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'R, U', 'csv_files/izzet_cards.csv')
print('Checking for simic_cards.csv.')
while True:
try:
with open('csv_files/simic_cards.csv', 'r', encoding='utf-8'):
print('simic_cards.csv exists.\n')
break
except FileNotFoundError:
print('simic_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'G, U', 'csv_files/simic_cards.csv')
print('Checking for rakdos_cards.csv.')
while True:
try:
with open('csv_files/rakdos_cards.csv', 'r', encoding='utf-8'):
print('rakdos_cards.csv exists.\n')
break
except FileNotFoundError:
print('rakdos_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B, R', 'csv_files/rakdos_cards.csv')
print('Checking for golgari_cards.csv.')
while True:
try:
with open('csv_files/golgari_cards.csv', 'r', encoding='utf-8'):
print('golgari_cards.csv exists.\n')
break
except FileNotFoundError:
print('golgari_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B, G', 'csv_files/golgari_cards.csv')
print('Checking for gruul_cards.csv.')
while True:
try:
with open('csv_files/gruul_cards.csv', 'r', encoding='utf-8'):
print('gruul_cards.csv exists.\n')
break
except FileNotFoundError:
print('gruul_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'G, R', 'csv_files/gruul_cards.csv')
break
print('Checking for three-color sets.\n')
while True:
print('Checking for bant_cards.csv.')
while True:
try:
with open('csv_files/bant_cards.csv', 'r', encoding='utf-8'):
print('bant_cards.csv exists.\n')
break
except FileNotFoundError:
print('bant_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'G, U, W', 'csv_files/bant_cards.csv')
print('Checking for esper_cards.csv.')
while True:
try:
with open('csv_files/esper_cards.csv', 'r', encoding='utf-8'):
print('esper_cards.csv exists.\n')
break
except FileNotFoundError:
print('esper_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B, U, W', 'csv_files/esper_cards.csv')
print('Checking for grixis_cards.csv.')
while True:
try:
with open('csv_files/grixis_cards.csv', 'r', encoding='utf-8'):
print('grixis_cards.csv exists.\n')
break
except FileNotFoundError:
print('grixis_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B, R, U', 'csv_files/grixis_cards.csv')
print('Checking for jund_cards.csv.')
while True:
try:
with open('csv_files/jund_cards.csv', 'r', encoding='utf-8'):
print('jund_cards.csv exists.\n')
break
except FileNotFoundError:
print('jund_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B, G, R', 'csv_files/jund_cards.csv')
print('Checking for naya_cards.csv.')
while True:
try:
with open('csv_files/naya_cards.csv', 'r', encoding='utf-8'):
print('naya_cards.csv exists.\n')
break
except FileNotFoundError:
print('naya_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'G, R, W', 'csv_files/naya_cards.csv')
print('Checking for abzan_cards.csv.')
while True:
try:
with open('csv_files/abzan_cards.csv', 'r', encoding='utf-8'):
print('abzan_cards.csv exists.\n')
break
except FileNotFoundError:
print('abzan_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B, G, W', 'csv_files/abzan_cards.csv')
print('Checking for jeskai_cards.csv.')
while True:
try:
with open('csv_files/jeskai_cards.csv', 'r', encoding='utf-8'):
print('jeskai_cards.csv exists.\n')
break
except FileNotFoundError:
print('jeskai_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'R, U, W', 'csv_files/jeskai_cards.csv')
print('Checking for sultai_cards.csv.')
while True:
try:
with open('csv_files/sultai_cards.csv', 'r', encoding='utf-8'):
print('sultai_cards.csv exists.\n')
break
except FileNotFoundError:
print('sultai_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B, G, U', 'csv_files/sultai_cards.csv')
print('Checking for temur_cards.csv.')
while True:
try:
with open('csv_files/temur_cards.csv', 'r', encoding='utf-8'):
print('temur_cards.csv exists.\n')
break
except FileNotFoundError:
print('temur_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'G, R, U', 'csv_files/temur_cards.csv')
break
print('Checking for four color sets.\n')
while True:
print('Checking for dune_cards.csv.')
while True:
try:
with open('csv_files/dune_cards.csv', 'r', encoding='utf-8'):
print('dune_cards.csv exists.\n')
break
except FileNotFoundError:
print('dune_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B, G, R, W', 'csv_files/dune_cards.csv')
print('Checking for glint_cards.csv.')
while True:
try:
with open('csv_files/glint_cards.csv', 'r', encoding='utf-8'):
print('glint_cards.csv exists.\n')
break
except FileNotFoundError:
print('glint_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B, G, R, U', 'csv_files/glint_cards.csv')
print('Checking for ink_cards.csv.')
while True:
try:
with open('csv_files/ink_cards.csv', 'r', encoding='utf-8'):
print('ink_cards.csv exists.\n')
break
except FileNotFoundError:
print('ink_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'G, R, U, W', 'csv_files/ink_cards.csv')
print('Checking for witch_cards.csv.')
while True:
try:
with open('csv_files/witch_cards.csv', 'r', encoding='utf-8'):
print('witch_cards.csv exists.\n')
break
except FileNotFoundError:
print('witch_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B, G, U, W', 'csv_files/witch_cards.csv')
print('Checking for yore_cards.csv.')
while True:
try:
with open('csv_files/yore_cards.csv', 'r', encoding='utf-8'):
print('yore_cards.csv exists.\n')
break
except FileNotFoundError:
print('yore_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B, R, U, W', 'csv_files/yore_cards.csv')
break
print('Checking for wubrg_cards.csv.\n')
while True:
try:
with open('csv_files/wubrg_cards.csv', 'r', encoding='utf-8'):
print('wubrg_cards.csv exists.\n')
break
except FileNotFoundError:
print('wubrg_cards.csv not found, creating it.')
filter_by_color(df, 'colorIdentity', 'B, G, R, U, W', 'csv_files/wubrg_cards.csv')
def regenerate_csvs():
"""
Pull the original cards.csv file and remake the {color}_cards.csv files.
This is useful if a new set has since come out to ensure the databases are up-to-date
"""
print('Downloading cards.csv from mtgjson')
url = 'https://mtgjson.com/api/v5/csv/cards.csv'
r = requests.get(url)
with open('csv_files/cards.csv', 'wb') as outputfile:
outputfile.write(r.content)
df = pd.read_csv('csv_files/cards.csv', low_memory=False)
df['colorIdentity'] = df['colorIdentity'].fillna('None')
# Color identity sorted cards
print('Regenerating color identity sorted files.\n')
print('Regenerating colorless_cards.csv.')
# Colorless
cards = 'colorless'
print(f'Regenerating {cards}_cards.csv.')
filter_by_color(df, 'colorIdentity', 'None', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
# Mono color
print('Regenerating mono-color card lists.\n')
while True:
# White cards
cards = 'white'
print(f'Regenerating {cards}_cards.csv.')
filter_by_color(df, 'colorIdentity', 'W', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
# Blue cards
cards = 'blue'
print('Regenerating blue_cards.csv.')
filter_by_color(df, 'colorIdentity', 'U', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
# Black cards
cards = 'black'
print('Regenerating black_cards.csv.')
filter_by_color(df, 'colorIdentity', 'B', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
# Red cards
cards = 'red'
print('Regenerating red_cards.csv.')
filter_by_color(df, 'colorIdentity', 'R', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
# Green cards
cards = 'green'
print('Regenerating green_cards.csv.')
filter_by_color(df, 'colorIdentity', 'G', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
break
# Color pairs
print('Regenerating color-pair lists.\n')
while True:
# Azorius cards
cards = 'azorius'
print(f'Regenerating {cards}_cards.csv.')
filter_by_color(df, 'colorIdentity', 'U, W', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
# Orzhov cards
cards = 'orzhov'
print(f'Regenerating {cards}_cards.csv.')
filter_by_color(df, 'colorIdentity', 'B, W', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
# Boros cards
cards = 'boros'
print(f'Regenerating {cards}_cards.csv.')
filter_by_color(df, 'colorIdentity', 'R, U', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
# Selesnya
cards = 'selesnya'
print(f'Regenerating {cards}_cards.csv.')
filter_by_color(df, 'colorIdentity', 'G, W', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
# Dimir
cards = 'dimir'
print(f'Regenerating {cards}_cards.csv.')
filter_by_color(df, 'colorIdentity', 'B, U', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
# Izzet
cards = 'izzet'
print(f'Regenerating {cards}_cards.csv.')
filter_by_color(df, 'colorIdentity', 'R, U', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
# Simic
cards = 'Simic'
print(f'Regenerating {cards}_cards.csv.')
filter_by_color(df, 'colorIdentity', 'G, U', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
# Rakdos
cards = 'rakdos'
print(f'Regenerating {cards}_cards.csv.')
filter_by_color(df, 'colorIdentity', 'B, R', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
# Golgari
cards = 'golgari'
print(f'Regenerating {cards}_cards.csv.')
filter_by_color(df, 'colorIdentity', 'B, G', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
# Gruul
cards = 'gruul'
print(f'Regenerating {cards}_cards.csv.')
filter_by_color(df, 'colorIdentity', 'G, R', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
break
# Color trios
print('Regenerating three-color sets.\n')
while True:
# Bant
cards = 'Bant'
print(f'Regenerating {cards}_cards.csv.')
filter_by_color(df, 'colorIdentity', 'G, U, W', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
# Esper
cards = 'esper'
print(f'Regenerating {cards}_cards.csv.')
filter_by_color(df, 'colorIdentity', 'B, U, W', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
# Grixis
cards = 'grixis'
print(f'Regenerating {cards}_cards.csv.')
filter_by_color(df, 'colorIdentity', 'B, R, U', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
# Jund
cards = 'jund'
print(f'Regenerating {cards}_cards.csv.')
filter_by_color(df, 'colorIdentity', 'B, G, R', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
# Naya
cards = 'naya'
print(f'Regenerating {cards}_cards.csv.')
filter_by_color(df, 'colorIdentity', 'G, R, W', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
# Abzan
cards = 'abzan'
print(f'Regenerating {cards}_cards.csv.')
filter_by_color(df, 'colorIdentity', 'B, G, W', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
# Jeskai
cards = 'jeskai'
print(f'Regenerating {cards}_cards.csv.')
filter_by_color(df, 'colorIdentity', 'R, U, W', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
# Mardu
cards = 'mardu'
print(f'Regenerating {cards}_cards.csv.')
filter_by_color(df, 'colorIdentity', 'B, R, W', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
# Sultai
cards = 'sultai'
print(f'Regenerating {cards}_cards.csv.')
filter_by_color(df, 'colorIdentity', 'B, G, U', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
# Temur
cards = 'temur'
print(f'Regenerating {cards}_cards.csv.')
filter_by_color(df, 'colorIdentity', 'G, R, U', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
break
# Four color
print('Regenerating four color sets.\n')
while True:
# Dune
cards = 'dune'
print(f'Regenerating {cards}_cards.csv.')
filter_by_color(df, 'colorIdentity', 'B, G, R, W', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
# Glint
cards = 'glint'
print(f'Regenerating {cards}_cards.csv.')
filter_by_color(df, 'colorIdentity', 'B, G, R, U', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
# Ink
cards = 'ink'
print(f'Regenerating {cards}_cards.csv.')
filter_by_color(df, 'colorIdentity', 'G, R, U, W', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
# Witch
cards = 'witch'
print(f'Regenerating {cards}_cards.csv.')
filter_by_color(df, 'colorIdentity', 'B, G, U, W', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
# Yore
cards = 'yore'
print(f'Regenerating {cards}_cards.csv.')
filter_by_color(df, 'colorIdentity', 'B, R, U, W', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
break
# WUBRG
cards = 'wubrg'
print(f'Regenerating {cards}_cards.csv.')
filter_by_color(df, 'colorIdentity', 'B, G, R, U, W', f'csv_files/{cards}_cards.csv')
print(f'A new {cards}_cards.csv file has been made\n')
def generate_staple_lists():
colors = ['colorless', 'white', 'blue', 'black', 'green', 'red',
'azorius', 'orzhov', 'selesnya', 'boros', 'dimir',
'simic', 'izzet', 'golgari', 'rakdos', 'gruul',
'bant', 'esper', 'grixis', 'jund', 'naya',
'abzan', 'jeskai', 'mardu', 'sultai', 'temur',
'dune', 'glint', 'ink', 'witch', 'yore', 'wubrg']
for color in colors:
staples = []
print(f'Checking for {color} staples file.')
try:
with open(f'staples/{color}.txt', 'r') as file:
staples = file.read().split('\n')
del staples[-1]
print(f'{color.capitalize()} staples:')
print('\n'.join(staples), '\n')
except FileNotFoundError:
print(f'{color.capitalize()} staples file not found.')
print(f'Generating {color} staples list.')
df = pd.read_csv(f'csv_files/{color}_cards.csv')
df['edhrecRank'] = pd.to_numeric(df['edhrecRank'], downcast='integer', errors='coerce')
df = df.dropna(subset=['edhrecRank'])
df['edhrecRank'] = df['edhrecRank'].astype(int)
columns_to_keep = ['name', 'edhrecRank', 'type']
df = df[columns_to_keep]
df.sort_values(by='edhrecRank', key=lambda col: col, inplace=True)
i = 1
while len(staples) < 20:
for index, row in df.iterrows():
if row['edhrecRank'] == i:
if 'Land' not in row['type'] and row['name'] not in banned_cards:
staples.append(row['name'])
i += 1
with open(f'staples/{color}.txt', 'w') as f:
for items in staples:
f.write('%s\n' %items)