tradingbot_starter.py

import logging
import os
import sys
from logging import handlers

from strategy_type import StrategyFactory
from strategy_type.AbstractStrategy import AbstractStrategy
from trading_connector import TradingConnectorFactory
from trading_connector.AbstractTradingConnector import AbstractTradingConnector
from util.ColorFormatter import ColorFormatter, MethodLoggerFormatter, FileLoggerFormatter, CoinInfoLoggerFormatter

trading_connector_list = ["coin", "stock"]
strategy_list = ["SVB", "CMM"]


def logger_setting():
    logger.setLevel(logging.DEBUG)
    ch = logging.StreamHandler()
    ch.setLevel(logging.DEBUG)
    ch.setFormatter(ColorFormatter())
    logger.addHandler(ch)

    os.makedirs('./logs', exist_ok=True)
    ch = handlers.TimedRotatingFileHandler("./logs/trading.log", when="midnight", interval=1, encoding="UTF-8")
    ch.namer = lambda name: name.replace(".log", "") + ".log"
    ch.setFormatter(FileLoggerFormatter())
    ch.setLevel(logging.DEBUG)
    logger.addHandler(ch)

    method_logger.setLevel(logging.DEBUG)
    ch = logging.StreamHandler()
    ch.setLevel(logging.DEBUG)
    ch.setFormatter(MethodLoggerFormatter())
    method_logger.addHandler(ch)

    coin_info_logger.setLevel(logging.DEBUG)
    ch = logging.StreamHandler()
    ch.setLevel(logging.DEBUG)
    ch.setFormatter(CoinInfoLoggerFormatter())
    coin_info_logger.addHandler(ch)


def is_valid_strategy(strategy_name):
    if strategy_name in strategy_list:
        logger.info(f'strategy: "{strategy_name}"')
        return True
    else:
        logger.error(f'잘못된 strategy("{strategy_name}"). {strategy_list}에서 선택.')
        return False


def is_valid_trading(trading_name):
    if trading_name in trading_connector_list:
        logger.info(f'trading_type: "{trading_name}"')
        return True
    else:
        logger.error(f'잘못된 trading_connector("{trading_name}"). {trading_connector_list}에서 선택.')
        return False


if __name__ == "__main__":
    logger = logging.getLogger("SystemLogger")
    method_logger = logging.getLogger("MethodLogger")
    coin_info_logger = logging.getLogger("CoinInfoLogger")
    logger_setting()
    logger.info("NEW START")
    if len(sys.argv) != 3:
        logger.error("입력 인자 개수가 부족합니다. (사용예시: 'python tradingbot_starter.py coin CMM')")
        exit()
    trading_connector_type = sys.argv[1]
    strategy_type = sys.argv[2]
    if not is_valid_trading(trading_connector_type) or not is_valid_strategy(strategy_type):
        exit()
    trading_connector: AbstractTradingConnector = TradingConnectorFactory.create(trading_connector_type)
    strategy: AbstractStrategy = StrategyFactory.create(strategy_type, trading_connector)
    strategy.run_strategy()