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

Removed need to create seperate land_cards file in setup

Browse source 
Adjusted tagger to keep the 'layout' and 'side' columns
This commit is contained in:
mwisnowski 2024-12-26 11:12:56 -08:00
parent ae83d0f66f
commit 1f8beeadef
2 changed files with 4 additions and 72 deletions
Download patch file Download diff file Expand all files Collapse all files

69
setup.py
View file

@ -51,64 +51,6 @@ def filter_by_color(df, column_name, value, new_csv_name):
filtered_df.to_csv(new_csv_name, index=False)
def set_lands():
print('Generating land_cards.csv.')
# Filter dataframe
while True:
try:
with open(f'{csv_directory}/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(f'{csv_directory}/cards.csv', 'wb') as outputfile:
outputfile.write(r.content)
# Load cards.csv file into pandas dataframe so it can be further broken down
df = pd.read_csv(f'{csv_directory}/cards.csv', low_memory=False)
legendary_options = ['Land']
df['colorIdentity'] = df['colorIdentity'].fillna('Colorless')
filtered_df = df[df['type'].str.contains('|'.join(legendary_options))]
"""
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.
"""
rows_to_drop = []
non_legel_sets = ['PHTR', 'PH17', 'PH18' ,'PH19', 'PH20', 'PH21', 'UGL', 'UND', 'UNH', 'UST',]
for index, row in filtered_df.iterrows():
for illegal_set in non_legel_sets:
if illegal_set in row['printings']:
rows_to_drop.append(index)
filtered_df = filtered_df.drop(rows_to_drop)
filtered_df.sort_values('name')
filtered_df = filtered_df.loc[filtered_df['layout'] != 'reversible_card']
filtered_df = filtered_df[filtered_df['availability'].str.contains('paper')]
filtered_df = filtered_df.loc[filtered_df['promoTypes'] != 'playtest']
filtered_df = filtered_df.loc[filtered_df['securityStamp'] != 'heart']
filtered_df = filtered_df.loc[filtered_df['securityStamp'] != 'acorn']
for card in banned_cards:
filtered_df = filtered_df[~filtered_df['name'].str.contains(card)]
card_types = ['Plane —', 'Conspiracy', 'Vanguard', 'Scheme', 'Phenomenon', 'Stickers', 'Attraction', 'Hero', 'Contraption']
for card_type in card_types:
filtered_df = filtered_df[~filtered_df['type'].str.contains(card_type)]
filtered_df['faceName'] = filtered_df['faceName'].fillna(filtered_df['name'])
filtered_df.drop_duplicates(subset='faceName', keep='first', inplace=True)
columns_to_keep = ['name', 'faceName','edhrecRank','colorIdentity', 'colors', 'manaCost', 'manaValue', 'type', 'layout', 'text', 'power', 'toughness', 'keywords']
filtered_df = filtered_df[columns_to_keep]
filtered_df.sort_values(by='edhrecRank', inplace=True)
filtered_df.to_csv(f'{csv_directory}/land_cards.csv', index=False)
print('land_cards.csv file generated.')
def determine_commanders():
print('Generating commander_cards.csv, containing all cards elligible to be commanders.')
@ -214,9 +156,6 @@ def initial_setup():
# Once by-color lists have been made, Determine legendary creatures
determine_commanders()
# Lastly, create a file with all lands, or cards that have a land on at least one face
set_lands()
# Once Legendary creatures are determined, generate staple lists
# generate_staple_lists()
@ -256,10 +195,7 @@ def regenerate_csvs_all():
# Once files are regenerated, create a new legendary list
determine_commanders()
# Lastly, create a file with all lands, or cards that have a land on at least one face
set_lands()
def regenerate_csv_by_color(color):
"""
Pull the original cards.csv file and remake the {color}_cards.csv files
@ -288,9 +224,6 @@ def regenerate_csv_by_color(color):
# Once files are regenerated, create a new legendary list
determine_commanders()
# Lastly, create a file with all lands, or cards that have a land on at least one face
set_lands()
def add_tags():
pass