21a52cc5c2
Fixed issues with some sets, added more set codes for draftable sets, added CSV output file.
105 lines
3.6 KiB
Python
105 lines
3.6 KiB
Python
import json
|
|
import statistics
|
|
|
|
# Counts words on each card from each set using MTGJSON data.
|
|
# I downloaded "all set files" from https://mtgjson.com/downloads/all-files/
|
|
|
|
def load_set(code):
|
|
with open(f'mtgjson/sets/{code}.json', encoding='utf8') as f:
|
|
data = json.load(f)
|
|
return data
|
|
baseSetSize = data['baseSetSize']
|
|
cards = data['cards']
|
|
|
|
|
|
def analyze_set(code):
|
|
d = load_set(code)
|
|
if 'data' in d:
|
|
d = d['data']
|
|
if 'cards' in d:
|
|
cards = d['cards']
|
|
else:
|
|
print(f'No card data found in {code}.json')
|
|
return
|
|
faces = {}
|
|
#list text for all cards
|
|
for card in cards:
|
|
if 'text' in card:
|
|
faces[card['uuid']] = card['text']
|
|
else:
|
|
faces[card['uuid']] = ""
|
|
#update text for double faced
|
|
for card in cards:
|
|
if 'side' in card:
|
|
if card['side'] == 'a':
|
|
if 'text' not in card:
|
|
card['text'] = ""
|
|
elif card['otherFaceIds']:
|
|
card['text'] += " " + faces[card['otherFaceIds'][0]]
|
|
|
|
# make card number an int
|
|
if 'number' in card:
|
|
try:
|
|
card['number'] = int(card['number'])
|
|
except ValueError:
|
|
numstring = ""
|
|
for c in card['number']:
|
|
if c.isdigit():
|
|
numstring += c
|
|
if len(numstring) == 0:
|
|
card['number'] = 999999
|
|
else:
|
|
card['number'] = int(numstring)
|
|
# Generate a list of cards in the set
|
|
set_cards = []
|
|
for card in cards:
|
|
if card['name'] in ['Plains', 'Island', 'Swamp', 'Mountain', 'Forest']:
|
|
continue
|
|
if 'side' in card and card['side'] != 'a':
|
|
continue
|
|
if card['number'] > d['baseSetSize']:
|
|
continue
|
|
set_cards += [card]
|
|
# Create a list of word counts by rarity
|
|
rarities = {'common': [], 'uncommon': [], 'rare': [], 'mythic': []}
|
|
for card in set_cards:
|
|
if 'text' not in card:
|
|
words = 0
|
|
else:
|
|
words = len(card['text'].split())
|
|
rarities[card['rarity']] += [words]
|
|
line = [d['code']]
|
|
for r in ['common', 'uncommon', 'rare', 'mythic']:
|
|
if len(rarities[r]) > 0:
|
|
mean = round(sum(rarities[r]) / len(rarities[r]), 1)
|
|
median = statistics.median(rarities[r])
|
|
else:
|
|
mean = 0
|
|
median = 0
|
|
line += [mean, median]
|
|
line = [str(x) for x in line]
|
|
textline = ",".join(line)
|
|
print(textline)
|
|
return textline
|
|
|
|
|
|
codes = []
|
|
codes += ['LEA','3ED','4ED','5ED','6ED','7ED','8ED','9ED','10E','M10','M11','M12','M13','M14','M15','ORI','M19','M20',
|
|
'M21']
|
|
codes += ['POR','P02','PTK','S99']
|
|
codes += ['ARN','ATQ','LEG','DRK','FEM','HML']
|
|
codes += ['ICE','ALL','MIR','VIS','WTH','TMP','STH','EXO','USG','ULG','UDS','MMQ','NEM','PCY','INV','PLS','APC',
|
|
'ODY','TOR','JUD','ONS','LGN','SCG','MRD','DST','5DN','CHK','BOK','SOK','RAV','GPT','DIS','CSP',
|
|
'TSP','PLC','FUT','LRW','MOR','SHM','EVE','ALA','CON_','ARB','ZEN','WWK','ROE','SOM','MBS','NPH',
|
|
'ISD','DKA','AVR','RTR','GTC','DGM','THS','BNG','JOU','KTK','FRF','DTK','BFZ','OGW','SOI','EMN','KLD','AER',
|
|
'AKH','HOU','XLN','RIX','DOM','GRN','RNA','WAR','ELD','THB','IKO','ZNR']
|
|
codes += ['MMA','MM2','EMA','MM3','IMA','A25','UMA','2XM']
|
|
codes += ['MH1','CMR','JMP']
|
|
|
|
output = open('retail_set_words.csv', 'w+')
|
|
output.write('Set,c (mean),c (med),u (mean),u (med),r (mean),r (med),m (mean),m (med)\n')
|
|
for code in codes:
|
|
line = analyze_set(code)
|
|
output.write(line + '\n')
|
|
output.close()
|
|
|