Saving new Variables in Data.csv

Author: dancelestini

Data.csv

@@ -1,8 +1,10 @@
+OR,Promoter,Driver,Transgene,Reporter
 Endogenous,OR42b,Gal4,Knockin,GCaMP7f
-DmOR1,OR47b,Boosted Gal4,Transgene,
-DmOR2,OR59b,,other,
-DmOR3,OR22ab,,,
-DmOR4,ORCO,,,
+Empty,OR47b,Boosted Gal4,Transgene,
+DmOR1,OR59b,,other,
+DmOR2,OR22ab,,,
+DmOR3,ORCO,,,
+DmOR4,Empty,,,
 DmOR5,,,,
 DmOR6,,,,
 DmOR7,,,,

Lumar_GUI.py

@@ -3,6 +3,7 @@
 from tkinter.ttk import *
 from tkcalendar import *
 from ttkwidgets.autocomplete import *
+import pandas as pd
 import re
 import csv
 
@@ -70,7 +71,27 @@ def __init__(self, master):
         self.master = master
 
         ### Variables for controlled parameters ###
-
+        
+        df = pd.read_csv("Data.csv")
+        
+        self.OR = list(df.OR[df.OR.notna()].values)
+        self.promotor = list(df.Promoter[df.Promoter.notna()].values)
+        self.KIKO = list(df.Driver[df.Driver.notna()].values)
+        self.replaced = list(df.Transgene[df.Transgene.notna()].values)
+        self.reporter = list(df.Reporter[df.Reporter.notna()].values)
+
+        # for row in csv_reader:
+        #     for i in range(len(row)):
+        #         if (i == 0) & (row[i] !=pd.NA): #ORs
+        #             self.OR.append(row[i])
+        #         elif (i == 1) & (row[i] != pd.NA): #Promoter
+        #              self.promotor.append(row[i])
+        #         elif (i == 2) & (row[i] != pd.NA): #Driver
+        #              self.replaced.append(row[i])
+        #         elif (i == 3) & (row[i] != pd.NA): #Transgene
+        #              self.KIKO.append(row[i])
+        #         elif (i == 4) & (row[i] != pd.NA): #Reporter
+        #              self.reporter.append(row[i])
 
 
 
@@ -80,12 +101,6 @@ def __init__(self, master):
 
         self.userDict = {"Nicolas": 1, "Yann": 2,
                          "Ilham": 3, "Ivan": 4, "Dan": 5,"Axel": 6}
-        self.Driver = ["OR42b", "OR47b", "OR59b", "OR22ab", "ORCO"]
-        self.promotor = ["OR42b", "OR47b", "OR59b", "OR22ab", "ORCO"]
-        self.replaced = ["Gal4","Boosted Gal4"]
-        self.KIKO = ["Knockin", "Transgene", "other"]
-        self.OR = ["Endogenous","Empty"] + ["DmOR%d" % id for id in range(1, 30)]
-        self.reporter = ["GCaMP7f"]
         self.T2A = ["F", "TB", "TA"]
         self.sex = ["M", "F"]
         self.Age = list(range(1, 10))
@@ -536,6 +551,22 @@ def createNewElement(self, Newmaster,category):
                     globals()[f"{i}"]["values"] = newlist
                     # set the current selection to the new OR created
                     globals()[f"{i}"].current((len(globals()[f"{i}"]["values"])-1))
+            
+            df = pd.read_csv("Data.csv")
+            NAs = list(df.loc[pd.isna(df[category]),:].index)
+            if len(NAs) == 0:
+                print(df.columns.get_loc(category))
+                col_index = df.columns.get_loc(category)
+                to_append_df = [pd.NA]*len(df.columns)
+                to_append_df[col_index] = value
+                print(to_append_df)
+                df = df.append(pd.Series(to_append_df, index=df.columns), ignore_index=True)
+                print(df.head)
+            else:
+                index = NAs[0]
+                df.loc[index,category] = value
+            
+            df.to_csv("Data.csv",index=False)
 
             Newmaster.destroy()  # quit the new window
         else:

Write_data.py

@@ -1,16 +1,18 @@
 import csv
 from itertools import zip_longest
+import pandas as pd
 
 with open ("Data.csv","w", encoding='UTF8', newline='') as f:
     writer = csv.writer(f,delimiter=",")
 
+    writer.writerow(["OR","Promoter","Driver","Transgene","Reporter"])
     OR = ["Endogenous","Empty"] + ["DmOR%d" % id for id in range(1, 30)]
     promotor = ["OR42b", "OR47b", "OR59b", "OR22ab", "ORCO"]
     replaced = ["Gal4","Boosted Gal4"]
     KIKO = ["Knockin", "Transgene", "other"]
     reporter = ["GCaMP7f"]
     d = [OR,promotor,replaced,KIKO,reporter]
-    rows = zip_longest(*d, fillvalue = '')
+    rows = zip_longest(*d, fillvalue = pd.NA)
     writer.writerows(rows)
 
 def load_csv(filename):
@@ -19,13 +21,26 @@ def load_csv(filename):
         file = open(filename,"r")
         # Reading file
         csv_reader = csv.reader(file)
-        first_line = True
+        OR = []
+        Promoter = []
+        Driver = []
+        Transgene = []
+        Reporter = []
+
         for row in csv_reader:
-            if first_line:
-                first_line=False
-                continue
-            odour[row[0]] = row[1]
-        
-        return odour
+            for i in range(len(row)):
+                if (i == 0) & (row[i] != ''): #ORs
+                    OR.append(row[i])
+                elif (i == 1) & (row[i] != ''): #Promoter
+                     Promoter.append(row[i])
+                elif (i == 2) & (row[i] != ''): #Driver
+                     Driver.append(row[i])
+                elif (i == 3) & (row[i] != ''): #Transgene
+                     Transgene.append(row[i])
+                elif (i == 4) & (row[i] != ''): #Reporter
+                     Reporter.append(row[i])    
+
+        return (OR,Promoter,Driver,Transgene,Reporter)
+
 
 print(load_csv("Data.csv"))