# 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
# MAKE SURE THE setAbbr VARIABLE IS UPDATED TO THE SET YOU WANT TO CLEAR OUT

from httplib import HTTP
from urlparse import urlparse
from urllib import urlopen
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!
setAbbr = 'mbs' 

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

for fileName in nameList:
	# if file doesn't exist continue
	filePath = os.path.join(folder, fileName)
	
	print filePath
	
	if os.path.isfile(filePath) == False:
		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:
			card.lines += line +'\n'
			
		line = file.readline().strip()

	file = open(os.path.join(folder, fileName), 'w')
	file.write(card.lines)
			
	file.write('End')
	file.close()
	err.write(card.name + '... Updated\n')

err.close()