diff --git a/setup.py b/setup.py index 6c01717b32b4e02740a0808d5a8c87d17aa7b3be..d6ebf3b743ef19b267223b74773d9cf6b22e2c9e 100644 --- a/setup.py +++ b/setup.py @@ -1,10 +1,24 @@ #!/usr/bin/env python from distutils.core import setup +import subprocess + +# Write a version file containing the git hash and info +git_log = subprocess.check_output(['git', 'log', '-1', '--pretty=%h %ai']) +git_diff = (subprocess.check_output(['git', 'diff', '.']) + + subprocess.check_output(['git', 'diff', '--cached', '.'])) +if git_diff == '': + status = '(CLEAN) ' + git_log +else: + status = '(UNCLEAN) ' + git_log +version_file = '.version' +with open('tupak/' + version_file, 'w+') as f: + f.write(status) setup(name='tupak', version='0.1', packages=['tupak'], package_dir={'tupak': 'tupak'}, - package_data={'tupak': ['noise_curves/*.txt', 'prior_files/*.txt']} + package_data={'tupak': ['noise_curves/*.txt', 'prior_files/*.txt', + version_file]} ) diff --git a/tupak/utils.py b/tupak/utils.py index eb1e20025603d802f76596582e223208f62ed79a..a6405f5883a583d40967d01b102eabaad37cc3ae 100644 --- a/tupak/utils.py +++ b/tupak/utils.py @@ -322,6 +322,11 @@ def setup_logger(outdir=None, label=None, log_level='info'): file_handler.setLevel(LEVEL) logger.addHandler(file_handler) + version_file = os.path.join(os.path.dirname(__file__), '.version') + with open(version_file, 'r') as f: + version = f.readline() + logging.info('Running tupak version: {}'.format(version)) + def get_progress_bar(module='tqdm'): if module in ['tqdm']: