data_prep.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import json
from pprint import pprint
import codecs
import sys
from get_ability import *
import re
from mtgjson import CardDb


#create db
db = CardDb.from_file(db_file='AllSets.json')


#read in our list of cards to check
with open('ORI.json', encoding="utf-8") as f:
    lines = f.read()
    print(type(lines))
    lines = json.loads(lines)
    #x = str(lines).encode("utf-8")
#print(x)
#print(str(lines['cards']).encode('utf-8'))
#getlist of cards
card_list = []


for key in lines['cards']:
    if "<class 'str'>" in key['name']:
        continue
    name =key['name'].encode('utf-8').decode('UTF-8')
    card = db.cards_by_name[name]


    if "Creature" == card.types[0]:
        if len(card.colors) == 1:
            card_list.append(card)


    
 

#get the values we want from the list of cards
for item in card_list:
    print("##",item.name)
    if item.power == "*":
        adjusted_power = "*"
    elif item.power == "0":
        adjusted_power = -1
    elif "+" in item.power:
        value = item.power.split("+")
        adjusted_power = int(value[0]) / int(item.cmc)
    
    else:
        adjusted_power =  int(item.power)/int(item.cmc) 

    if item.toughness == "*":
        adjusted_toughness = "*"
    elif item.toughness == "0":
        adjusted_toughness = -1
    elif "+" in item.toughness:
        value = item.toughness.split("+")
        adjusted_toughness = int(value[0]) / int(item.cmc)
    

    else:
        adjusted_toughness = int(item.toughness)/int(item.cmc)
    
    tf = ""
    if len(item.subtypes) == 2:
        tf = item.subtypes[1] 

    ability = get_ability(item)  
    
    print(str(item.cmc) + "," + str(adjusted_power) + "," + str(adjusted_toughness) + "," + item.subtypes[0] + "," + tf  + "," + ability + "," + item.colors[0])