49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
import abc
|
|
import os
|
|
import re
|
|
from typing import Union, Set, Dict
|
|
|
|
|
|
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[str] = set()
|
|
mimetypes: Set[str] = set()
|
|
|
|
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
|
|
|
|
:raises RuntimeError: Raised if the cleaning process went wrong.
|
|
"""
|
|
|
|
@abc.abstractmethod
|
|
def remove_all(self) -> bool:
|
|
"""
|
|
Remove all the metadata of the current file
|
|
|
|
:raises RuntimeError: Raised if the cleaning process went wrong.
|
|
"""
|