Source code for pipeline.src.entities.control_units

from dataclasses import dataclass
from datetime import datetime
from typing import List

import pandas as pd

from src.helpers.dates import Period


@dataclass

[docs]
class ControlUnit:

[docs]
    control_unit_id: int


[docs]
    control_unit_name: str


[docs]
    administration: str


[docs]
    emails: List[str]


[docs]
    phone_numbers: List[str]




@dataclass

[docs]
class ControlUnitActions:
    """
    Control unit and its fisheries control actions between two dates.
    """


[docs]
    control_unit: ControlUnit


[docs]
    period: Period


[docs]
    land_controls: pd.DataFrame


[docs]
    sea_controls: pd.DataFrame


[docs]
    air_controls: pd.DataFrame


[docs]
    air_surveillances: pd.DataFrame




@dataclass

[docs]
class ControlUnitActionsSentMessage:

[docs]
    control_unit_id: int


[docs]
    control_unit_name: str


[docs]
    email_address: str


[docs]
    sending_datetime_utc: datetime


[docs]
    actions_min_datetime_utc: datetime


[docs]
    actions_max_datetime_utc: datetime


[docs]
    number_of_actions: int


[docs]
    success: bool


[docs]
    error_code: int


[docs]
    error_message: str