mtgotraders.py

#!/usr/bin/python -tt
# -*- coding: utf-8 -*-
'''
A set of helper methods to help get pricing information from MTGO Traders.

Author: John Cleaver
License: BSD 3 Clause License
'''

import requests
import card as mtgcard
import config
import datetime
import codecs
import sys

UTF8Writer = codecs.getwriter('utf8')
sys.stdout = UTF8Writer(sys.stdout)


def get_raw_list(url):
    r = requests.get(url)
    r.raise_for_status()
    return r.text


def parse_list(price_text):
    for line in price_text.splitlines():
        line_list = line.replace(u"<br>", u"").strip().split(u"|")
        if line_list[0] != u"BOOSTER":
            try:
                card_release = mtgcard.find_release_by_name(line_list[3],
                                                            line_list[0])
            except mtgcard.ReleaseNotFoundError as e:
                card = mtgcard.find_card(line_list[3])
                expansion = mtgcard.find_expansion(line_list[0])
                card_release = mtgcard.models.CardRelease()
                card_release.expansion = expansion
                card_release.card = card
                card_release.mtgoprice = mtgcard.models.MTGOPrice()
                if line_list[2] == u"R":
                    card_release.mtgoprice.price = float(line_list[5])
                else:
                    card_release.mtgoprice.foil_price = float(line_list[5])
                card_release.mtgoprice.last_fetch = datetime.datetime.now()
            except mtgcard.CardError as e:
                pass
            except IndexError as e:
                pass
            else:
                if line_list[2] == u"R":
                    card_release.mtgoprice.price = float(line_list[5])
                else:
                    card_release.mtgoprice.foil_price = float(line_list[5])
                card_release.mtgoprice.last_fetch = datetime.datetime.now()


def main():
    mtgcard.models.setup()
    parse_list(get_raw_list(config.mtgotraders_api_url))
    mtgcard.models.close()

if __name__ == '__main__':
    main()