mirror of
https://github.com/Card-Forge/forge.git
synced 2025-11-16 02:38:02 +00:00
265 lines
6.5 KiB
Python
Executable File
265 lines
6.5 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
# This python script is designed to handle the following: individual cards located in /res/cardsfolder
|
|
# Insert of SetInfo data into data files from magiccards.info
|
|
|
|
from httplib import HTTP
|
|
from urlparse import urlparse
|
|
from urllib import urlopen
|
|
import os,fnmatch
|
|
|
|
def getURL(url):
|
|
return urlopen(url).read()
|
|
|
|
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 initSets():
|
|
# Base Sets
|
|
allSets['Limited Edition Alpha'] = 'LEA'
|
|
allSets['Limited Edition Beta'] = 'LEB'
|
|
allSets['Unlimited Edition'] = '2ED'
|
|
allSets['Revised Edition'] = '3ED'
|
|
allSets['Fourth Edition'] = '4ED'
|
|
allSets['Fifth Edition'] = '5ED'
|
|
allSets['Classic Sixth Edition'] = '6ED'
|
|
allSets['Seventh Edition'] = '7ED'
|
|
allSets['Eighth Edition'] = '8ED'
|
|
allSets['Ninth Edition'] = '9ED'
|
|
allSets['Tenth Edition'] = '10E'
|
|
allSets['Magic 2010'] = 'M10'
|
|
allSets['Magic 2011'] = 'M11'
|
|
allSets['Magic 2012'] = 'M12'
|
|
|
|
# Multiplayer Sets
|
|
allSets['Commander'] = 'COM'
|
|
|
|
# Portal
|
|
allSets['Portal'] = 'POR'
|
|
allSets['Portal Second Age'] = 'PO2'
|
|
allSets['Portal Three Kingdoms'] = 'PTK'
|
|
|
|
# Starter
|
|
allSets['Starter 1999'] = 'S99'
|
|
allSets['Starter 2000'] = 'S00'
|
|
|
|
# Early Sets
|
|
allSets['Arabian Nights'] = 'ARN'
|
|
allSets['Antiquities'] = 'ATQ'
|
|
allSets['Legends'] = 'LEG'
|
|
allSets['The Dark'] = 'DRK'
|
|
allSets['Fallen Empires'] = 'FEM'
|
|
allSets['Homelands'] = 'HML'
|
|
|
|
# Ice Age
|
|
allSets['Ice Age'] = 'ICE'
|
|
allSets['Alliances'] = 'ALL'
|
|
allSets['Coldsnap'] = 'CSP'
|
|
|
|
# Mirage
|
|
allSets['Mirage'] = 'MIR'
|
|
allSets['Visions'] = 'VIS'
|
|
allSets['Weatherlight'] = 'WTH'
|
|
|
|
# Rath Cycle
|
|
allSets['Tempest'] = 'TMP'
|
|
allSets['Stronghold'] = 'STH'
|
|
allSets['Exodus'] = 'EXO'
|
|
|
|
# Artifacts Cycle
|
|
allSets['Urza\'s Saga'] = 'USG'
|
|
allSets['Urza\'s Legacy'] = 'ULG'
|
|
allSets['Urza\'s Destiny'] = 'UDS'
|
|
|
|
# Masques
|
|
allSets['Mercadian Masques'] = 'MMQ'
|
|
allSets['Nemesis'] = 'NMS'
|
|
allSets['Prophecy'] = 'PCY'
|
|
|
|
# Invasion
|
|
allSets['Invasion'] = 'INV'
|
|
allSets['Planeshift'] = 'PLS'
|
|
allSets['Apocalypse'] = 'APC'
|
|
|
|
# Odyssey
|
|
allSets['Odyssey'] = 'ODY'
|
|
allSets['Torment'] = 'TOR'
|
|
allSets['Judgment'] = 'JUD'
|
|
|
|
# Onslaught
|
|
allSets['Onslaught'] = 'ONS'
|
|
allSets['Legions'] = 'LGN'
|
|
allSets['Scourge'] = 'SCG'
|
|
|
|
# Mirrodin
|
|
allSets['Mirrodin'] = 'MRD'
|
|
allSets['Darksteel'] = 'DST'
|
|
allSets['Fifth Dawn'] = '5DN'
|
|
|
|
# Kamigawa
|
|
allSets['Champions of Kamigawa'] = 'CHK'
|
|
allSets['Betrayers of Kamigawa'] = 'BOK'
|
|
allSets['Saviors of Kamigawa'] = 'SOK'
|
|
|
|
# Ravnica
|
|
allSets['Ravnica: City of Guilds'] = 'RAV'
|
|
allSets['Guildpact'] = 'GPT'
|
|
allSets['Dissension'] = 'DIS'
|
|
|
|
# Time Spiral
|
|
allSets['Time Spiral'] = 'TSP'
|
|
allSets['Time Spiral "Timeshifted"'] = 'TSB'
|
|
allSets['Planar Chaos'] = 'PLC'
|
|
allSets['Future Sight'] = 'FUT'
|
|
|
|
# Lorwyn
|
|
allSets['Lorwyn'] = 'LRW'
|
|
allSets['Morningtide'] = 'MOR'
|
|
|
|
# Shadowmoor
|
|
allSets['Shadowmoor'] = 'SHM'
|
|
allSets['Eventide'] = 'EVE'
|
|
|
|
# Alara
|
|
allSets['Shards of Alara'] = 'ALA'
|
|
allSets['Conflux'] = 'CFX'
|
|
allSets['Alara Reborn'] = 'ARB'
|
|
|
|
# Zendikar
|
|
allSets['Zendikar'] = 'ZEN'
|
|
allSets['Worldwake'] = 'WWK'
|
|
allSets['Rise of the Eldrazi'] = 'ROE'
|
|
|
|
# Scars of Mirrodin
|
|
allSets['Scars of Mirrodin'] = 'SOM'
|
|
allSets['Mirrodin Besieged'] = 'MBS'
|
|
allSets['New Phyrexia'] = 'NPH'
|
|
|
|
def addSets(card):
|
|
html = getURL('http://magiccards.info/query?q=!'+card.name)
|
|
start = html.find('<br><u><b>Editions:</b></u><br>')
|
|
end = html.find('<br><u><b>Languages:</b></u><br>', start)
|
|
block = html[start:end]
|
|
|
|
print card.name
|
|
|
|
for edition in allSets.keys():
|
|
edIndex = block.find('>'+edition+'<') # Portal/Mirrodin issue
|
|
|
|
if edIndex == -1:
|
|
edIndex = block.find('>'+edition+' (') # Single set issue
|
|
if edIndex == -1:
|
|
continue
|
|
|
|
# Scrape rarity
|
|
rarityIndex = block.find('(',edIndex)
|
|
rarity = block[rarityIndex+1:block.find(')',rarityIndex)]
|
|
raritySpace = rarity.find(' ')
|
|
if raritySpace != -1:
|
|
rarity = rarity[0:raritySpace] # For older cards
|
|
|
|
# What to do with TimeShifted cards?
|
|
if rarity == 'Special' and edition != 'Time Spiral "Timeshifted"':
|
|
continue
|
|
|
|
# Get setAbbreviation and setNumber
|
|
dataIndex = block.rfind('"/',0,edIndex)
|
|
data = block[dataIndex+2:edIndex-1] # 1 instead of 2 because of Portal/Mirrodin Issue
|
|
|
|
splitData = data.split('/')
|
|
setAbbr = splitData[0]
|
|
setNum = splitData[2].replace('.html', '')
|
|
|
|
if len(setNum) > 4:
|
|
# Setnum not available here for most recent set. Switch to the .jpg used on page
|
|
jpgIndex = html.find('.jpg')
|
|
data = html[html.rfind('scans/en/', 0, jpgIndex):jpgIndex]
|
|
|
|
# data = scans/en/[set]/[num]
|
|
splitData = data.split('/')
|
|
setAbbr = splitData[2]
|
|
setNum = splitData[3]
|
|
|
|
image = 'http://magiccards.info/scans/en/' + setAbbr + '/' + setNum + '.jpg'
|
|
|
|
card.sets[allSets[edition]] = SetInfo(allSets[edition], rarity, image)
|
|
|
|
return
|
|
|
|
|
|
#get master card list and drop into a dictionary
|
|
folder = "cardsfolder"
|
|
err = open('setInfoScript.log','w')
|
|
allSets = {}
|
|
initSets()
|
|
cardDict = {}
|
|
setStr = 'SetInfo:'
|
|
|
|
for root, dirnames, filenames in os.walk(folder):
|
|
for fileName in fnmatch.filter(filenames, '*.txt'):
|
|
if fileName.startswith('.'):
|
|
continue
|
|
|
|
# parse cardsfolder for Card Lines and Rarity/Picture SVars. Filling in any gaps
|
|
file = open(os.path.join(root, fileName))
|
|
cleanName = fileName.replace('.txt', '')
|
|
|
|
line = file.readline().strip()
|
|
# Handle name and creation
|
|
name = line.replace('Name:','')
|
|
|
|
card = Card(name.replace(' ','+'), cleanName) #This makes it work on Mac OS X. Will test Windows and FreeBSD when I can.
|
|
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't
|
|
if line == 'End':
|
|
break
|
|
|
|
if line.find(setStr) != -1:
|
|
info = line.replace('SetInfo:','')
|
|
parts = info.split('|')
|
|
|
|
card.hasSet = True
|
|
card.sets[parts[0]] = SetInfo(parts[0], parts[1], parts[2])
|
|
else:
|
|
card.lines += line +'\n'
|
|
|
|
line = file.readline().strip()
|
|
|
|
if not card.hasSet:
|
|
addSets(card)
|
|
card.hasSet = True
|
|
|
|
file = open(os.path.join(root, fileName), 'w')
|
|
file.write(card.lines)
|
|
if card.hasSet:
|
|
for s in card.sets.values():
|
|
file.write('SetInfo:'+ s.set + '|' + s.rarity + '|' + s.image + '\n')
|
|
|
|
file.write('End')
|
|
|
|
err.write(card.name + '... Updated\n')
|
|
|
|
file.close()
|
|
err.close()
|