main.py

import sys

from pydantic import ValidationError

from parser import Parser
from module import Graph, Drone
from simulation import Engine, UnsolvableGraphError


class Application:
    def __init__(self, argv: list[str]) -> None:
        """Store the command-line arguments.

        Args:
            argv: Command-line arguments passed to the application.
        """
        self.argv = argv

    def run(self) -> None:
        """Execute the application workflow.

        This loads the map, builds the graph, computes drone paths, and
        runs the simulation.
        """
        if len(self.argv) != 2:
            print("Usage: python main.py <file>")
            sys.exit(1)

        try:
            parser = Parser(self.argv[1])
            self.data = parser.load_map_data()

            self.graph = Graph()
            for zone in self.data.zone_list:
                self.graph.add_nodes(zone)

            for connection in self.data.connection_list:
                self.graph.add_edges(connection)

            engine = Engine(self.data, self.graph)

            start = self.data.zone_list[0]
            self.drones = [
                Drone(f"D{i + 1}", start,
                      engine.dijkstra(
                          start)[1:]
                      ) for i in range(self.data.nb_drones)
            ]

            engine.simulation(self.drones)

        except ValidationError as e:
            for error in e.errors():
                print(error["msg"])
            sys.exit(1)
        except ValueError as e:
            print(e)
            sys.exit(1)
        except UnsolvableGraphError as e:
            print(e)
            sys.exit(1)

    def test(self) -> None:
        """Print the planned path for each drone."""
        for d in self.drones:
            print(d.path)


if __name__ == "__main__":
    app = Application(sys.argv)

    app.run()