Skip to content
Snippets Groups Projects
Commit 183667a7 authored by Julien (jvoisin) Voisin's avatar Julien (jvoisin) Voisin
Browse files

Improve a bit the typing, again

parent 8143b63e
No related branches found
No related tags found
No related merge requests found
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
......
...@@ -5,7 +5,7 @@ class HarmlessParser(abstract.AbstractParser): ...@@ -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,7 +2,7 @@ from . import abstract ...@@ -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:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment