mirror of
https://github.com/aljazceru/bitfinex-api-py.git
synced 2025-12-20 23:34:21 +01:00
73 lines
2.6 KiB
Python
73 lines
2.6 KiB
Python
import asyncio
|
|
import aiohttp
|
|
import time
|
|
import json
|
|
|
|
from ..utils.CustomLogger import CustomLogger
|
|
from ..utils.auth import generate_auth_headers
|
|
from ..models import Wallet
|
|
|
|
class BfxRest:
|
|
|
|
def __init__(self, API_KEY, API_SECRET, host='https://api.bitfinex.com/v2', loop=None,
|
|
logLevel='INFO', *args, **kwargs):
|
|
self.loop = loop or asyncio.get_event_loop()
|
|
self.API_KEY = API_KEY
|
|
self.API_SECRET = API_SECRET
|
|
self.host = host
|
|
self.logger = CustomLogger('BfxRest', logLevel=logLevel)
|
|
|
|
async def fetch(self, endpoint):
|
|
url = '{}/{}'.format(self.host, endpoint)
|
|
async with aiohttp.ClientSession() as session:
|
|
async with session.get(url) as resp:
|
|
text = await resp.text()
|
|
if resp.status is not 200:
|
|
raise Exception('GET {} failed with status {} - {}'
|
|
.format(url, resp.status, text))
|
|
return await resp.json(text)
|
|
|
|
async def post(self, endpoint, data={}):
|
|
url = '{}/{}'.format(self.host, endpoint)
|
|
sData = json.dumps(data)
|
|
headers = generate_auth_headers(
|
|
self.API_KEY, self.API_SECRET, endpoint, sData)
|
|
headers["content-type"] = "application/json"
|
|
async with aiohttp.ClientSession() as session:
|
|
async with session.post(url, headers=headers, data=sData) as resp:
|
|
text = await resp.text()
|
|
if resp.status is not 200:
|
|
raise Exception('POST {} failed with status {} - {}'
|
|
.format(url, resp.status, text))
|
|
return await resp.json()
|
|
|
|
async def get_seed_candles(self, symbol):
|
|
endpoint = 'candles/trade:1m:{}/hist?limit=5000&_bfx=1'.format(symbol)
|
|
time_difference = (1000 * 60) * 5000
|
|
# get now to the nearest min
|
|
now = int(round((time.time() // 60 * 60) * 1000))
|
|
task_batch = []
|
|
for x in range(0, 10):
|
|
start = x * time_difference
|
|
end = now - (x * time_difference) - time_difference
|
|
e2 = endpoint + '&start={}&end={}'.format(start, end)
|
|
task_batch += [asyncio.ensure_future(self.fetch(e2))]
|
|
self.logger.info("Downloading seed candles from Bitfinex...")
|
|
# call all fetch requests async
|
|
done, _ = await asyncio.wait(*[ task_batch ])
|
|
candles = []
|
|
for task in done:
|
|
candles += task.result()
|
|
candles.sort(key=lambda x: x[0], reverse=True)
|
|
self.logger.info("Downloaded {} candles.".format(len(candles)))
|
|
return candles
|
|
|
|
##################################################
|
|
# Wallets #
|
|
##################################################
|
|
|
|
async def get_wallets(self):
|
|
endpoint = "auth/r/wallets"
|
|
raw_wallets = await self.post(endpoint)
|
|
return [ Wallet(rw[0], rw[1], rw[2], rw[3]) for rw in raw_wallets ]
|