working on #35, fixes lint

AsyncAlgoTrading · Jul 2, 2019 · e363821 · e363821
1 parent ef8a206
commit e363821
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 4 deletions.
diff --git a/aat/exchanges/gemini.py b/aat/exchanges/gemini.py
@@ -8,7 +8,7 @@
 from datetime import datetime
 from functools import lru_cache
 from ..define import EXCHANGE_MARKET_DATA_ENDPOINT
-from ..enums import TickType, TickType_from_string, OrderType, OrderSubType
+from ..enums import TickType, TickType_from_string, PairType
 from ..exchange import Exchange
 from ..logging import log
 from ..structs import MarketData, Instrument
@@ -121,7 +121,7 @@ def tickToData(self, jsn: dict) -> MarketData:
         volume = float(jsn.get('amount', 0.0))
 
         s = jsn.get('type').upper()
-        if s in ('BLOCK_TRADE', 'FILL'): # Market data can't trigger fill event
+        if s in ('BLOCK_TRADE', 'FILL'):  # Market data can't trigger fill event
             typ = TickType.TRADE
         elif s in ('AUCTION_INDICATIVE', 'AUCTION_OPEN', 'BOOKED', 'INITIAL'):
             typ = TickType.OPEN

diff --git a/aat/exchanges/poloniex.py b/aat/exchanges/poloniex.py
@@ -24,6 +24,7 @@ def heartbeat(self):
     def tickToData(self, jsn: dict) -> MarketData:
         raise NotImplementedError()
 
+
 POLONIEX_CURRENCY_ID = {
     '1CR': '1',
     'ABY': '2',

diff --git a/aat/persistence.py b/aat/persistence.py
@@ -0,0 +1,12 @@
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy import Column, Integer
+# from sqlalchemy import ForeignKey
+# from sqlalchemy.orm import relationship
+
+# Base sqlalchemy
+Base = declarative_base()
+
+
+class Record(Base):
+    __tablename__ = 'records'
+    id = Column(Integer, primary_key=True)
diff --git a/aat/utils.py b/aat/utils.py
@@ -4,15 +4,15 @@
 import pytz
 from datetime import datetime
 from functools import lru_cache
-from .enums import ExchangeType, ExchangeType_from_string, ExchangeTypes, CurrencyType, OrderType, Side, PairType, TradeResult
+from .enums import ExchangeType, ExchangeType_from_string, ExchangeTypes, CurrencyType, OrderType, Side, PairType
 from .exceptions import AATException
 from .logging import log
 
 
 @lru_cache(100)
 def parse_date(indate: str) -> datetime:
     '''parse date
-    
+
     Args:
         indate (string, int, or datetime): input to convert to datetime
     Returns: