Mise à jour de 'src/Main.py'

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: