Add authentication for rest client

bfxapi/rest/BfxRest.py

@@ -4,12 +4,16 @@ 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)
 
@@ -19,9 +23,23 @@ class BfxRest:
       async with session.get(url) as resp:
         text =  await resp.text()
         if resp.status is not 200:
-          raise Exception('Unable to seed trades. Received status {} - {}'
-            .format(resp.status, text))
-        return json.loads(text)
+          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)
@@ -43,3 +61,12 @@ class BfxRest:
     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 ]