119 lines
3.2 KiB
Python
119 lines
3.2 KiB
Python
import argparse
|
|
import datetime
|
|
import json
|
|
import logging
|
|
import logging.handlers
|
|
import time
|
|
from decimal import Decimal
|
|
from pathlib import Path
|
|
|
|
import fin_defs
|
|
import fin_depo
|
|
|
|
from . import (
|
|
AutoSellConfig,
|
|
log_results,
|
|
order_csv,
|
|
run_auto_sell,
|
|
)
|
|
from . import logger as module_logger
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
################################################################################
|
|
# Application Setup #
|
|
|
|
|
|
def setup_logging(path_log_file: Path):
|
|
"""Enables logging for the terminal and to a log file."""
|
|
path_log_file.parent.mkdir(parents=True, exist_ok=True)
|
|
file_handler = logging.handlers.WatchedFileHandler(filename=path_log_file)
|
|
file_handler.setFormatter(
|
|
logging.Formatter(
|
|
'%(levelname)s:%(asctime)s: %(message)s',
|
|
datefmt='%Y-%m-%d %H:%M:%S',
|
|
),
|
|
)
|
|
|
|
stream_handler = logging.StreamHandler()
|
|
try:
|
|
import logging_color
|
|
|
|
stream_handler = logging_color.ColorStreamHandler()
|
|
except ImportError:
|
|
pass
|
|
|
|
logging.basicConfig(handlers=[stream_handler, file_handler])
|
|
|
|
logger.setLevel('INFO')
|
|
module_logger.setLevel('INFO')
|
|
|
|
|
|
CLI_DESCRIPTION = """
|
|
Sells financial assets from an online account.
|
|
""".strip()
|
|
|
|
|
|
def load_config(config_path: Path, path_trades_file: Path) -> AutoSellConfig:
|
|
logger.info('Loading configuration')
|
|
|
|
from . import secrets_config
|
|
|
|
seller_backend = fin_depo.defi_kucoin.KucoinDepoFetcher(
|
|
kucoin_secret=secrets_config.KUCOIN_SECRET,
|
|
kucoin_key=secrets_config.KUCOIN_KEY,
|
|
kucoin_pass=secrets_config.KUCOIN_PASS,
|
|
allow_trades=True,
|
|
)
|
|
|
|
with open(config_path) as f:
|
|
json_config = json.load(f)
|
|
|
|
return AutoSellConfig(
|
|
input_amount_range=(
|
|
Decimal(json_config['input_amount_low']),
|
|
Decimal(json_config['input_amount_high']),
|
|
),
|
|
interval_range=(
|
|
datetime.timedelta(minutes=json_config['interval_minutes_low']),
|
|
datetime.timedelta(minutes=json_config['interval_minutes_high']),
|
|
),
|
|
input_asset=fin_defs.WELL_KNOWN_SYMBOLS[json_config['input_asset']],
|
|
output_asset=fin_defs.WELL_KNOWN_SYMBOLS[json_config['output_asset']],
|
|
exit_when_empty=True,
|
|
seller=seller_backend,
|
|
sleep=time.sleep,
|
|
log_order_to_csv=order_csv.CsvFileLogger(path_trades_file),
|
|
)
|
|
|
|
|
|
def parse_args():
|
|
parser = argparse.ArgumentParser('crypto_seller', description=CLI_DESCRIPTION)
|
|
parser.add_argument('--config', type=Path, dest='config_file', required=True)
|
|
parser.add_argument('--output', type=Path, dest='output_folder', required=True)
|
|
return parser.parse_args()
|
|
|
|
|
|
def main():
|
|
"""Initializes the program."""
|
|
|
|
args = parse_args()
|
|
|
|
# Setup output paths
|
|
path_output = args.output_folder.absolute()
|
|
path_output.mkdir(parents=True, exist_ok=True)
|
|
path_log_file = path_output / 'log.txt'
|
|
path_trades_file = path_output / 'trades.csv'
|
|
setup_logging(path_log_file )
|
|
|
|
logger.info('Initializing crypto_seller')
|
|
|
|
auto_sell_config = load_config(args.config_file, path_trades_file)
|
|
results = run_auto_sell(auto_sell_config)
|
|
logging.info('Sell-offs complete')
|
|
log_results(results)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|