mtg_json_get.py

#extracts data from MTG-JSON file
#Dakota Madden-Fong May 2018

import json

#unzip if zipped only
from pathlib import Path
unzipped = Path('AllSets.json')
if not(unzipped.is_file()):
    print ('Unzipping zip file containing the json file')
    import zipfile
    zip_ref = zipfile.ZipFile('AllSets.json.zip', 'r')
    zip_ref.extractall()
    zip_ref.close()

#read in json file
try:
    jsonsets = json.loads(open('AllSets.json',encoding="utf8").read())
except MemoryError:
    raise Exception('Please ensure you are running 64 bit Python')
print('json file loaded')

def getSets():
    #return alphabetized list of all sets, removing sets for which no card images exist
    retsets = []
    for set in (list(jsonsets.keys())):
        cards = jsonsets[set]['cards']
        empties = False
        exists = False
        multiverse_ids = []
        for card in cards:
            try:
                multiverse_ids.append(card['multiverseId'])
                exists = True
            except:
                multiverse_ids.append(None)
                empties = True
        if (empties and (not exists)):
            pass
        elif (empties and exists):
            pass
        else:
            retsets.append(set)
    retsets = sorted(retsets)
    return retsets

class card_set_json:
    # gather info on a specified set
    
    def __init__(self, setcode):
        try:
            self.cards = jsonsets[setcode]['cards']
        except:
            raise NameError('No set found with that setcode')
        
        self.names = []
        for card in self.cards:
            self.names.append(card['name'])
        # self.names = list of all card names
        self.uids = []
        for card in self.cards:
            self.uids.append(card['uuid'])
        # self.uids = list of all card unique ids
            
        self.multiverse_ids = []
        
        empties = False
        exists = False
        for card in self.cards:
            try:
                self.multiverse_ids.append(card['multiverseId'])
                exists = True
            except:
                self.multiverse_ids.append(None)
                empties = True
        if (empties and (not exists)):
            print('No images accessible')
        elif (empties and exists):
            print('Some images not accessible')
        else:
            print('All images accessible')
            
        
        # self.imgurls = list of all URLs to gatherer card images
        self.imgurls = []
        for id in self.multiverse_ids:
            if (id == None):
                self.imgurls.append(None)
            else: 
                self.imgurls.append('http://gatherer.wizards.com/Handlers/Image.ashx?multiverseid='+str(id)+'&type=card')