diff options
Diffstat (limited to '')
| -rw-r--r-- | libmat2/abstract.py | 7 | ||||
| -rw-r--r-- | libmat2/harmless.py | 2 | ||||
| -rw-r--r-- | libmat2/torrent.py | 2 |
3 files changed, 6 insertions, 5 deletions
diff --git a/libmat2/abstract.py b/libmat2/abstract.py index e4838a9..9cb9320 100644 --- a/libmat2/abstract.py +++ b/libmat2/abstract.py | |||
| @@ -1,12 +1,13 @@ | |||
| 1 | import abc | 1 | import abc |
| 2 | import os | 2 | import os |
| 3 | from typing import Set | ||
| 3 | 4 | ||
| 4 | 5 | ||
| 5 | class AbstractParser(abc.ABC): | 6 | class AbstractParser(abc.ABC): |
| 6 | meta_list = set() | 7 | meta_list = set() # type: Set[str] |
| 7 | mimetypes = set() | 8 | mimetypes = set() # type: Set[str] |
| 8 | 9 | ||
| 9 | def __init__(self, filename: str): | 10 | def __init__(self, filename: str) -> None: |
| 10 | self.filename = filename | 11 | self.filename = filename |
| 11 | fname, extension = os.path.splitext(filename) | 12 | fname, extension = os.path.splitext(filename) |
| 12 | self.output_filename = fname + '.cleaned' + extension | 13 | self.output_filename = fname + '.cleaned' + extension |
diff --git a/libmat2/harmless.py b/libmat2/harmless.py index aa00582..a63407f 100644 --- a/libmat2/harmless.py +++ b/libmat2/harmless.py | |||
| @@ -5,7 +5,7 @@ class HarmlessParser(abstract.AbstractParser): | |||
| 5 | """ This is the parser for filetypes that do not contain metadata. """ | 5 | """ This is the parser for filetypes that do not contain metadata. """ |
| 6 | mimetypes = {'application/xml', 'text/plain'} | 6 | mimetypes = {'application/xml', 'text/plain'} |
| 7 | 7 | ||
| 8 | def __init__(self, filename: str): | 8 | def __init__(self, filename: str) -> None: |
| 9 | super().__init__(filename) | 9 | super().__init__(filename) |
| 10 | self.filename = filename | 10 | self.filename = filename |
| 11 | self.output_filename = filename | 11 | self.output_filename = filename |
diff --git a/libmat2/torrent.py b/libmat2/torrent.py index cb4b5e3..3a819fb 100644 --- a/libmat2/torrent.py +++ b/libmat2/torrent.py | |||
| @@ -2,7 +2,7 @@ from . import abstract | |||
| 2 | 2 | ||
| 3 | 3 | ||
| 4 | class TorrentParser(abstract.AbstractParser): | 4 | class TorrentParser(abstract.AbstractParser): |
| 5 | mimetypes = {'application/x-bittorrent', } | 5 | mimetypes = {b'application/x-bittorrent', } |
| 6 | whitelist = {b'announce', b'announce-list', b'info'} | 6 | whitelist = {b'announce', b'announce-list', b'info'} |
| 7 | 7 | ||
| 8 | def get_meta(self) -> dict: | 8 | def get_meta(self) -> dict: |
