SQLAlchemy Load CSV Files

gistfile1.py

# from MODELS_MODULE import MODEL
import pandas as pd

CSV_FOLDER = 'csvs'


def load_csv(file, model_name):
    df = pd.read_csv('{}/{}.csv'.format(CSV_FOLDER, file))
    csv = df.where((pd.notnull(df)), None)  # replace Nan with None
    for row in csv.itertuples(index=False):
        model_class = getattr(sys.modules[__name__], model_name)
        model = model_class(**row._asdict())
        model.save()
        
"""
Usage:
tables = {'Model': 'table_name'}
for model_name, table in tables.items():
    load_csv(table, model_name)
"""