loaders.py

# third parties imports
import pandas as pd
import os
from pprint import pprint as pp
# import display

# local imports
from constants import Constant as C


def load_ratings(surprise_format=False):
    df_ratings = pd.read_csv(C.EVIDENCE_PATH / C.RATINGS_FILENAME)
    if surprise_format:
        pass
    else:
        return df_ratings


def load_items():
    df_items = pd.read_csv(C.CONTENT_PATH / C.ITEMS_FILENAME) # ce qui se trouve dans le movie csv
    df_items = df_items.set_index(C.ITEM_ID_COL) # movie id
    
    return df_items


def export_evaluation_report():
    df_ratings = load_ratings()
    df_items = load_items()
    
    df_merge = pd.merge(df_ratings,df_items, on = 'movieId')

    report_name = f"evaluation_report_{pd.Timestamp.now().strftime('%Y-%m-%d')}.csv"
    export_path = os.path.join("data", "small", "evaluations", report_name)
    df_merge.to_csv(export_path, index=False)
    print("The data has been exported to the evaluation report")
    return df_merge

# Display
print("\n")
print("Display the ratings")
pp(load_ratings())
print("\n\n")
print("Display the movie data")
pp(load_items())
# display(load_items())# pas pratique dans .py vscode


# Call the function to register the report 
export_evaluation_report()