Schemas/__init__.py

from .common import RouteContract
from .crop_simulation_current_farm_chart import CONTRACT as CROP_SIMULATION_CURRENT_FARM_CHART_CONTRACT
from .crop_simulation_growth import CONTRACT as CROP_SIMULATION_GROWTH_CONTRACT
from .crop_simulation_growth_status import CONTRACT as CROP_SIMULATION_GROWTH_STATUS_CONTRACT
from .crop_simulation_harvest_prediction import CONTRACT as CROP_SIMULATION_HARVEST_PREDICTION_CONTRACT
from .crop_simulation_yield_prediction import CONTRACT as CROP_SIMULATION_YIELD_PREDICTION_CONTRACT
from .economy_overview import CONTRACT as ECONOMY_OVERVIEW_CONTRACT
from .farm_alerts import CONTRACTS as FARM_ALERTS_CONTRACTS
from .farm_data_upsert import CONTRACT as FARM_DATA_UPSERT_CONTRACT
from .farm_detail import CONTRACT as FARM_DETAIL_CONTRACT
from .farm_parameter import CONTRACT as FARM_PARAMETER_CONTRACT
from .fertilization_recommend import CONTRACT as FERTILIZATION_RECOMMEND_CONTRACT
from .irrigation_methods import CONTRACTS as IRRIGATION_METHOD_CONTRACTS
from .irrigation_recommend import CONTRACT as IRRIGATION_RECOMMEND_CONTRACT
from .irrigation_water_stress import CONTRACT as IRRIGATION_WATER_STRESS_CONTRACT
from .pest_disease import CONTRACTS as PEST_DISEASE_CONTRACTS
from .plant import CONTRACTS as PLANT_CONTRACTS
from .rag_chat import CONTRACT as RAG_CHAT_CONTRACT
from .soil_data import CONTRACTS as SOIL_DATA_CONTRACTS
from .soile_anomaly_detection import CONTRACT as SOILE_ANOMALY_DETECTION_CONTRACT
from .soile_health_summary import CONTRACT as SOILE_HEALTH_SUMMARY_CONTRACT
from .soile_moisture_heatmap import CONTRACT as SOILE_MOISTURE_HEATMAP_CONTRACT
from .weather_farm_card import CONTRACT as WEATHER_FARM_CARD_CONTRACT
from .weather_water_need_prediction import CONTRACT as WEATHER_WATER_NEED_PREDICTION_CONTRACT

ALL_ROUTE_CONTRACTS: list[RouteContract] = [
    RAG_CHAT_CONTRACT,
    *FARM_ALERTS_CONTRACTS,
    *SOIL_DATA_CONTRACTS,
    SOILE_MOISTURE_HEATMAP_CONTRACT,
    SOILE_HEALTH_SUMMARY_CONTRACT,
    SOILE_ANOMALY_DETECTION_CONTRACT,
    FARM_DATA_UPSERT_CONTRACT,
    FARM_DETAIL_CONTRACT,
    FARM_PARAMETER_CONTRACT,
    WEATHER_FARM_CARD_CONTRACT,
    WEATHER_WATER_NEED_PREDICTION_CONTRACT,
    ECONOMY_OVERVIEW_CONTRACT,
    *PLANT_CONTRACTS,
    *PEST_DISEASE_CONTRACTS,
    *IRRIGATION_METHOD_CONTRACTS,
    IRRIGATION_RECOMMEND_CONTRACT,
    IRRIGATION_WATER_STRESS_CONTRACT,
    FERTILIZATION_RECOMMEND_CONTRACT,
    CROP_SIMULATION_GROWTH_CONTRACT,
    CROP_SIMULATION_GROWTH_STATUS_CONTRACT,
    CROP_SIMULATION_CURRENT_FARM_CHART_CONTRACT,
    CROP_SIMULATION_HARVEST_PREDICTION_CONTRACT,
    CROP_SIMULATION_YIELD_PREDICTION_CONTRACT,
]

ROUTE_CONTRACTS: dict[str, RouteContract] = {
    f'{contract.method} {contract.path}': contract
    for contract in ALL_ROUTE_CONTRACTS
}

__all__ = ['ALL_ROUTE_CONTRACTS', 'ROUTE_CONTRACTS', 'RouteContract']