mtg/database.py

59 lines
No EOL
2.1 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
}
elif card and line:
# Load the data into the card
if not card['cost'] and re.match('^X?\d*[WBURG]*$', line):
card['cost'] = line
continue
if not card['type']:
attributes = line.split(' -- ')
type = attributes.pop(0)
card['type'] = type
card['attributes'] = attributes
if not card['power'] and not card['toughness'] and re.match('^\d+([+-]?\*)?\/\d+([+-]?\*)?$', line):
(card['power'], card['toughness']) = line.split('/')
elif not line:
# Finished with current record
if card:
# We're done here
break
else:
# Prepare to read in the next record
inRecord = False
return Card(card['name'], card['type'], card['attributes'], card['cost'], card['power'], card['toughness'])
if __name__ == '__main__':
db = TextDB('db.txt')
c = db.getCard('Aura Gnarlid')
print c