catch newly thrown ValueErrors on get_file_parserv0.7.0

2 files changed, 45 insertions, 39 deletions

diff --git a/matweb/frontend.py b/matweb/frontend.py
index d29a2fb..395606f 100644
--- a/matweb/frontend.py
+++ b/matweb/frontend.py
@@ -64,11 +64,11 @@ def upload_file():
             current_app.logger.error('Invalid Filename in upload')
             return redirect(request.url)
 
-        parser, mime = utils.get_file_parser(filepath)
-
-        if parser is None:
-            flash('The type %s is not supported' % mime)
-            current_app.logger.error('Unsupported type %s', mime)
+        try:
+            parser, mime = utils.get_file_parser(filepath)
+        except ValueError:
+            flash('The filetype is not supported')
+            current_app.logger.error('Unsupported filetype',)
             return redirect(url_for('routes.upload_file'))
 
         try:
diff --git a/matweb/rest_api.py b/matweb/rest_api.py
index c8edead..d44d838 100644
--- a/matweb/rest_api.py
+++ b/matweb/rest_api.py
@@ -41,14 +41,11 @@ class APIUpload(Resource):
             current_app.logger.error('Upload - Invalid file name')
             abort(400, message='Invalid Filename')
 
-        parser, mime = utils.get_file_parser(filepath)
-
-        if parser is None:
-            current_app.logger.error('Upload - Invalid mime type %s', mime)
-            abort(415, message='The type %s is not supported' % mime)
         try:
+            parser, mime = utils.get_file_parser(filepath)
             if not parser.remove_all():
-                raise ValueError()
+                current_app.logger.error('Upload - Cleaning failed with mime: %s', mime)
+                abort(400, message='Unable to clean %s' % mime)
             meta = parser.get_meta()
             key, secret, meta_after, output_filename = utils.cleanup(parser, filepath,
                                                                      current_app.config['UPLOAD_FOLDER'])
@@ -68,7 +65,10 @@ class APIUpload(Resource):
                     _external=True
                 )
             ), 201
-        except (ValueError, RuntimeError):
+        except ValueError:
+            current_app.logger.error('Upload - Invalid mime type')
+            abort(415, message='The filetype is not supported')
+        except RuntimeError:
             current_app.logger.error('Upload - Cleaning failed with mime: %s', mime)
             abort(400, message='Unable to clean %s' % mime)
 
@@ -107,18 +107,19 @@ class APIClean(Resource):
             current_app.logger.error('Clean - Invalid Filename')
             abort(400, message='Invalid Filename')
 
-        parser, mime = utils.get_file_parser(filepath)
-
-        if parser is None:
-            current_app.logger.error('Clean - The type %s is not supported', mime)
-            abort(415, message='The type %s is not supported' % mime)
-
-        if parser.remove_all() is not True:
+        try:
+            parser, mime = utils.get_file_parser(filepath)
+            if parser is None:
+                raise ValueError()
+            parser.remove_all()
+            _, _, _, output_filename = utils.cleanup(parser, filepath, current_app.config['UPLOAD_FOLDER'])
+        except ValueError:
+            current_app.logger.error('Upload - Invalid mime type')
+            abort(415, message='The filetype is not supported')
+        except RuntimeError:
             current_app.logger.error('Clean - Unable to clean %s', mime)
             abort(500, message='Unable to clean %s' % mime)
 
-        _, _, _, output_filename = utils.cleanup(parser, filepath, current_app.config['UPLOAD_FOLDER'])
-
         @after_this_request
         def remove_file(response):
             os.remove(os.path.join(current_app.config['UPLOAD_FOLDER'], output_filename))
@@ -180,26 +181,31 @@ class APIBulkDownloadCreator(Resource):
                 current_app.logger.error('BulkDownload -  Validating Zip failed: %s', e)
                 abort(400, message='Validating Zip failed')
 
-        parser, mime = utils.get_file_parser(zip_path)
-        if not parser.remove_all():
+        try:
+            parser, mime = utils.get_file_parser(zip_path)
+            parser.remove_all()
+            key, secret, meta_after, output_filename = utils.cleanup(parser, zip_path, current_app.config['UPLOAD_FOLDER'])
+            return {
+                       'inactive_after_sec': utils.get_file_removal_max_age_sec(),
+                       'output_filename': output_filename,
+                       'mime': mime,
+                       'key': key,
+                       'secret': secret,
+                       'meta_after': meta_after,
+                       'download_link': url_for(
+                           'api_bp.apidownload',
+                           key=key,
+                           secret=secret,
+                           filename=output_filename,
+                           _external=True
+                       )
+                   }, 201
+        except ValueError:
+            current_app.logger.error('BulkDownload - Invalid mime type')
+            abort(415, message='The filetype is not supported')
+        except RuntimeError:
             current_app.logger.error('BulkDownload -  Unable to clean Zip')
             abort(500, message='Unable to clean %s' % mime)
-        key, secret, meta_after, output_filename = utils.cleanup(parser, zip_path, current_app.config['UPLOAD_FOLDER'])
-        return {
-                   'inactive_after_sec': utils.get_file_removal_max_age_sec(),
-                   'output_filename': output_filename,
-                   'mime': mime,
-                   'key': key,
-                   'secret': secret,
-                   'meta_after': meta_after,
-                   'download_link': url_for(
-                       'api_bp.apidownload',
-                       key=key,
-                       secret=secret,
-                       filename=output_filename,
-                       _external=True
-                   )
-               }, 201
 
 
 class APISupportedExtensions(Resource):