122 lines
2.7 KiB
Python
122 lines
2.7 KiB
Python
import subprocess
|
|
import re
|
|
from conf import CLASS_BLACKLIST
|
|
import random
|
|
import string
|
|
import sys
|
|
import os
|
|
|
|
def replace_classnames(code):
|
|
char_set = string.ascii_lowercase
|
|
result = code
|
|
for item in CLASS_BLACKLIST:
|
|
if "%s(" % item in result:
|
|
rand = ''.join(random.sample(char_set*6,6))
|
|
result = result.replace(item, rand)
|
|
return result
|
|
|
|
|
|
def run(cmd):
|
|
p = subprocess.Popen(cmd, stdout=subprocess.PIPE, shell=True)
|
|
output, error = p.communicate()
|
|
yield output
|
|
|
|
|
|
def sanitize(string):
|
|
"Converts parameters and stuff to be correctly evaluated."
|
|
# Remove double parentesis
|
|
sane = string.strip().replace('))', ')')
|
|
# Boolean values
|
|
sane = sane.strip().replace('false', 'False')
|
|
sane = sane.strip().replace('true', 'True')
|
|
# Convert float values to string
|
|
regex = re.compile('([\-]?\d?\.?\d\F)')
|
|
sane = regex.sub("'\1'", sane)
|
|
# Convert rest t
|
|
regex = re.compile('([, |\(])([\w+\d+\.]+)')
|
|
sane = regex.sub(r'\1"\2"', sane)
|
|
sane = replace_classnames(sane)
|
|
return sane
|
|
|
|
# Functions
|
|
class Colors:
|
|
END = '\033[0m'
|
|
|
|
def __init__(self, c='ter,'):
|
|
if c == 'custom':
|
|
self.custom()
|
|
else:
|
|
self.term()
|
|
|
|
def custom(self):
|
|
self.WHITE = '\033[89m'
|
|
self.BLACK = '\033[90m'
|
|
self.RED = '\033[91m'
|
|
self.GREEN = '\033[92m'
|
|
self.YELLOW = '\033[93m'
|
|
self.BLUE = '\033[94m'
|
|
self.PURPLE = '\033[95m'
|
|
self.CYAN = '\033[96m'
|
|
|
|
def term(self):
|
|
self.WHITE = '\033[29m'
|
|
self.BLACK = '\033[30m'
|
|
self.RED = '\033[31m'
|
|
self.GREEN = '\033[32m'
|
|
self.YELLOW = '\033[33m'
|
|
self.BLUE = '\033[34m'
|
|
self.PURPLE = '\033[35m'
|
|
self.CYAN = '\033[36m'
|
|
|
|
colors = Colors('custom')
|
|
|
|
|
|
def echo(string, end='\r\n', color=None):
|
|
if color:
|
|
sys.stdout.write("%s%s%s" % (
|
|
color,
|
|
string,
|
|
colors.END
|
|
))
|
|
else:
|
|
sys.stdout.write(string)
|
|
if end:
|
|
sys.stdout.write("\n")
|
|
sys.stdout.flush()
|
|
|
|
|
|
# Shortcurts
|
|
def check_status(status=None, words=['done', 'failed']):
|
|
if status is not None:
|
|
if status == 0:
|
|
echo(words[0], color=colors.GREEN)
|
|
else:
|
|
echo(words[1], color=colors.RED)
|
|
pass
|
|
#exit(-1)
|
|
|
|
|
|
def exists(path):
|
|
return os.path.exists(path)
|
|
|
|
|
|
def title(string):
|
|
print("")
|
|
echo("[==] %s" % string, color=colors.PURPLE)
|
|
|
|
|
|
def info(string):
|
|
echo("[ i] %s" % string, color=colors.BLUE)
|
|
|
|
|
|
def error(string):
|
|
echo("[ E] %s" % string, color=colors.RED)
|
|
|
|
|
|
def success(string):
|
|
echo("[OK] %s" % string, color=colors.GREEN)
|
|
|
|
|
|
def sub(string, end='', color=None):
|
|
echo(" %s " % string, end=end, color=color)
|