Detect missing setuptools-scm in setup.py

axnsan12 · Dec 21, 2018 · 8057ce7 · 8057ce7
1 parent 3b31c54
commit 8057ce7
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -65,13 +65,15 @@ def drf_yasg_setup(**kwargs):
 
 
 try:
+    import setuptools_scm
     drf_yasg_setup(use_scm_version=True)
-except LookupError as e:
+except (ImportError, LookupError) as e:
     if os.getenv('CI', 'false') == 'true' or os.getenv('TRAVIS', 'false') == 'true':
         # don't silently fail on travis - we don't want to accidentally push a dummy version to PyPI
         raise
 
-    if 'setuptools-scm' in str(e):
+    err_msg = str(e)
+    if 'setuptools-scm' in err_msg or 'setuptools_scm' in err_msg:
         import time
 
         timestamp_ms = int(time.time() * 1000)