SPACEX FALCON 9 – esempio di previsione del successo o meno dell’atterraggio del primo stadio sulla base di alcuni dati di input – I parte –

ENDPOINTS = {
    'launches': 'https://api.spacexdata.com/v5/launches/past',
    'launchpads': 'https://api.spacexdata.com/v4/launchpads',
    'payloads': 'https://api.spacexdata.com/v4/payloads',
    'rockets': 'https://api.spacexdata.com/v4/rockets'
}

import requests
import sqlite3
import pandas as pd
import json

# Definiamo gli endpoint
ENDPOINTS = {
    'launches': 'https://api.spacexdata.com/v5/launches/past',
    'launchpads': 'https://api.spacexdata.com/v4/launchpads',
    'payloads': 'https://api.spacexdata.com/v4/payloads',
    'rockets': 'https://api.spacexdata.com/v4/rockets'
}

def Get_Data_Endpoint(endpoint_url):
    """
    Estrae i dati dall'endpoint specificato e li restituisce come un DataFrame pandas.
    """
    response = requests.get(endpoint_url)
    response.raise_for_status()  # Genera un'eccezione se la richiesta non ha avuto successo
    data = response.json()
    return pd.DataFrame(data)
def Converti_Tipo_Data(dataframe):
    """
    Converte i tipi di dati non supportati in SQLite (ad es. liste, dizionari) in stringhe JSON.
    """
    for col in dataframe.columns:
        # Controlla se il tipo di dati della colonna è una lista o un dizionario
        if dataframe[col].apply(lambda x: isinstance(x, (list, dict))).any():
            dataframe[col] = dataframe[col].apply(json.dumps)
    return dataframe

def Save_Data_To_Sqlite(dataframe, table_name, db_name):
    """
    Salva un DataFrame pandas in una tabella SQLite specificata in un database specificato.
    """
    data = Converti_Tipo_Data(dataframe)
    # Crea una connessione al database (lo crea se non esiste)
    with sqlite3.connect(db_name) as conn:
        # Usa la connessione per salvare il DataFrame come tabella nel database
        data.to_sql(table_name, conn, if_exists='replace', index=False)  # Se la tabella esiste già, la sostituisce

def main():
    # Itera su ogni endpoint
    for table_name, endpoint_url in ENDPOINTS.items():
        print(f"Estrae dati da {endpoint_url}...")
        data = Get_Data_Endpoint(endpoint_url)
        print(f"Salva dati nella tabella {table_name}...")
        Save_Data_To_Sqlite(data, table_name, 'launches_def.sqlite3')
    print("Operazione completata con successo!")

if __name__ == "__main__":
    main()