diff --git a/src/Main.py b/src/Main.py index 0177bf4..fbea8fd 100644 --- a/src/Main.py +++ b/src/Main.py @@ -19,6 +19,22 @@ import getpass ################# Fin Tips ################# +def insert_Country(connection, filename: str): + cur = connection.cursor() + df = pand.DataFrame(pand.read_csv(filename)).dropna() + + id_table=0 + for row in df.itertuples(): + for country in row.country_lastfm.split('; '): + id_table+=1 + cur.execute("INSERT INTO Country VALUES (%s, %s) ON CONFLICT ON CONSTRAINT doublons_country DO NOTHING;", + (id_table, + str(country)) + ) + + connection.commit() + cur.close() + def create_tables(connection, filename): cur = connection.cursor() with open(filename) as f: