zkb_crest.py

from evesrp.killmail import CRESTMail


class SubmittedCRESTZKillmail(CRESTMail):
    """Accepts and validates CREST killmail links, but submits them to
    ZKillboard and substitutes the zKB link in as the canonical link
    """
    
    def __init__(self, url, **kwargs):
        # Let CRESTMail validate the CREST link
        super(self.__class__, self).__init__(url, **kwargs)
        # Submit the CREST URL to ZKillboard
        resp = self.requests_session.post('https://zkillboard.com/post/',
                data={'killmailurl': url})
        # Use the URL we get from ZKillboard as the new URL
        self.url = resp.url
        # Grab zkb's data from their API
        api_url = ('https://zkillboard.com/api/no-attackers/'
                   'no-items/killID/{}').format(self.kill_id)
        zkb_api = self.requests_session.get(api_url)
        retrieval_error = LookupError(u"Error retrieving killmail data (zKB): {}"
                .format(zkb_api.status_code))
        if zkb_api.status_code != 200:
            raise retrieval_error
        try:
            json = zkb_api.json()
        except ValueError as e:
            raise retrieval_error
        try:
            json = json[0]
        except IndexError as e:
            raise LookupError(u"Invalid killmail: {}".format(url))
        # The only thing CREST doesn't give us is the value, which new versions
        # of zkb return
        # Python2 hates passing these classes around, so we're stuck importing the 
        from decimal import Decimal
        try:
            self.value = Decimal(json[u'zkb'][u'totalValue'])
        except KeyError:
            self.value = Decimal(0)