34 lines
894 B
Python
34 lines
894 B
Python
"""collection of configured data pipelines, intended to be invoked from C#"""
|
|
|
|
from datetime import datetime as Datetime
|
|
|
|
from delta_barth.analysis import forecast
|
|
from delta_barth.management import SESSION
|
|
from delta_barth.types import JsonExportResponse
|
|
|
|
|
|
def pipeline_sales_forecast(
|
|
company_id: int | None,
|
|
start_date: Datetime | None,
|
|
) -> JsonExportResponse:
|
|
result = forecast.pipeline_sales_forecast(
|
|
SESSION, company_id=company_id, start_date=start_date
|
|
)
|
|
export = JsonExportResponse(result.model_dump_json())
|
|
|
|
return export
|
|
|
|
|
|
def pipeline_sales_forecast_dummy(
|
|
company_id: int | None,
|
|
start_date: Datetime | None,
|
|
) -> JsonExportResponse:
|
|
result = forecast.pipeline_sales_dummy(
|
|
SESSION,
|
|
company_id=company_id,
|
|
start_date=start_date,
|
|
)
|
|
export = JsonExportResponse(result.model_dump_json())
|
|
|
|
return export
|