fastapi snippets

service_protocol.py

from typing import Protocol, Generator
from common.logger.protocol import LoggerProtocol

from common.db.db import GeneratorCMDBSession
from common.db.eq import EventQueue, EQType, EQEvent, EQEventEnum
from scheduler.dto import *


class SchedulerServiceProtocol(Protocol):
    log: LoggerProtocol
    get_cdb: GeneratorCMDBSession
    eq: EQType

    def __init__(self, log: LoggerProtocol, get_cdb: GeneratorCMDBSession, eq: EQType): ...
    def stop(self): ...

    def toggle_task(self, task_id: int, enable: bool): ...
    def get_state(self) -> SchedulerStatsDTO: ...
    def update_state(self, new_state: SchedulerStatsDTO): ...


GeneratorSchedulerService = Generator[SchedulerServiceProtocol, None, None]