security-and-privacy/mat2
security-and-privacy/mat2
1
0
Fork 0
mirror of synced 2024-11-22 09:14:23 +01:00
Code Releases Activity

Add more typing hints

Browse Source
This commit is contained in:
jvoisin 2018-04-02 19:11:59 +02:00
parent 6868f20065
commit 23bd22b305
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/parser_factory.py
View File

@ -4,6 +4,10 @@ import pkgutil
from . import abstract from . import abstract
from typing import Type, TypeVar
T = TypeVar('T', bound='abstract.AbstractParser')
for module_loader, name, ispkg in pkgutil.walk_packages('.src'): for module_loader, name, ispkg in pkgutil.walk_packages('.src'):
if not name.startswith('src.'): if not name.startswith('src.'):
continue continue
@ -11,10 +15,9 @@ for module_loader, name, ispkg in pkgutil.walk_packages('.src'):
continue continue
importlib.import_module(name) importlib.import_module(name)
def get_parser(filename: str): def get_parser(filename: str) -> (T, str):
mtype, _ = mimetypes.guess_type(filename) mtype, _ = mimetypes.guess_type(filename)
for c in abstract.AbstractParser.__subclasses__(): for c in abstract.AbstractParser.__subclasses__():
if mtype in c.mimetypes: if mtype in c.mimetypes:
return c(filename), mtype return c(filename), mtype
print('factory: %s is not supported' % mtype)
return None, mtype return None, mtype