cloud_scope.py

import argparse

import toml

from src.api_throttle import APIThrottle
from src.config import Config
from src.monitor import Monitor

if __name__ == "__main__":
    # Parse command line arguments
    parser = argparse.ArgumentParser(description="""
    Monitors the availability of Lambda Labs' cloud instances 
    with customizable monitoring for specific instance types and regions. 
    Features configurable alerts via email or email-to-text for newly available instances. 
    Additionally, it can launch new instances upon availability and execute predefined scripts.
    """)
    parser.add_argument("config_file", help="Path to the configuration file.")
    parser.add_argument("namespace", help="TOML configuration namespace.")
    args = parser.parse_args()

    # Attempt to load the TOML configuration file
    try:
        with open(args.config_file, "r") as file:
            config_data = toml.load(file)
    except FileNotFoundError:
        raise FileNotFoundError(f"The configuration file '{args.config_file}' was not found.")

    # Ensure the specified namespace exists
    if args.namespace not in config_data:
        raise ValueError(f"Namespace '{args.namespace}' not found in the configuration file.")

    # Get the configuration for the specified namespace
    namespace_config = config_data[args.namespace]

    # Create an instance of the Config class, which also sets up logging
    config = Config(**namespace_config)

    # Create an instance of the Monitor class
    monitor = Monitor(config=config)

    # Create an instance of the APIThrottle class
    api_throttle = APIThrottle(request_interval_ms=config.min_poll_delay_ms)

    # Start monitoring instance availability with wait intervals
    while True:
        monitor.poll()
        api_throttle.wait_for_next_request()