amount.py

__author__ = 'gsibble'

from decimal import Decimal
from coinbase.config import CENTS_PER_BITCOIN, CENTS_PER_OTHER

class CoinbaseAmount(Decimal):

    def __new__(self, amount, currency):
        return Decimal.__new__(self, amount)

    def __init__(self, amount, currency):
        super(CoinbaseAmount, self).__init__()
        self.currency = currency

    def __eq__(self, other, *args, **kwargs):
        if isinstance(other, self.__class__) and self.currency != other.currency:
            return False
        return super(CoinbaseAmount, self).__eq__(other, *args, **kwargs)

    @classmethod
    def from_cents(cls, cents, currency):
        if currency == 'BTC':
            return cls(Decimal(cents)/CENTS_PER_BITCOIN, currency)
        else:
            return cls(Decimal(cents)/CENTS_PER_OTHER, currency)