Version info: handle betas

pypa · Oct 16, 2019 · bdf9ec1 · bdf9ec1
1 parent 1486517
commit bdf9ec1
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pipx/main.py b/pipx/main.py
@@ -6,6 +6,7 @@
 import argparse
 import functools
 import logging
+from pkg_resources import parse_version
 import shlex
 import sys
 import textwrap
@@ -29,7 +30,11 @@
 
 
 __version__ = "0.14.0.0"
-__version_info__ = tuple(int(n) for n in __version__.split("."))
+_, __version_info__, subver, *_ = parse_version(__version__)._key  # type: ignore
+if subver:
+    while len(__version_info__) < 4:
+        __version_info__ += (0,)
+    __version_info__ += subver
 
 
 def print_version() -> None: