o #jm9@sdZddlmZddlmZmZmZddlmZmZddl m Z m Z m Z ddl mZmZmZmZmZmZmZddlmZmZddlmZdd lmZdd lmZdd lmZed e Z!ed Z"e!j#d dgdeedddddZ$ e!j#ddgdddZ%e!j#ddgdddZ&e!j#ddgdddZ'e!j#ddgdedd Z(e!j#d!dgdej)ed"d#Z*e!j#d$dgdd%d&Z+e!j#d'dgdej)e ed(d)Z,e!j#d*dgdej)e ed+d,Z-e!j#d-dgdej)e ed.d/Z.e!j#d0dgdej)d1d2Z/e!j#d3dgdej)d4d5Z0e!j#d0dgdej)d6d7Z1e!j#d8dgdej)e ed9d:Z2e!j#d;dgdej)ddgdej)d?d@Z4e!j#dAdgdej)dBdCZ5e!j#dDdgdej)dEdFZ6e!j#dGdgdej)e edHdIZ7e!j#dJdgdej)e edKdLZ8e!j#dMdgdej)e edNdOZ9e!j#dPdgdej)e edQdRZ:e!j#dSdgdej)dTdUZ;e!j#dVdgdej)dWdXZd[d\d]Z?e!>d^d_d`Z@e!>eAdadbZBdcS)dz # File: pyapi/app/routes/api.py # Description: API routes for the OTT Reports service, including caching, validation, and error handling )datetime) Blueprintjsonifyrequest) get_loggerperformance_log) validate_jsonvalidate_queryvalidate_multiple)ProphetTrainRequestProphetForecastRequestProphetPlotRequestProphetValidateRequestRegressionTrainRequestCollectionRiskScoreRequestSalesTargetPredictRequest) CacheServicecache_response)auth_middleware)ProphetController)ReportController)RegressionControllerapiz/healthGET)methods health_check<)timeoutc Csz[dtttd}t}|}t|dr ||d<n2t|dr+|j |d<n't |t rNt |dkrN|d}t|drF||d<n dd d |d<n||d<t |}d |_|WSty}z#td t|t d t|ttd}d|_|WYd}~Sd}~ww)z+Health check endpoint with cache statisticshealthy)status timestampcacheget_json databasesjsonrunknownz(Unable to parse database health response)rerrorzHealth check failed: unhealthy)rr'r N)strrnowr get_statsrget_database_healthhasattrr"r$ isinstancetuplelenr status_code Exceptionloggerr') health_datareport_controllerdb_health_response response_dataresponseer<,/var/www/html/oms-prod-mcp/app/routes/api.pyr7s>      z/dataPOSTcCs tj} td|d}d|_|S)Nz Data created)messagedata)rr$rr3)r@r:r<r<r= create_dataus rBz/data/cCst|dd}d|_|S)z Endpoint to retrieve data by ID This is a placeholder for demonstration purposes. In a real application, you would retrieve data from a database or service. z Sample data)data_idr@r(rr3rCr:r<r<r=get_datarFDELETEcCstd|d}d|_|S)z Endpoint to delete data by ID This is a placeholder for demonstration purposes. In a real application, you would delete data from a database or service. z Data deleted)r?rCrDrEr<r<r= delete_datarGrJz /cache/statsc Cszt}td|ttd}d|_|WSty@}zt dt|tdt|d}d|_|WYd}~Sd}~ww) zGet cache statisticssuccess)rr@r r(zError getting cache stats: r'rr?r*N) rr-rr+rr,r3r4r5r')statsr:r;r<r<r=get_cache_statssrNz /cache/clearc Cszt}t|r dnd|rdndttd}d|_|WStyH}zt dt|tdt|d }d |_|WYd }~Sd }~ww) z Clear all cache (admin endpoint)rKwarningzCache cleared successfullyzCache clear may have failed)rr?r r(zError clearing cache: r'rLr*N) rclearrr+rr,r3r4r5r')rKr:r;r<r<r= clear_caches&   rQz/reports/database/healthcCt}|S)z"Database connectivity health check)rr. controllerr<r<r=get_reports_database_healthsrUz/prophet/traincCt}||S)z!Train a Prophet forecasting model)r train_modelvalidated_jsonrTr<r<r=train_prophet_model rZz/prophet/forecast/cCt}|||S)z/Generate forecast using a trained Prophet model)rgenerate_forecastmodel_idrYrTr<r<r=generate_prophet_forecast r`z/prophet/plot/cCr\)z2Generate forecast plot for a trained Prophet model)rget_forecast_plotr^r<r<r=generate_prophet_plotrarcz/prophet/models/cCrV)z-Get information about a trained Prophet model)rget_model_infor_rTr<r<r=get_prophet_model_info rfz/prophet/modelscCrR)zList all trained Prophet models)r list_modelsrSr<r<r=list_prophet_modelsricCrV)zDelete a trained Prophet model)r delete_modelrer<r<r=delete_prophet_modelrgrlz/prophet/validatecCrV)z,Validate time series data format for Prophet)r validate_datarXr<r<r=validate_prophet_datar[rnz/prophet/train/dbcCs t}tjddp i}||S)z2Train a Prophet model using data from OMS databaseT)silent)rrr"train_from_database)rTr@r<r<r=train_prophet_from_databases rqz/prophet/tablescCrR)z1Get available OMS tables for time series analysis)rget_available_tablesrSr<r<r=get_prophet_tablesrjrsz#/prophet/tables//previewcCst}tj}|||S)z,Get preview of time series data from a table)rrargsto_dictget_data_preview) table_keyrTparamsr<r<r=get_prophet_data_preview%s  ryz$/prophet/tables//validatecCrV)zValidate a table configuration)rvalidate_table_config)rwrTr<r<r=validate_prophet_table.rgr{z!/regression/train/collection-riskcCrV)z6Train XGBoost collection risk classifier for a company)rtrain_collection_riskrXr<r<r=train_collection_risk_model:r[r}z/regression/train/sales-targetcCrV)z=Train XGBoost sales target attainment regressor for a company)rtrain_sales_targetrXr<r<r=train_sales_target_modelCr[rz!/regression/score/collection-riskcCrV)z+Score clients for payment delay probability)rscore_collection_riskrXr<r<r=rLr[rz /regression/predict/sales-targetcCrV)u<Predict attainment % for district×product_head combinations)rpredict_sales_targetrXr<r<r=rUr[rz/regression/modelscCrR)z1List all trained regression/classification models)rrhrSr<r<r=list_regression_models^rjrz/regression/models/cCrV)z(Get info for a specific regression model)rrdrer<r<r=get_regression_model_infofrgrcCrV)zDelete a regression model)rrkrer<r<r=delete_regression_modelnrgrcCs(tdtjtddi}d|_|S)Nz404 error for path: r'zEndpoint not foundr)r5rOrpathrr3r'r:r<r<r= not_foundvs rcCs0tdtjdtjtddi}d|_|S)Nz405 error for  r'zMethod not allowedr)r5rOrmethodrrr3rr<r<r=method_not_allowed~s rcCs,tdt|tddd}d|_|S)zECatch any unhandled exceptions in API routes and ensure JSON responsezUnhandled API exception: zInternal server errorzAn unexpected error occurred)r'r?r*)r5r'r+rr3rr<r<r=handle_api_exceptionsrN)C__doc__rflaskrrrapp.utils.helpersrrapp.decorators.validationrr r app.schemas.request_schemasr r r rrrrapp.services.cache_servicerrapp.middleware.auth_middlewarer"app.controllers.prophet_controllerr!app.controllers.report_controllerr%app.controllers.regression_controllerr__name__api_bpr5routerrBrFrJrNrequire_api_keyrQrUrZr`rcrfrirlrnrqrsryr{r}rrrrrr errorhandlerrrr4rr<r<r<r=s $      0