init2

res/reprintSetInfo.py

@@ -0,0 +1,115 @@
+#!/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:'
+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:
+            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()