main.py

import sqlite3

import pandas as pd


def load_to_db(table_name: str = "supply_chain_data.db", df: pd.DataFrame = None):
    conn = sqlite3.connect(table_name)
    df.to_sql(table_name, conn, if_exists="replace", index=False)
    conn.close()


def query_db(
    query: str = "SELECT * FROM table_name", table_name: str = "supply_chain_data.db"
):
    conn = sqlite3.connect(table_name)
    cursor = conn.cursor()
    cursor.execute(query)
    rows = cursor.fetchall()
    conn.close()

    return rows


def main():
    df_logistics = pd.read_csv("supply_chain_data.csv")
    print(df_logistics.columns)
    # print(df_logistics.head())

    df_tickets = pd.read_csv("customer_support_tickets.csv")
    print(df_tickets.columns)
    # print(df_tickets.head())

    load_to_db("supply_chain_data.db", df_logistics)
    load_to_db("customer_support_tickets.db", df_tickets)

    data = query_db(
        "SELECT DISTINCT SKU FROM supply_chain_data", "supply_chain_data.db"
    )
    print(data)

    return data


if __name__ == "__main__":
    main()