FinancialMarketPython/main.py at master · gabrielpalassi/FinancialMarketPython · 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
"""
FinancialMarketPython - Main Entry Point

A collection of financial analysis tools for portfolio management,
risk assessment, strategy backtesting, and market data analysis.
"""

import sys
import subprocess
from pathlib import Path
import os

# Add the src directory to the path to import utils
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "src")))
from utils import print_with_separator


def clear_screen():
    """Clear the terminal screen."""
    os.system("cls" if os.name == "nt" else "clear")


def display_menu():
    """Display the main menu with all available tools."""
    clear_screen()
    print_with_separator(
        [
            "Portfolio Analysis",
            "   1. Portfolio Backtest",
            "   2. Markowitz Portfolio Optimization",
            "   3. Drawdown Calculator",
            "   4. Value at Risk (VaR) Calculator",
            "Strategy Backtests",
            "   5. Last Month Performance (LMP) Strategy Backtest",
            "   6. Moving Average (MA) Strategy Backtest",
            "   7. Bollinger Bands Backtest",
            "Market Data",
            "   8. Brazilian Central Bank Historical Data",
            "   9. Brazilian Central Bank Market Expectations",
            "Other",
            "   0. Exit",
        ]
    )


def run_script(script_path: Path):
    """Run a Python script and wait for it to complete."""
    try:
        subprocess.run([sys.executable, str(script_path)], check=True)
    except subprocess.CalledProcessError as e:
        print(f"Error running script: {e}")
    except KeyboardInterrupt:
        print("Script interrupted by user.")


def get_script_path(category: str, script_name: str) -> Path:
    """Get the full path to a script."""
    base_path = Path(__file__).parent / "src" / category
    return base_path / script_name


def main():
    """Main application loop."""
    scripts = {
        "1": ("portfolio_analysis", "portfolio_backtest.py", "Portfolio Backtest"),
        "2": ("portfolio_analysis", "markowitz_optimization.py", "Markowitz Optimization"),
        "3": ("portfolio_analysis", "drawdown.py", "Drawdown Calculator"),
        "4": ("portfolio_analysis", "value_at_risk.py", "Value at Risk"),
        "5": ("strategy_backtests", "lmp_strategy.py", "LMP Strategy Backtest"),
        "6": ("strategy_backtests", "ma_strategy.py", "MA Strategy Backtest"),
        "7": ("strategy_backtests", "bollinger_bands_strategy.py", "Bollinger Bands Backtest"),
        "8": ("market_data", "bcb_historical_data.py", "BCB Historical Data"),
        "9": ("market_data", "bcb_market_expectations.py", "BCB Market Expectations"),
    }

    while True:
        display_menu()
        choice = input("Select an option (0-9): ").strip()

        if choice == "0":
            clear_screen()
            sys.exit(0)

        if choice in scripts:
            category, script_name, display_name = scripts[choice]
            script_path = get_script_path(category, script_name)

            if not script_path.exists():
                print(f"Script not found at {script_path}")
                input("Press Enter to continue...")
                continue

            clear_screen()
            print(f"Running {display_name}...")

            run_script(script_path)
            input("Press Enter to return to menu...")
        else:
            print("Invalid option. Please select a number between 0 and 8.")
            input("Press Enter to continue...")


if __name__ == "__main__":
    try:
        main()
    except KeyboardInterrupt:
        clear_screen()
        sys.exit(0)