From cbd5705e57b9d16e067eaab5fb155c77baec599a Mon Sep 17 00:00:00 2001
From: dkaslovsky <dkaslovsky@gmail.com>
Date: Sat, 5 Sep 2020 14:31:34 -0600
Subject: [PATCH] validate parsed version

---
 setup.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index bc98050..5879738 100644
--- a/setup.py
+++ b/setup.py
@@ -1,4 +1,5 @@
 import os
+import re
 from setuptools import setup
 
 
@@ -22,11 +23,16 @@ def parse_version(version_str: str) -> str:
     Parse the package version
     :param version_str: string of the form "__version__ = 'x.y.z'"
     """
+    version_str = version_str.strip('\n')
     try:
         version = version_str.split('=')[1]
     except IndexError:
-        return 'unknown'
-    return version.strip(' \'\"\n')
+        raise RuntimeError(f'cannot parse version from [{version_str}]')
+    version = version.strip(' \'\"')
+    # validate version
+    if not re.match(r'^(\d+).(\d+).(\d+)$', version):
+        raise RuntimeError(f'parsed invalid version [{version}] from [{version_str}]')
+    return version
 
 
 setup(