stable/freqtrade/exchange/__init__.py

165 lines
3.8 KiB
Python
Raw Normal View History

2017-05-12 17:11:56 +00:00
import enum
2017-05-14 12:14:16 +00:00
import logging
2017-11-05 14:21:16 +00:00
from random import randint
2017-11-05 15:12:58 +00:00
from typing import List, Dict, Any
2017-09-01 19:11:46 +00:00
2017-10-06 10:22:04 +00:00
import arrow
from freqtrade.exchange.bittrex import Bittrex
from freqtrade.exchange.interface import Exchange
2017-05-12 17:11:56 +00:00
2017-05-14 12:14:16 +00:00
logger = logging.getLogger(__name__)
2017-05-12 17:11:56 +00:00
2017-09-08 21:10:22 +00:00
# Current selected exchange
_API: Exchange = None
2017-10-06 10:22:04 +00:00
_CONF: dict = {}
2017-05-12 17:11:56 +00:00
2017-11-05 14:21:16 +00:00
# Holds all open sell orders for dry_run
2017-11-05 15:12:58 +00:00
_DRY_RUN_OPEN_ORDERS: Dict[str, Any] = {}
2017-11-05 14:21:16 +00:00
2017-05-12 17:11:56 +00:00
2017-10-06 10:22:04 +00:00
class Exchanges(enum.Enum):
"""
Maps supported exchange names to correspondent classes.
"""
BITTREX = Bittrex
2017-05-12 17:11:56 +00:00
def init(config: dict) -> None:
"""
Initializes this module with the given config,
it does basic validation whether the specified
exchange and pairs are valid.
:param config: config to use
:return: None
"""
global _CONF, _API
2017-09-08 21:10:22 +00:00
_CONF.update(config)
if config['dry_run']:
logger.info('Instance is running with dry_run enabled')
2017-10-06 10:22:04 +00:00
exchange_config = config['exchange']
# Find matching class for the given exchange name
2017-10-07 15:38:33 +00:00
name = exchange_config['name']
try:
exchange_class = Exchanges[name.upper()].value
except KeyError:
2017-10-06 10:22:04 +00:00
raise RuntimeError('Exchange {} is not supported'.format(name))
_API = exchange_class(exchange_config)
# Check if all pairs are available
2017-10-06 10:22:04 +00:00
validate_pairs(config['exchange']['pair_whitelist'])
2017-10-01 21:28:09 +00:00
def validate_pairs(pairs: List[str]) -> None:
"""
Checks if all given pairs are tradable on the current exchange.
Raises RuntimeError if one pair is not available.
:param pairs: list of pairs
:return: None
"""
markets = _API.get_markets()
2017-10-01 21:28:09 +00:00
for pair in pairs:
if pair not in markets:
raise RuntimeError('Pair {} is not available at {}'.format(pair, _API.name.lower()))
def buy(pair: str, rate: float, amount: float) -> str:
2017-09-08 21:10:22 +00:00
if _CONF['dry_run']:
2017-11-05 14:21:16 +00:00
global _DRY_RUN_OPEN_ORDERS
order_id = 'dry_run_buy_{}'.format(randint(0, 1e6))
_DRY_RUN_OPEN_ORDERS[order_id] = {
'pair': pair,
'rate': rate,
'amount': amount,
'type': 'LIMIT_BUY',
'remaining': 0.0,
'opened': arrow.utcnow().datetime,
'closed': arrow.utcnow().datetime,
}
return order_id
2017-10-06 10:22:04 +00:00
return _API.buy(pair, rate, amount)
def sell(pair: str, rate: float, amount: float) -> str:
2017-09-08 21:10:22 +00:00
if _CONF['dry_run']:
2017-11-05 14:21:16 +00:00
global _DRY_RUN_OPEN_ORDERS
order_id = 'dry_run_sell_{}'.format(randint(0, 1e6))
_DRY_RUN_OPEN_ORDERS[order_id] = {
'pair': pair,
'rate': rate,
'amount': amount,
'type': 'LIMIT_SELL',
'remaining': 0.0,
'opened': arrow.utcnow().datetime,
'closed': arrow.utcnow().datetime,
}
return order_id
2017-10-06 10:22:04 +00:00
return _API.sell(pair, rate, amount)
def get_balance(currency: str) -> float:
2017-09-08 21:10:22 +00:00
if _CONF['dry_run']:
return 999.9
2017-10-06 10:22:04 +00:00
return _API.get_balance(currency)
2017-10-30 23:36:35 +00:00
def get_balances():
2017-11-01 00:25:48 +00:00
if _CONF['dry_run']:
return []
return _API.get_balances()
2017-10-30 23:36:35 +00:00
def get_ticker(pair: str) -> dict:
return _API.get_ticker(pair)
2017-10-06 10:22:04 +00:00
def get_ticker_history(pair: str, minimum_date: arrow.Arrow):
return _API.get_ticker_history(pair, minimum_date)
def cancel_order(order_id: str) -> None:
2017-09-08 21:10:22 +00:00
if _CONF['dry_run']:
2017-10-06 10:22:04 +00:00
return
return _API.cancel_order(order_id)
def get_order(order_id: str) -> Dict:
2017-09-08 21:10:22 +00:00
if _CONF['dry_run']:
order = _DRY_RUN_OPEN_ORDERS[order_id]
2017-11-05 14:21:16 +00:00
order.update({
'id': order_id
})
return order
return _API.get_order(order_id)
def get_pair_detail_url(pair: str) -> str:
return _API.get_pair_detail_url(pair)
def get_markets() -> List[str]:
return _API.get_markets()
def get_name() -> str:
return _API.name
def get_sleep_time() -> float:
return _API.sleep_time
def get_fee() -> float:
return _API.fee