#!/usr/bin/env python

# This script grabs the list of all cards in a set and clears out the setInfo
# After running this script, re-run setInfoScript to fill in the slots
# Run this Script with a command line argument, as below

# $ python reprintSetInfo.py <SetAbbr>
# $ python reprintSetInfo.py M12

from httplib import HTTP
from urlparse import urlparse
from urllib import urlopen
import sys
import os

class SetInfo:
    def __init__(self, set, rarity, image):
        self.set = set
        self.rarity = rarity
        self.image = image

class Card:
    def __init__(self, name, cleanName):
        self.name = name
        self.cleanName = cleanName
        self.hasSet = False
        self.sets = {}

def clean(name):
    return name.replace(' ','_').replace('\'','').replace('-','_').replace('"','').replace(',','').lower()

def getCardsInSet():
    html = urlopen('http://magiccards.info/query?q=e:'+setAbbr+'&v=olist').read()

    start = html.find('<th><b>Card name</b></th>')
    end = html.find('</table>', start)
    block = html[start:end]

    while True:
        nameIndex = block.find('.html">')

        if nameIndex == -1:
            break

        nameEnd = block.find('<',nameIndex)

        name = block[nameIndex+7:nameEnd]

        # Add name to array
        nameList.append(clean(name)+'.txt')

        block = block[nameEnd:]

    return

folder = "cardsfolder"
err = open('reprintSetInfo.log','w')

# THIS NEEDS TO BE UPDATED TO THE SET YOU WANT TO UPDATE
# SOME ARE THREE LETTER ABBR. BUT SOME ARE JUST TWO. CHECK BEFORE YOU RUN!
print "Using Set: " + sys.argv[1]
setAbbr = sys.argv[1]

cardDict = {}
setStr = 'SetInfo:'
oracleStr = 'Oracle:'
nameList = []
getCardsInSet()

for fileName in nameList:
	# Join new folder convention cardsfolder/<char>/<cardName>
    filePath = os.path.join(folder, fileName[0].lower(), fileName)
    print filePath

	# if file doesn't exist continue
    if not os.path.isfile(filePath):
        continue

    file = open(filePath)
    cleanName = fileName.replace('.txt', '')

    line = file.readline().strip()
    # Handle name and creation
    name = line.replace('Name:','')

    card = Card(name, cleanName)
    cardDict[cleanName] = card
    card.lines = line + '\n'

    # Start parsing the rest of the data file
    line = file.readline().strip()

    while line != 'End':
        # Skip empty lines
        if line == '':
            line = file.readline().strip()
            continue

        # We really shouldn
        if line == 'End':
            break

        # Skip SetInfo lines
        if line.find(setStr) == -1 and line.find(oracleStr) == -1:
            card.lines += line +'\n'

        line = file.readline().strip()

    file = open(filePath, 'w')
    file.write(card.lines)

    file.write('End')
    file.close()
    err.write(card.name + '... Updated\n')

err.close()