Denis Lebedev
Return mtga_card_id in predict output
8f19b3a unverified
import json
import os
#TODO: load inside a class
current_dir = os.path.dirname(os.path.abspath(__file__))
cards_by_mtga_id = {}
cards_by_name = {}
# Load DSK
with open(os.path.join('data', 'dsk_cards.json'), 'r') as file:
dsk_cards = json.load(file)
with open(os.path.join('data', 'dsk_lands.json'), 'r') as file:
dsk_lands = json.load(file)[0]['data']
for card in dsk_cards:
cards_by_mtga_id[card['arena_id']] = card
cards_by_name[card['name']] = card
for card in dsk_lands:
cards_by_mtga_id[card['arena_id']] = card
cards_by_name[card['name']] = card
assert cards_by_mtga_id[92372]['name'] == 'Plains'
# Load FDN
with open(os.path.join('data', 'fdn_cards.json'), 'r') as file:
fdn_cards = json.load(file)
with open(os.path.join('data', 'fdn_lands.json'), 'r') as file:
fdn_lands = json.load(file)[0]['data']
for card in fdn_cards:
cards_by_mtga_id[card['arena_id']] = card
cards_by_name[card['name']] = card
for card in fdn_lands:
cards_by_mtga_id[card['arena_id']] = card
cards_by_name[card['name']] = card
assert cards_by_mtga_id[95200]['name'] == 'Forest'
def mtga_id_to_card_name(id):
return cards_by_mtga_id[id]['name']
def card_name_to_image_url(name):
return cards_by_name[name]['image_uris']['normal']
def card_name_to_mtga_id(name):
return cards_by_name[name]['arena_id']
def mtga_id_to_image_url(id):
return cards_by_mtga_id[id]['image_uris']['normal']