Improve get_future_imports implementation.

[etc/vim.git] / setup.py

diff --git a/setup.py b/setup.py
index 4e9a5c97acbf02f0defa01f7aead84265db754b7..7ec262daf595fe64f1b01f726ebe7207a6de0e00 100644 (file)
--- 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))
 
 
@@ -40,10 +41,10 @@ setup(
     package_data={"blib2to3": ["*.txt"]},
     python_requires=">=3.6",
     zip_safe=False,
-    install_requires=["click", "attrs>=17.4.0", "appdirs"],
+    install_requires=["click>=6.5", "attrs>=17.4.0", "appdirs", "toml>=0.9.4"],
     test_suite="tests.test_black",
     classifiers=[
-        "Development Status :: 3 - Alpha",
+        "Development Status :: 4 - Beta",
         "Environment :: Console",
         "Intended Audience :: Developers",
         "License :: OSI Approved :: MIT License",