Text card database support
This commit is contained in:
parent
e43445af78
commit
4cc0bb905b
2 changed files with 77985 additions and 0 deletions
59
database.py
Normal file
59
database.py
Normal file
|
@ -0,0 +1,59 @@
|
||||||
|
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
|
Loading…
Reference in a new issue