main.py

import sys
from Trading212 import Trading212
from DataProcessor import ProcessorV1
import os
from TradeAlgo import TradeAlgorithm
from MessageHandler import Messager
from dotenv import load_dotenv
import logging
import logging.config
import yaml
import time
import watchtower

load_dotenv()


def getData(ticker):
    connection = Trading212()
    responseData = connection.getTickerData(ticker, 'TEN_MINUTES')
    print(responseData)

    return


def generateData(tickers):
    processor = ProcessorV1(Trading212())
    processor.generateData(tickers, output=os.path.abspath(os.getcwd()) + '/storage/datasets/datasetV1.json')

    return


def getDecision(tickers):
    tradeAlgo = TradeAlgorithm(tickers)

    return tradeAlgo.getResult()


def testMessager():
    messageHandler = Messager()

    messageHandler.enqueue('BUY/EURUSD', 'MessageQueue-dev.fifo')
    messageHandler.enqueue('SELL/GBPCHF',  'MessageQueue-dev.fifo')
    messageHandler.enqueue('this is a debug message', 'MessageQueue-dev.fifo')

    return


def main():
    arguments = sys.argv

    if len(arguments) == 1:
        return

    match arguments[1]:
        case 'help':
            print('Enter an option to run that function, this was helpful.')
        case 'getData':
            return getData(arguments[2])
        case 'generateData':
            return generateData(arguments[2])
        case 'getDecisions':
            return getDecision(arguments[2].split(','))
        case 'testMessage':
            return testMessager()
        case _:
            logging.error(f'Option({arguments[1]}) not found.')


if __name__ == '__main__':
    with open('logging.yml') as log_config:
        config_yml = log_config.read()
        config_dict = yaml.safe_load(config_yml)
        logging.config.dictConfig(config_dict)

    logging.info('-----------Application Starting-----------')
    startTime = time.time()
    main()
    logging.info('-----------Application Finished-----------')
    logging.info(f'Runtime: {round(time.time() - startTime, 2)}s')
    logging.info('------------------------------------------')