diff --git a/requirements/base.txt b/requirements/base.txt index de39df0..7776cb0 100644 --- a/requirements/base.txt +++ b/requirements/base.txt @@ -4,3 +4,4 @@ lxml==3.2.3 psutil==1.1.3 Pillow==2.3.0 pylibmc==1.2.3 +GitPython diff --git a/vmdriver.py b/vmdriver.py index f692e9e..781c47e 100644 --- a/vmdriver.py +++ b/vmdriver.py @@ -619,10 +619,22 @@ def get_ram_size(): @celery.task +def get_driver_version(): + from git import Repo + repo = Repo(path=os.getcwd()) + lc = repo.log()[0] + return {'branch': repo.active_branch, + 'commit': lc.id_abbrev, + 'commit_text': lc.summary, + 'is_dirty': repo.is_dirty} + + +@celery.task def get_info(): return {'core_num': get_core_num(), 'ram_size': get_ram_size(), - 'architecture': get_architecture()} + 'architecture': get_architecture(), + 'driver_version': get_driver_version()} @celery.task