def __init__(self, key=None, secret=None, session=None, engine=None, red=None, cfg=None):
self.cfg = get_config(self.NAME.lower()) if cfg is None else cfg
self.key = key if key is not None else self.cfg.get(self.NAME.lower(), 'key')
self.secret = secret if secret is not None else self.cfg.get(self.NAME.lower(), 'secret')
self.session = session
self.engine = engine
self.red = red
self.logger = None
def test_setup_logger(self):
cfg = get_config("test")
assert cfg.get('log', 'LOGFILE') == '/tmp/test.log'
logger = setup_logging('test', cfg=cfg)
assert isinstance(logger, logging.getLoggerClass())
message = 'testing 1.2.3.'
logger.exception(message)
assert os.path.isfile('/tmp/test.log')
def get_active_markets(exchange):
active_markets = red.get('%s_active_markets' % exchange)
if active_markets is not None and len(active_markets) > 2: # is not "[]"
markets = json.loads(active_markets)
else:
cfg = get_config(name=exchange)
markets = json.loads(cfg.get(exchange, 'active_markets'))
return markets
def test_get_environment_config(self):
path = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'env_cfg.ini')
os.environ["TEST_CONFIG_FILE"] = path
cfg = get_config("test")
assert cfg.get('db', 'SA_ENGINE_URI') == 'sqlite:////tmp/env_test.db'
assert cfg.get('log', 'LOGFILE') == '/tmp/env_test.log'
def test_get_installed_config(self):
cfg = get_config("test_env")
assert cfg.get('db', 'SA_ENGINE_URI') == 'sqlite:////tmp/env_test.db'
assert cfg.get('log', 'LOGFILE') == '/tmp/env_test.log'
def test_get_default_config(self):
cfg = get_config("test_not_known")
assert cfg.get('db', 'SA_ENGINE_URI') == 'sqlite:////tmp/test.db'
assert cfg.get('log', 'LOGFILE') == '/tmp/test.log'
def setUp(self):
self.ses, self.eng = create_session_engine(cfg=get_config("helper"))
setup_database(self.eng, modules=[um, em, wm])
def test_create_session_engine_cfg():
cfg = get_config("helper")
ses, eng = create_session_engine(cfg=cfg)
assert str(eng.url) == cfg.get('db', 'SA_ENGINE_URI')
"""
The main trade_manager module. Initializes sessions, and connections.
"""
from sqlalchemy_models import wallet as wm, exchange as em, user as um, create_session_engine, setup_database
from tapp_config import get_config
CFG = get_config('trade_manager')
ses, eng = create_session_engine(cfg=CFG)
setup_database(eng, modules=[wm, em, um])
NETWORK_COMMODITY_MAP = {'BTC': 'Bitcoin', 'DASH': 'Dash', 'ETH': 'Ethereum', 'LTC': 'Litecoin'}
EXCHANGES = ['kraken', 'bitfinex', 'poloniex']
import datetime
import json
from ledger import Amount
from sqlalchemy_models import (setup_database,
create_session_engine, wallet as wm,
user as um)
from tapp_config import get_config, setup_logging
CFG = get_config('desw')
ses, eng = create_session_engine(cfg=CFG)
setup_database(eng, modules=[wm, um])
logger = setup_logging('desw', prefix='desw', cfg=CFG)
def confirm_send(address, amount, ref_id=None, session=ses):
"""
Confirm that a send has completed. Does not actually check confirmations,
but instead assumes that if this is called, the transaction has been
completed. This is because we assume our own sends are safe.
:param session:
:param str ref_id: The updated ref_id for the transaction in question
:param str address: The address that was sent to
:param Amount amount: The amount that was sent
"""
debitq = session.query(wm.Debit)
debitq = debitq.filter(wm.Debit.address == address)
debitq = debitq.filter(wm.Debit.amount == amount)
debit = debitq.filter(wm.Debit.transaction_state == 'unconfirmed').first()
