CryptoAutotrader/run.py at main · FunnyRabbitIsAHabbit/CryptoAutotrader · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
"""
Main bot module

@Developer: Stan
"""

# Python default library ---------
import argparse
import sys
from os.path import abspath, dirname, join
from typing import Any, Callable
# --------------------------------

# Own modules --------------------
from config import TestData
from output_integration import OutputIntegration
from predict import PredictionApp
from trading_bot import TradingBot
# --------------------------------


def global_main() -> None:
    """
    Function to call for main run logic

    :return: None
    """

    console_arguments_parser = argparse.ArgumentParser(
        prog="run.py",
        description="run.py will place trades in accordance with specified parameters. "
                    "Use with `test` command to only run default data through prediction API; "
                    "use with `run` command to run main functionality.",
        epilog="Extremely caution is advised, don't run the program unless knowing EXACTLY what will happen."
    )
    default_main_environment_filename = "main.env"
    default_prediction_environment_filename = "probability_llm.env"
    subparsers = console_arguments_parser.add_subparsers(
        dest="running_mode",
        required=True
    )
    parser_test_predict_api = subparsers.add_parser("test")
    parser_test_predict_api.add_argument(
        "-p", "--predictions",
        default=default_prediction_environment_filename,
        type=str,
        required=False,
    )
    parser_run = subparsers.add_parser("run")
    parser_run.add_argument(
        "-e", "--env",
        default=default_main_environment_filename,
        type=str,
        required=False,
    )
    parser_run.add_argument(
        "-p", "--predictions",
        default=default_prediction_environment_filename,
        type=str,
        required=False,
    )
    # New argument for base-output mode:
    parser_run.add_argument(
        "-b", "--base",
        action="store_true",
        help="Launch the app in base output mode (NOT IMPLEMENTED)"
    )

    console = console_arguments_parser.parse_args()
    mode = console.running_mode

    # Paths
    current_path: str = dirname(abspath(__file__))
    predictions_env_path: str  = join(current_path, console.predictions)

    # Predictions
    prediction_app: PredictionApp = PredictionApp(env_file_path=predictions_env_path)
    prediction_function: Callable[[Any], str] = prediction_app.predict_up_or_down

    match mode:
        case "run":
            main_trading_env_path = join(current_path, console.env)

            # If -b or --base flags have been used to run the script
            if console.base:
                print("[START]\tRunning in `run` mode with altered base output logic.")
                trading_bot: TradingBot = TradingBot(
                    prediction_api=prediction_function,
                    output_integration=OutputIntegration("base"),
                    env_file_path = main_trading_env_path
                )

            # No -b or --base flag has been given
            else:
                print("[START]\tStarted module in `run` mode (console).")
                trading_bot: TradingBot = TradingBot(
                    prediction_api=prediction_function,
                    output_integration=OutputIntegration("console"),
                    env_file_path=main_trading_env_path
                )

            # Regardless of usage of -d or --dashboard flags
            sys.exit(trading_bot.main(infinite_loop_condition=True))

        case "test":
            print("[START]\tSTARTED module in `test` mode.")
            print("\t[INFO]\tUptrend recognized ?",
                  prediction_function(TestData.DEFAULT_DATA_TO_TEST_API_UP))
            print("\t[INFO]\tDowntrend recognized ?",
                  prediction_function(TestData.DEFAULT_DATA_TO_TEST_API_DOWN))
            sys.exit(print("END] Test mode exited."))


if __name__ == "__main__":
    global_main()