Merge pull request #5 from theTaikun/development

Error handling for CSV with invalid data

__init__.py

@@ -2,8 +2,8 @@
     "name": "PlotRock",
     "description": "Create 3D plots from CSV data",
     "author": "Isaac Phillips (theTaikun)",
-    "version": (0,0,1),
-    "version_code": 1, # not used by blender, but keeping track here
+    "version": (0,0,2),
+    "version_code": 2, # not used by blender, but keeping track here
     "blender": (2, 92, 0),
     "location": "File > Import > Import CSV for plotting",
     "tracker_url": "https://github.com/theTaikun/plotrock/issues/new",

plot.py

@@ -9,15 +9,26 @@ def convertData(csv_textdata, entry_delimiter=",", has_headers=True):
     print("converting data")
     raw_data = csv_textdata.as_string()
     reader = csv.reader(StringIO(raw_data), delimiter=entry_delimiter) # read csv string as csv file
+    print("READER: {}".format(reader))
     if(has_headers):
         headers = next(reader)
         print("headers: {}".format(headers))
     else:
         headers = None
     string_list = list(reader)
-    pos_list = [list(map(float, x)) for x in string_list] # convert list of strings to list of floats
+    print("string_list: {}".format(string_list))
+    pos_list = [i for i in sanitizeData(string_list)]
+    print("pos_list: {}".format(pos_list))
     return pos_list, headers
 
+def sanitizeData(data):
+    for i in data:
+        try:
+            x = list(map(float, i))
+            yield x
+        except:
+            pass
+
 # findRoot function courtesy of MMDTools addon
 def findRoot(obj):
     if obj:
@@ -50,6 +61,8 @@ def execute(self, **args):        # execute() is called when running the operato
         #self.delimiter = args.get("delimiter")
         self.pos_list, self.headers = convertData(self.csv_textdata, self.csv_textdata['delimiter'], self.has_headers)
 
+        if self.pos_list is None:
+            return {'FINISHED'}
 
         if self.obj is None:
             print("no class obj")
@@ -271,8 +284,9 @@ def execute(self, context):
 
         self.pos_list, self.headers = convertData(self.csv_textdata, self.delimiter, self.has_headers)
 
-        self.update_curve()
-        self.update_axis()
+        if self.pos_list is not None:
+            self.update_curve()
+            self.update_axis()
         return {"FINISHED"}