Correct error for CSV

johnsamuelwrites · Mar 6, 2021 · a447b21 · a447b21
1 parent 90cda8c
commit a447b21
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/RELEASE.md b/RELEASE.md
@@ -1,4 +1,8 @@
-# v0.8 (in progress)
+# v0.9
+===============================================================================
+* Correct error related to handling of CSV files 
+
+# v0.8 
 ===============================================================================
 * Add support for spreadsheet files (.ods, .xls, .xlsx)
 * Apply pycodestyle (PEP8)

diff --git a/shexstatements/application.py b/shexstatements/application.py
@@ -28,11 +28,15 @@ def generateshex():
                 if ".csv" == file_extension.lower():
                     shex = CSV.generate_shex_from_csv(
                         shexstatements, delim=delim, filename=False)
-                else:
+                elif file_extension.lower() in {".ods", ".xls", ".xlsx"}:
                     shexstatements = request.files["csvfileupload"].stream.read(
                     )
                     shex = Spreadsheet.generate_shex_from_spreadsheet(
                         stream=shexstatements, filepath=filepath)
+                else:
+                    shex = CSV.generate_shex_from_csv(
+                        shexstatements, delim=delim, filename=False)
+
             data["input"] = shexstatements
             data["output"] = shex
             return render_template('shexstatements.html', data=data)