From 8057ce7a4aee1aef381cb76751a1ef6c3e6e797a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Cristi=20V=C3=AEjdea?= Date: Fri, 21 Dec 2018 12:41:06 +0200 Subject: [PATCH] Detect missing setuptools-scm in setup.py --- setup.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 7ab033d..baa3de0 100755 --- 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)