diff options
Diffstat (limited to '')
| -rw-r--r-- | sploit/__init__.py | 2 | ||||
| -rw-r--r-- | sploit/util.py | 20 |
2 files changed, 18 insertions, 4 deletions
diff --git a/sploit/__init__.py b/sploit/__init__.py index f5a82fc..5bac38f 100644 --- a/sploit/__init__.py +++ b/sploit/__init__.py @@ -8,3 +8,5 @@ from sploit import ( util, rev, ) + +__version__ = util.git_version() diff --git a/sploit/util.py b/sploit/util.py index c44ab78..3a2b842 100644 --- a/sploit/util.py +++ b/sploit/util.py @@ -1,14 +1,26 @@ +from os import path from subprocess import run -def run_cmd(cmd): - return run(cmd,capture_output=True).stdout.decode('utf-8').split('\n')[:-1] +def run_cmd(cmd,cwd=None): + return run(cmd,cwd=cwd,capture_output=True,text=True,check=True).stdout.split('\n')[:-1] __RUN_CACHE__ = {} -def run_cmd_cached(cmd): +def run_cmd_cached(cmd,cwd=None): key = ''.join(cmd) if key in __RUN_CACHE__: return __RUN_CACHE__[key] else: - result = run_cmd(cmd) + result = run_cmd(cmd,cwd) __RUN_CACHE__[key] = result return result + +#try to get the version through git +def git_version(): + try: + cwd = path.dirname(path.realpath(__file__)) + version = run_cmd(["git","describe","--always","--first-parent","--dirty"],cwd=cwd)[0] + #PEP440 compliance + version = version.replace('-','+',1).replace('-','.') + return version + except: + return "0+unknown.version" |