71 lines
2.8 KiB
Python
71 lines
2.8 KiB
Python
import re
|
|
from mtg import *
|
|
|
|
class Database:
|
|
def getCard(name):
|
|
pass
|
|
|
|
""" Text file database class
|
|
|
|
Loads the card database files found at http://www.yawgatog.com/resources/oracle/
|
|
"""
|
|
class TextDB(Database):
|
|
def __init__(self, filename):
|
|
self.filename = filename
|
|
def getCard(self, name):
|
|
card = None
|
|
with open(self.filename, 'r') as f:
|
|
inRecord = False
|
|
recordName = False
|
|
for line in f:
|
|
line = line.strip()
|
|
if not inRecord:
|
|
recordName = line
|
|
inRecord = True
|
|
if recordName.lower() == name.lower():
|
|
card = {
|
|
'name': recordName,
|
|
'type': '',
|
|
'attributes': [],
|
|
'cost': '',
|
|
'power': 0,
|
|
'toughness': 0,
|
|
'rarity': None,
|
|
'text': [],
|
|
}
|
|
elif card and line:
|
|
# Load the data into the card
|
|
if not card['cost'] and re.match('^X?\d*[{0}]*$'.format(' '.join(Mana.types.keys())), line):
|
|
card['cost'] = line
|
|
continue
|
|
elif not card['type']:
|
|
attributes = line.split(' -- ')
|
|
type = attributes.pop(0)
|
|
card['type'] = type
|
|
card['attributes'] = attributes
|
|
elif not card['power'] and not card['toughness'] and re.match('^\d+([+-]?\*)?\/\d+([+-]?\*)?$', line):
|
|
(card['power'], card['toughness']) = line.split('/')
|
|
elif re.match('^([A-Z0-9]+-[{0}](, )?)+$'.format(''.join(Card.rarities.keys())), line):
|
|
info = line.split(', ')
|
|
card['sets'] = [i.split('-').pop(0) for i in info]
|
|
currentSet = info.pop()
|
|
card['rarity'] = currentSet.split('-').pop()
|
|
else:
|
|
# Ability text
|
|
card['text'].append(line)
|
|
elif not line:
|
|
# Finished with current record
|
|
if card:
|
|
# We're done here
|
|
break
|
|
else:
|
|
# Prepare to read in the next record
|
|
inRecord = False
|
|
if not card:
|
|
return None
|
|
return Card(card['name'], card['type'], card['attributes'], card['cost'], card['power'], card['toughness'], card['sets'], card['rarity'], card['text'])
|
|
|
|
if __name__ == '__main__':
|
|
db = TextDB('db.txt')
|
|
c = db.getCard('Aura Gnarlid')
|
|
print c
|