summaryrefslogtreecommitdiffstats
path: root/nsploit/util/cmd.py
blob: 3a2b8422bc75acc8741521df694f1311875c81a0 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
from os import path
from subprocess import run

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,cwd=None):
    key = ''.join(cmd)
    if key in __RUN_CACHE__:
        return __RUN_CACHE__[key]
    else:
        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"