From effe68f08f19ff26c37ede5a7a92d5d4452434fe Mon Sep 17 00:00:00 2001
From: jvoisin
Date: Wed, 16 May 2018 22:00:37 +0200
Subject: The CLI shouldn't display unsupported file extensions

---
 main.py | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

(limited to 'main.py')

diff --git a/main.py b/main.py
index 0cda955..ab07641 100755
--- a/main.py
+++ b/main.py
@@ -8,7 +8,7 @@ import mimetypes
 import argparse
 import multiprocessing
 
-from src import parser_factory
+from src import parser_factory, unsupported_extensions
 
 __version__ = '0.1.0'
 
@@ -74,8 +74,15 @@ def show_parsers():
     print('[+] Supported formats:')
     for parser in parser_factory._get_parsers():
         for mtype in parser.mimetypes:
-            extensions = ', '.join(mimetypes.guess_all_extensions(mtype))
-            print('  - %s (%s)' % (mtype, extensions))
+            extensions = set()
+            for extension in mimetypes.guess_all_extensions(mtype):
+                if extension[1:] not in unsupported_extensions:  # skip the dot
+                    extensions.add(extension)
+            if not extensions:
+                # we're not supporting a single extension in the current
+                # mimetype, so there is not point in showing the mimetype at all
+                continue
+            print('  - %s (%s)' % (mtype, ', '.join(extensions)))
 
 
 def __get_files_recursively(files):
-- 
cgit v1.3