Spec custom filetype for output

InseeFrLab · Sep 20, 2023 · 83e5a1a · 83e5a1a
1 parent 4560c78
commit 83e5a1a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/cartiflette/download/download.py b/cartiflette/download/download.py
@@ -757,6 +757,7 @@ def download_unzip(
 
             if "7-zip" in filetype:
                 file_locations = datafile.unzip(pattern, ext=ext)
+                filetype = "SHAPEFILE"
             elif "Unicode text" in filetype:
                 # copy in temp directory without processing
                 location = tempfile.mkdtemp()
@@ -770,9 +771,10 @@ def download_unzip(
                         out.write(f.read())
                 logger.debug(f"Storing CSV to {location}")
                 file_locations = (path,)
+                filetype = "CSV"
 
             else:
-                raise Exception(f"{filetype} encountered")
+                raise NotImplementedError(f"{filetype} encountered")
         except Exception as e:
             raise e
         finally: