Replace libmat2 with modified version #10

Author: CalebUsadi

pyproject.toml

@@ -3,8 +3,4 @@ name = "metadata_remover"
 version = "1.6"
 description = "Tool for removing metadata from images and videos"
 license = {file = "LICENSE.txt"}
-readme = "README.md"
-
-dependencies = [
-    "mat2"
-]
+readme = "README.md"

src/metadata_remover/libmat2/__init__.py

@@ -0,0 +1,53 @@
+
+import enum
+import importlib
+from typing import Dict, Optional, Union
+
+# A set of extension that aren't supported, despite matching a supported mimetype
+UNSUPPORTED_EXTENSIONS = {
+    '.asc',
+    '.bat',
+    '.brf',
+    '.c',
+    '.h',
+    '.ksh',
+    '.pl',
+    '.pot',
+    '.rdf',
+    '.srt',
+    '.wsdl',
+    '.xpdl',
+    '.xsd',
+    '.xsl',
+    }
+
+DEPENDENCIES = {
+    'Mutagen': {
+        'module': 'mutagen',
+        'required': True,
+    },
+}
+
+
+def check_dependencies() -> Dict[str, Dict[str, bool]]:
+    ret = dict()  # type: Dict[str, dict]
+
+    for key, value in DEPENDENCIES.items():
+        ret[key] = {
+            'found': True,
+            'required': value['required'],
+        }
+        try:
+            importlib.import_module(value['module'])  # type: ignore
+        except ImportError:  # pragma: no cover
+            ret[key]['found'] = False
+
+
+    return ret
+
+
+@enum.unique
+class UnknownMemberPolicy(enum.Enum):
+    ABORT = 'abort'
+    OMIT = 'omit'
+    KEEP = 'keep'

src/metadata_remover/libmat2/abstract.py

@@ -0,0 +1,47 @@
+import abc
+import os
+import re
+from typing import Set, Dict, Union
+
+assert Set  # make pyflakes happy
+
+
+class AbstractParser(abc.ABC):
+    """ This is the base class of every parser.
+    It might yield `ValueError` on instantiation on invalid files,
+    and `RuntimeError` when something went wrong in `remove_all`.
+    """
+    meta_list = set()  # type: Set[str]
+    mimetypes = set()  # type: Set[str]
+
+    def __init__(self, filename: str) -> None:
+        """
+        :raises ValueError: Raised upon an invalid file
+        """
+        if re.search('^[a-z0-9./]', filename) is None:
+            # Some parsers are calling external binaries,
+            # this prevents shell command injections
+            filename = os.path.join('.', filename)
+
+        self.filename = filename
+        fname, extension = os.path.splitext(filename)
+
+        # Special case for tar.gz, tar.bz2, … files
+        if fname.endswith('.tar') and len(fname) > 4:
+            fname, extension = fname[:-4], '.tar' + extension
+
+        self.output_filename = fname + '.cleaned' + extension
+        self.lightweight_cleaning = False
+        self.sandbox = True
+
+    @abc.abstractmethod
+    def get_meta(self) -> Dict[str, Union[str, dict]]:
+        """Return all the metadata of the current file"""
+
+    @abc.abstractmethod
+    def remove_all(self) -> bool:
+        """
+        Remove all the metadata of the current file
+
+        :raises RuntimeError: Raised if the cleaning process went wrong.
+        """