From: Zsolt Dollenstein <zsol.zsol@gmail.com>
Date: Sun, 29 Apr 2018 07:50:08 +0000 (-0700)
Subject: fix type errors in setup.py (#179)
X-Git-Url: https://git.madduck.net/etc/vim.git/commitdiff_plain/188572118253f410eaf24d9945e369f744760f42

fix type errors in setup.py (#179)
---

diff --git a/setup.py b/setup.py
index 4e9a5c9..67a1e76 100644
--- a/setup.py
+++ b/setup.py
@@ -10,17 +10,18 @@ from pathlib import Path  # noqa E402
 CURRENT_DIR = Path(__file__).parent
 
 
-def get_long_description():
+def get_long_description() -> str:
     readme_md = CURRENT_DIR / "README.md"
     with open(readme_md, encoding="utf8") as ld_file:
         return ld_file.read()
 
 
-def get_version():
+def get_version() -> str:
     black_py = CURRENT_DIR / "black.py"
     _version_re = re.compile(r"__version__\s+=\s+(?P<version>.*)")
     with open(black_py, "r", encoding="utf8") as f:
-        version = _version_re.search(f.read()).group("version")
+        match = _version_re.search(f.read())
+        version = match.group("version") if match is not None else '"unknown"'
     return str(ast.literal_eval(version))