mtg/database.py

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 not card['rarity'] and re.match('^([A-Z0-9]+-[{0}](, )?)+$'.format(' '.join(Card.rarities.keys())), line):
                        sets = line.split(', ')
                        currentSet = sets.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['rarity'], card['text'])

if __name__ == '__main__':
    db = TextDB('db.txt')
    c = db.getCard('Aura Gnarlid')
    print c