utils.py

import json 
import logging
import mimetypes
import os
import requests
from urllib.parse import urlparse
from config import BASE_URL


def read_file_or_url(path):
    if path.startswith("http://") or path.startswith("https://"):
        response = requests.get(path)
        return response.content
    else:
        return open(path, "rb")
    
# used by "/forecastingapicall2"
def call_api(token: str, 
             apiroute: str,  
             filename: str,          
             method: str = 'Ridge',
             n_hidden_features: int = 5,
             lags: int = 20,
             type_pi: str = 'scp2-block-bootstrap',
             replications: int = 50,
             h: int = 10):    

    headers = {
    'Authorization': 'Bearer ' + token,
    }

    params = {'base_model': str(method), 
    'n_hidden_features': str(n_hidden_features),
    'lags': str(lags),
    'type_pi': str(type_pi),
    'replications': str(replications),
    'h': str(h)}

    # Prepare the file for uploading (file content, filename, and content_type)
    files = {
        'file': (filename, read_file_or_url(filename), 'text/csv')
    }

    try: 

        response = requests.post(BASE_URL + apiroute, 
            params=params, headers=headers, 
            files=files)        

        res = response.json()

        mean = res["mean"]
        lower = res["lower"]
        upper = res["upper"]
        sims = res["sims"]

        return mean, lower, upper, sims 

    except Exception as e:
        
        print(f"Error: {e}")    
        return None, None, None, None