security-and-privacy/mat2
security-and-privacy/mat2
1
0
Fork 0
mirror of synced 2024-11-25 10:44:23 +01:00
Code Releases Activity

Improve a bit the typing, again

Browse Source
This commit is contained in:
totallylegit 2018-06-04 20:39:27 +02:00
parent 8143b63ee3
commit 183667a7f9
3 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
libmat2/abstract.py
View File

@ -1,12 +1,13 @@
import abc import abc
import os import os
from typing import Set
class AbstractParser(abc.ABC): class AbstractParser(abc.ABC):
meta_list = set() meta_list = set() # type: Set[str]
mimetypes = set() mimetypes = set() # type: Set[str]
def __init__(self, filename: str): def __init__(self, filename: str) -> None:
self.filename = filename self.filename = filename
fname, extension = os.path.splitext(filename) fname, extension = os.path.splitext(filename)
self.output_filename = fname + '.cleaned' + extension self.output_filename = fname + '.cleaned' + extension

2
libmat2/harmless.py
View File

@ -5,7 +5,7 @@ class HarmlessParser(abstract.AbstractParser):
""" This is the parser for filetypes that do not contain metadata. """ """ This is the parser for filetypes that do not contain metadata. """
mimetypes = {'application/xml', 'text/plain'} mimetypes = {'application/xml', 'text/plain'}
def __init__(self, filename: str): def __init__(self, filename: str) -> None:
super().__init__(filename) super().__init__(filename)
self.filename = filename self.filename = filename
self.output_filename = filename self.output_filename = filename

2
libmat2/torrent.py
View File

@ -2,7 +2,7 @@ from . import abstract
class TorrentParser(abstract.AbstractParser): class TorrentParser(abstract.AbstractParser):
mimetypes = {'application/x-bittorrent', } mimetypes = {b'application/x-bittorrent', }
whitelist = {b'announce', b'announce-list', b'info'} whitelist = {b'announce', b'announce-list', b'info'}
def get_meta(self) -> dict: def get_meta(self) -> dict: