From 6c29e0eae256ddb3d32fd78e9949d0aa81be033e Mon Sep 17 00:00:00 2001
From: jvoisin <julien.voisin@dustri.org>
Date: Sun, 1 Apr 2018 17:13:34 +0200
Subject: [PATCH] Improve a bit the main.py file
---
main.py | 13 +++++++++----
1 file changed, 9 insertions(+), 4 deletions(-)
diff --git a/main.py b/main.py
index daebc00..3cff758 100644
--- a/main.py
+++ b/main.py
@@ -1,4 +1,5 @@
import sys
+import mimetypes
from shutil import copyfile
import argparse
@@ -18,10 +19,11 @@ def create_arg_parser():
help='list all the harmful metadata of a file without removing them')
return parser
-def show_meta(file_name:str):
- p = parser_factory.get_parser(file_name)
+def show_meta(filename:str):
+ p = parser_factory.get_parser(filename)
if p is None:
- print("[-] %s's format (%s) is not supported" % (file_name, p))
+ mtype, _ = mimetypes.guess_type(filename)
+ print("[-] %s's format (%s) is not supported" % (filename, mtype))
return
for k,v in p.get_meta().items():
print("%s: %s" % (k, v))
@@ -36,7 +38,10 @@ def main():
return 0
for f in args.files:
- p = parser_factory.get_parser(sys.argv[1])
+ p = parser_factory.get_parser(f)
+ if p is None:
+ print("[-] %s's format (%s) is not supported" % (f, "meh"))
+ continue
p.remove_all()
--
GitLab