mirror of
https://github.com/Card-Forge/forge.git
synced 2025-11-18 11:48:02 +00:00
279 lines
7.0 KiB
Python
279 lines
7.0 KiB
Python
# This python script is designed to handle the following: individual cards located in /res/cardsfolder
|
|
# Insert of rarity, picture, and sets of
|
|
|
|
from httplib import HTTP
|
|
from urlparse import urlparse
|
|
from urllib import urlopen
|
|
import os
|
|
|
|
def clean(name):
|
|
return name.lower().replace('-',' ').replace(',','').replace('_', ' ').replace('\'', '').replace('\"', '').replace('.', '').strip()
|
|
|
|
def checkURL(url):
|
|
p = urlparse(url)
|
|
h = HTTP(p[1])
|
|
h.putrequest('HEAD', p[2])
|
|
h.endheaders()
|
|
if h.getreply()[0] == 200: return 1
|
|
else: return 0
|
|
|
|
def getURL(url):
|
|
return urlopen(url).read()
|
|
|
|
def getSVarString(line, str):
|
|
start = line.find(str) + len(str) + 1
|
|
return line[start:].strip()
|
|
|
|
def getRarity(name, html):
|
|
# magiccards.info uses (<rarity>) on the page for rarity.
|
|
# since older editions had funky things like c2, and u1, only use open parenthesis for this search
|
|
landCount = html.count("(Land")
|
|
commonCount = html.count("(Common")
|
|
uncommonCount = html.count("(Uncommon")
|
|
rareCount = html.count("(Rare")
|
|
mythicCount = html.count("(Mythic Rare)")
|
|
if (landCount > 0):
|
|
return 'Land'
|
|
elif (commonCount + uncommonCount + rareCount + mythicCount == 0):
|
|
err.write("Bad magiccards.info Query: " + name + '\n')
|
|
return ''
|
|
elif (commonCount >= uncommonCount and commonCount >= rareCount and commonCount >= mythicCount):
|
|
return 'Common'
|
|
elif (commonCount < uncommonCount and uncommonCount >= rareCount and uncommonCount >= mythicCount):
|
|
return 'Uncommon'
|
|
elif (rareCount >= mythicCount):
|
|
return 'Rare'
|
|
else:
|
|
return 'Mythic'
|
|
|
|
def getPicture(name):
|
|
urlName = name.replace(' ', '_')
|
|
picUrl ='http://www.wizards.com/global/images/magic/general/' + urlName + '.jpg'
|
|
if not checkURL(picUrl):
|
|
err.write("Bad Picture URL " + name + '\n')
|
|
return ''
|
|
return picUrl
|
|
|
|
def getSets(name, html):
|
|
delimitedStr = ''
|
|
first = True;
|
|
for s in sets:
|
|
if (html.find(s + ' (') != -1) or (html.find(s + '</a> (') != -1):
|
|
if not first:
|
|
delimitedStr = delimitedStr + ','
|
|
first = False
|
|
delimitedStr = delimitedStr + sets[s]
|
|
|
|
return delimitedStr
|
|
|
|
def initSets():
|
|
# Base Sets
|
|
sets['Limited Edition Beta'] = 'LEB'
|
|
sets['Unlimited Edition'] = '2ED'
|
|
sets['Revised Edition'] = '3ED'
|
|
sets['Fourth Edition'] = '4ED'
|
|
sets['Fifth Edition'] = '5ED'
|
|
sets['Classic Sixth Edition'] = '6ED'
|
|
sets['Seventh Edition'] = '7ED'
|
|
sets['Eighth Edition'] = '8ED'
|
|
sets['Ninth Edition'] = '9ED'
|
|
sets['Tenth Edition'] = '10E'
|
|
sets['Magic 2010'] = 'M10'
|
|
sets['Magic 2011'] = 'M11'
|
|
|
|
# Portal
|
|
sets['Portal'] = 'POR'
|
|
sets['Portal Second Age'] = 'P02'
|
|
sets['Portal Three Kingdoms'] = 'PTK'
|
|
|
|
# Starter
|
|
sets['Starter 1999'] = 'S99'
|
|
sets['Starter 2000'] = 'S00'
|
|
|
|
# Early sets
|
|
sets['Arabian Nights'] = 'ARN'
|
|
sets['Antiquities'] = 'ATQ'
|
|
sets['Legends'] = 'LEG'
|
|
sets['The Dark'] = 'DRK'
|
|
sets['Fallen Empires'] = 'FEM'
|
|
sets['Homelands'] = 'HML'
|
|
|
|
# Ice Age
|
|
sets['Ice Age'] = 'ICE'
|
|
sets['Alliances'] = 'ALL'
|
|
sets['Coldsnap'] = 'CSP'
|
|
|
|
# Mirage
|
|
sets['Mirage'] = 'MIR'
|
|
sets['Visions'] = 'VIS'
|
|
sets['Weatherlight'] = 'WTH'
|
|
|
|
# Rath Cycle
|
|
sets['Tempest'] = 'TMP'
|
|
sets['Stronghold'] = 'STH'
|
|
sets['Exodus'] = 'EXO'
|
|
|
|
# Artifacts Cycle
|
|
sets['Urza\'s Saga'] = 'USG'
|
|
sets['Urza\'s Legacy'] = 'ULG'
|
|
sets['Urza\'s Destiny'] = 'UDS'
|
|
|
|
# Masques
|
|
sets['Mercadian Masques'] = 'MMQ'
|
|
sets['Nemesis'] = 'NMS'
|
|
sets['Prophecy'] = 'PCY'
|
|
|
|
# Invasion
|
|
sets['Invasion'] = 'INV'
|
|
sets['Planeshift'] = 'PLS'
|
|
sets['Apocalypse'] = 'APC'
|
|
|
|
# Odyssey
|
|
sets['Odyssey'] = 'ODY'
|
|
sets['Torment'] = 'TOR'
|
|
sets['Judgment'] = 'JUD'
|
|
|
|
# Onslaught
|
|
sets['Onslaught'] = 'ONS'
|
|
sets['Legions'] = 'LGN'
|
|
sets['Scourge'] = 'SCG'
|
|
|
|
# Mirrodin
|
|
sets['Mirrodin'] = 'MRD'
|
|
sets['Darksteel'] = 'DST'
|
|
sets['Fifth Dawn'] = '5DN'
|
|
|
|
# Kamigawa
|
|
sets['Champions of Kamigawa'] = 'CHK'
|
|
sets['Betrayers of Kamigawa'] = 'BOK'
|
|
sets['Saviors of Kamigawa'] = 'SOK'
|
|
|
|
# Ravnica
|
|
sets['Ravnica: City of Guilds'] = 'RAV'
|
|
sets['Guildpact'] = 'GPT'
|
|
sets['Dissension'] = 'DIS'
|
|
|
|
# Time Spiral
|
|
sets['Time Spiral'] = 'TSP'
|
|
sets['Planar Chaos'] = 'PLC'
|
|
sets['Future Sight'] = 'FUT'
|
|
|
|
# Lorwyn
|
|
sets['Lorwyn'] = 'LRW'
|
|
sets['Morningtide'] = 'MOR'
|
|
|
|
# Shadowmoor
|
|
sets['Shadowmoor'] = 'SHM'
|
|
sets['Eventide'] = 'EVE'
|
|
|
|
# Alara
|
|
sets['Shards of Alara'] = 'ALA'
|
|
sets['Conflux'] = 'CON'
|
|
sets['Alara Reborn'] = 'ARB'
|
|
|
|
# Zendikar
|
|
sets['Zendikar'] = 'ZEN'
|
|
sets['Worldwake'] = 'WWK'
|
|
sets['Rise of the Eldrazi'] = 'ROE'
|
|
|
|
# Scars of Mirrodin
|
|
sets['Scars of Mirrodin'] = 'SOM'
|
|
sets['Mirrodin Beseiged'] = 'MBS'
|
|
#sets['Unknown Mirrodin Set']
|
|
|
|
class Card:
|
|
def __init__(self, name, cleanName):
|
|
self.name = name
|
|
self.cleanName = cleanName
|
|
self.hasPicture = False
|
|
self.picURL = ''
|
|
self.hasRarity = False
|
|
self.rarity = ''
|
|
self.hasSets = False
|
|
self.sets = ''
|
|
self.lines = ''
|
|
|
|
#get master card list and drop into a dictionary
|
|
folder = "cardsfolder"
|
|
err = open('cardsScript.log','w')
|
|
cardDict = {}
|
|
rarityStr = 'SVar:Rarity'
|
|
pictureStr = 'SVar:Picture'
|
|
setStr = 'SVar:Sets'
|
|
sets = {}
|
|
initSets()
|
|
|
|
|
|
for fileName in os.listdir(folder):
|
|
# parse cardsfolder for Card Lines and Rarity/Picture SVars. Filling in any gaps
|
|
file = open(folder + '\\' + fileName)
|
|
cleanName = fileName.replace('.txt', '')
|
|
|
|
html = ''
|
|
|
|
line = file.readline().strip()
|
|
name = line.replace('Name:','')
|
|
|
|
card = Card(name, cleanName)
|
|
cardDict[cleanName] = card
|
|
card.lines = line + '\n'
|
|
|
|
line = file.readline().strip()
|
|
|
|
while line != 'End':
|
|
if line == '' or line == 'End':
|
|
break
|
|
|
|
if line.find(rarityStr) != -1:
|
|
card.hasRarity = True
|
|
card.rarity = getSVarString(line, rarityStr)
|
|
elif line.find(pictureStr) != -1:
|
|
card.hasPicture = True
|
|
card.picURL = getSVarString(line, pictureStr)
|
|
elif line.find(setStr) != -1:
|
|
card.hasSets = True
|
|
card.sets = getSVarString(line, setStr)
|
|
else:
|
|
card.lines += line +'\n'
|
|
|
|
line = file.readline().strip()
|
|
|
|
if card.hasRarity and card.hasPicture and card.hasSets:
|
|
err.write(card.name + '... No Changes' + '\n')
|
|
continue
|
|
|
|
if not card.hasRarity:
|
|
html = getURL('http://magiccards.info/query?q=!'+card.name)
|
|
rarity = getRarity(card.name, html)
|
|
if not rarity == '':
|
|
card.hasRarity = True
|
|
card.rarity = rarity
|
|
|
|
if not card.hasPicture:
|
|
pic = getPicture(card.cleanName)
|
|
if not pic == '':
|
|
card.hasPicture = True
|
|
card.picURL = pic
|
|
|
|
if not card.hasSets:
|
|
if html == '':
|
|
html = getURL('http://magiccards.info/query?q=!'+card.name)
|
|
allSets = getSets(card.name, html)
|
|
if not allSets == '':
|
|
card.hasSets = True
|
|
card.sets = allSets
|
|
|
|
file = open(folder + "/" + fileName, 'w')
|
|
file.write(card.lines)
|
|
if card.hasRarity:
|
|
file.write('SVar:Rarity:'+card.rarity + '\n')
|
|
if card.hasPicture:
|
|
file.write('SVar:Picture:'+card.picURL + '\n')
|
|
if card.hasSets:
|
|
file.write('SVar:Sets:'+card.sets + '\n')
|
|
file.write('End')
|
|
file.close()
|
|
err.write(card.name + '... Updated')
|
|
|
|
|
|
err.close() |