diff options
Diffstat (limited to 'sploit/util')
| -rw-r--r-- | sploit/util/__init__.py | 2 | ||||
| -rw-r--r-- | sploit/util/cmd.py | 26 | ||||
| -rw-r--r-- | sploit/util/log.py | 32 | 
3 files changed, 0 insertions, 60 deletions
| diff --git a/sploit/util/__init__.py b/sploit/util/__init__.py deleted file mode 100644 index 32a079b..0000000 --- a/sploit/util/__init__.py +++ /dev/null @@ -1,2 +0,0 @@ -from .cmd import * -from .log import * diff --git a/sploit/util/cmd.py b/sploit/util/cmd.py deleted file mode 100644 index 3a2b842..0000000 --- a/sploit/util/cmd.py +++ /dev/null @@ -1,26 +0,0 @@ -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" diff --git a/sploit/util/log.py b/sploit/util/log.py deleted file mode 100644 index 823b252..0000000 --- a/sploit/util/log.py +++ /dev/null @@ -1,32 +0,0 @@ -import codecs -import sys - -# https://docs.python.org/3/library/codecs.html#standard-encodings -ENCODING = None - -ERROR   = 31 -WARNING = 33 -STATUS  = 32 -NORMAL  =  0 -ALT     = 90 - -def enc_value(value, enc): -    if type(value) is bytes: -        if enc is not None: -            value = codecs.encode(value, enc) -        elif ENCODING is not None: -            value = codecs.encode(value, ENCODING) -        value = str(value)[2:-1] # strip b'' -    return str(value) - -def generic_log(*values, sep, end, file, flush, enc, color): -    string = sep.join([ enc_value(x, enc) for x in values ]) -    print(f'\033[{color}m{string}\033[0m', end=end, file=file, flush=flush) - -# For library internal use -def ilog(*values, sep=' ', end='\n', file=sys.stderr, flush=True, enc=None, color=STATUS): -    generic_log(*values, sep=sep, end=end, file=file, flush=flush, enc=enc, color=color) - -# For external use in user script (via print = elog) -def elog(*values, sep=' ', end='\n', file=sys.stdout, flush=True, enc=None, color=ALT): -    generic_log(*values, sep=sep, end=end, file=file, flush=flush, enc=enc, color=color) | 
