From f0e00d8f2abaef4bdbeacdbff14e6a86b8baa791 Mon Sep 17 00:00:00 2001 From: dtomlinson Date: Mon, 13 Jan 2020 00:25:07 +0000 Subject: [PATCH] updating header.py to set the default of panaetius if no __header__ or virtualenv has been detected --- src/panaetius/header.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/src/panaetius/header.py b/src/panaetius/header.py index ec9687b..4d365fa 100644 --- a/src/panaetius/header.py +++ b/src/panaetius/header.py @@ -11,14 +11,16 @@ try: __spec.loader.exec_module(__header__) __header__ = __header__.__header__ except FileNotFoundError: - venv = os.environ.get('VIRTUAL_ENV').split('/')[-1] - if venv is not None: + try: + venv = os.environ.get('VIRTUAL_ENV').split('/')[-1] __header__ = venv - else: - raise FileNotFoundError( + except AttributeError: + print( f'Cannot find a __header__.py file in {os.getcwd()} containing the' ' __header__ value of your project name and you are not working' ' from a virtual environment. Either make sure this file ' 'exists and the value is set or create and work from a virtual ' - 'environment and try again.' + 'environment and try again. \n The __header__ value has been ' + 'set to the default of panaetius.' ) + __header__ = 'panaetius'