mop.py

from market import Market
import time
import base64
import hmac
import urllib
import urllib2
import hashlib
import sys
import json
sys.path.append('../')
sys.path.append('.')
import config
from decimal import Decimal


class PrivateBitcoinCentral(Market):
    balance_url = "https://bitcoin-central.net/api/v1/balances/"
    trade_url = "https://bitcoin-central.net/api/v1/trade_orders/"

    def __init__(self):
        self.name = self.__class__.__name__
        self.username = config.bitcoincentral_username
        self.password = config.bitcoincentral_password
        self.currency = "EUR"
        self.get_info()

    def _create_nonce(self):
        return int(time.time() * 1000000)

    def _send_request(self, url, params=[], extra_headers=None):
        headers = {
            'Content-type': 'application/json',
            'Accept': 'application/json, text/javascript, */*; q=0.01',
            'User-Agent': 'Mozilla/4.0 (compatible; MSIE 5.5; Windows NT)'
        }
        if extra_headers is not None:
            for k, v in extra_headers.iteritems():
                headers[k] = v

        req = None
        if params:
            req = urllib2.Request(url, json.dumps(params), headers=headers)
        else:
            req = urllib2.Request(url, headers=headers)
        base64string = base64.encodestring('%s:%s' % (
            self.username, self.password)).replace('\n', '')
        req.add_header("Authorization", "Basic %s" % base64string)
        code = 422
        try:
            response = urllib2.urlopen(req)
            code = response.getcode()
        except urllib2.HTTPError, e:
            code = 422
        if code == 200:
            jsonstr = response.read()
            return json.loads(jsonstr)
        return None

    def trade(self, amount, ttype, price=None):
        # params = [("amount", amount), ("currency", self.currency), ("type",
        # ttype)]
        params = {"amount": amount, "currency": self.currency, "type": ttype}
        if price:
            params["price"] = price
        response = self._send_request(self.trade_url, params)
        return response

    def buy(self, amount, price=None):
        response = self.trade(amount, "buy", price)
        print response

    def sell(self, amount, price=None):
        response = self.trade(amount, "sell", price)
        print response

    def get_info(self):
        response = self._send_request(self.balance_url)
        if response:
            self.btc_balance = Decimal(response["BTC"])
            self.eur_balance = Decimal(response["EUR"])

    def __str__(self):
        return str({"btc_balance": self.btc_balance, "eur_balance": self.eur_balance})


if __name__ == "__main__":
    mtgox = PrivateBitcoinCentral()
#    mtgox.buy(0.01)
#    mtgox.sell(0.01)
    print mtgox