Source code for bbstrader.compat

import platform
import sys
from typing import Any




def setup_mock_modules() -> None:
    """Mock some modules not available on some OS to prevent import errors."""
    from unittest.mock import MagicMock

    class Mock(MagicMock):
        @classmethod
        def __getattr__(cls, name: str) -> Any:
            return MagicMock()

    MOCK_MODULES = []

    # Mock Metatrader5 on Linux and MacOS
    if platform.system() != "Windows":
        MOCK_MODULES.append("MetaTrader5")

    # Mock posix On windows
    if platform.system() == "Windows":
        MOCK_MODULES.append("posix")

    sys.modules.update((mod_name, Mock()) for mod_name in MOCK_MODULES)



setup_mock_modules()