Correct handling of unhandled files

2 files changed, 5 insertions, 4 deletions

diff --git a/cli.py b/cli.py
index 2215f86..a19b8ce 100755
--- a/cli.py
+++ b/cli.py
@@ -92,7 +92,8 @@ def main():
 
     for filename in filenames:
         class_file = mat.create_class_file(filename, args.backup)
-        func(class_file, filename)
+        if class_file is not None:
+            func(class_file, filename)
 
 if __name__ == '__main__':
     main()
diff --git a/lib/mat.py b/lib/mat.py
index 858fbc9..508864b 100644
--- a/lib/mat.py
+++ b/lib/mat.py
@@ -61,8 +61,8 @@ def create_class_file(name, backup):
     filename = hachoir_core.cmd_line.unicodeFilename(name)
     parser = hachoir_parser.createParser(filename)
     if not parser:
-        print("Unable to parse the file %s with hachoir-parser." % filename)
-        sys.exit(1)
+        print("[+] Unable to parse %s" % filename)
+        return
 
     editor = hachoir_editor.createEditor(parser)
     try:
@@ -74,7 +74,7 @@ def create_class_file(name, backup):
         stripper_class = strippers[editor.input.__class__]
     except KeyError:
         #Place for another lib than hachoir
-        print("Don't have stripper for file type: %s" % editor.description)
+        print("[+] Don't have stripper for file type %s" % editor.description)
         sys.exit(1)
     if editor.input.__class__ == hachoir_parser.misc.PDFDocument:
         return stripper_class(filename, realname, backup)