Skip to content
Snippets Groups Projects
parser_factory.py 863 B
Newer Older
  • Learn to ignore specific revisions
  • import mimetypes
    
    import importlib
    import pkgutil
    
    from . import abstract
    
    from typing import TypeVar
    
    
    T = TypeVar('T', bound='abstract.AbstractParser')
    
    
    # This loads every parser in a dynamic way
    
    for module_loader, name, ispkg in pkgutil.walk_packages('.src'):
        if not name.startswith('src.'):
    
        elif name == 'src.abstract':
    
            continue
        importlib.import_module(name)
    
    def _get_parsers() -> list:
        """ Get all our parsers!"""
        def __get_parsers(cls):
            return cls.__subclasses__() + \
                [g for s in cls.__subclasses__() for g in __get_parsers(s)]
        return __get_parsers(abstract.AbstractParser)
    
    
    def get_parser(filename: str) -> (T, str):
    
        mtype, _ = mimetypes.guess_type(filename)
    
    
        for c in _get_parsers():
    
            if mtype in c.mimetypes:
    
                return c(filename), mtype
        return None, mtype