Implement mimetype detection

3 files changed, 18 insertions, 6 deletions

diff --git a/src/parser_factory.py b/src/parser_factory.py
new file mode 100644
index 0000000..a93595a
--- /dev/null
+++ b/src/parser_factory.py
@@ -0,0 +1,10 @@
+import mimetypes
+
+from .parsers import abstract
+from .parsers import *
+
+def get_parser(filename: str):
+    mtype, _ = mimetypes.guess_type(filename)
+    for c in abstract.AbstractParser.__subclasses__():
+        if mtype in c.mimetypes:
+            return c(filename)
diff --git a/src/parsers/abstract.py b/src/parsers/abstract.py
index d0e7108..80bb812 100644
--- a/src/parsers/abstract.py
+++ b/src/parsers/abstract.py
@@ -3,6 +3,7 @@ class AbstractParser(object):
         self.filename = filename
         self.output_filename = filename + '.cleaned'
         self.meta_list = set()
+        self.mimetypes = set()
 
     def get_meta(self):
         raise NotImplementedError
diff --git a/src/parsers/pdf.py b/src/parsers/pdf.py
index 26985c6..e7bd00d 100644
--- a/src/parsers/pdf.py
+++ b/src/parsers/pdf.py
@@ -20,13 +20,14 @@ logging.basicConfig(level=logging.DEBUG)
 
 
 class PDFParser(abstract.AbstractParser):
+    mimetypes = {'application/pdf', }
+    meta_list = {'author', 'creation-date', 'creator', 'format', 'keywords',
+            'metadata', 'mod-date', 'producer', 'subject', 'title',
+            'viewer-preferences'}
+
     def __init__(self, filename):
         super().__init__(filename)
-        self.meta_list = {'author', 'creation-date', 'creator', 'format', 'keywords',
-                'metadata', 'mod-date', 'producer', 'subject', 'title',
-                'viewer-preferences'}
         self.uri = 'file://' + os.path.abspath(self.filename)
-        self.password = None
 
     def remove_all(self):
         """
@@ -35,7 +36,7 @@ class PDFParser(abstract.AbstractParser):
             PDF are removed via Poppler, because there is no way to tell
             cairo to not add "created by cairo" during rendering.
         """
-        document = Poppler.Document.new_from_file(self.uri, self.password)
+        document = Poppler.Document.new_from_file(self.uri, None)
         pages_count = document.get_n_pages()
 
         _, tmp_path = tempfile.mkstemp()
@@ -80,7 +81,7 @@ class PDFParser(abstract.AbstractParser):
         """ Return a dict with all the meta of the file
         """
         print("URI: %s", self.uri)
-        document = Poppler.Document.new_from_file(self.uri, self.password)
+        document = Poppler.Document.new_from_file(self.uri, None)
         metadata = {}
         for key in self.meta_list:
             if document.get_property(key):