27 lines
749 B
Python
27 lines
749 B
Python
|
import yaml
|
||
|
|
from fastapi import FastAPI
|
||
|
|
|
||
|
|
from core.module import Module
|
||
|
|
|
||
|
|
|
||
|
|
class MaSH:
|
||
|
|
def __init__(self, config_path: str) -> None:
|
||
|
|
self.server: FastAPI = FastAPI()
|
||
|
|
self.config: dict = None
|
||
|
|
|
||
|
|
self._load_config_(config_path)
|
||
|
|
|
||
|
|
def _load_config_(self, config_path: str) -> None:
|
||
|
|
try:
|
||
|
|
with open(config_path, "r", encoding="UTF-8") as fp:
|
||
|
|
self.config = yaml.safe_load(fp)
|
||
|
|
except FileNotFoundError:
|
||
|
|
raise f"Config file for MaSH server could not be opened at [{config_path}]."
|
||
|
|
|
||
|
|
def add_module(self, module: Module) -> None:
|
||
|
|
module.add_routes(self.server)
|
||
|
|
self.server.include_router(module.get_router())
|
||
|
|
|
||
|
|
def run(self):
|
||
|
|
self.server.run()
|