diff --git a/tg/config.py b/tg/config.py index c9ed8c9..aa9a794 100644 --- a/tg/config.py +++ b/tg/config.py @@ -5,7 +5,7 @@ overwritten by external config file import os import platform import runpy -from typing import Dict +from typing import Dict, Optional _os_name = platform.system() _darwin = "Darwin" @@ -15,7 +15,7 @@ _linux = "Linux" CONFIG_DIR = os.path.expanduser("~/.config/tg/") CONFIG_FILE = os.path.join(CONFIG_DIR, "conf.py") FILES_DIR = os.path.expanduser("~/.cache/tg/") -MAILCAP_FILE = None +MAILCAP_FILE: Optional[str] = None LOG_LEVEL = "INFO" LOG_PATH = os.path.expanduser("~/.local/share/tg/") diff --git a/tg/utils.py b/tg/utils.py index e8df9f2..3efd6bc 100644 --- a/tg/utils.py +++ b/tg/utils.py @@ -17,7 +17,7 @@ from functools import lru_cache from logging.handlers import RotatingFileHandler from subprocess import CompletedProcess from types import TracebackType -from typing import Any, Optional, Tuple, Type +from typing import Any, Dict, Optional, Tuple, Type from tg import config @@ -72,10 +72,10 @@ def get_mime(file_path: str) -> str: return mtype.split("/")[0] -def get_mailcap(): +def get_mailcap() -> Dict: if config.MAILCAP_FILE: with open(config.MAILCAP_FILE) as f: - return mailcap.readmailcapfile(f) + return mailcap.readmailcapfile(f) # type: ignore return mailcap.getcaps()