def registerPoc(self):
pocString = multipleReplace(self.pocString, POC_IMPORTDICT)
_, self.moduleName = filepathParser(self.pocName)
try:
importer = StringImporter(self.moduleName, pocString)
importer.load_module(self.moduleName)
except ImportError, ex:
logger.log(CUSTOM_LOGGING.ERROR, ex)
def registerJsonPoc(path):
_, pocname = filepathParser(path)
if pocname in kb.registeredPocs:
return
with open(path) as f:
jsonPoc = json.load(f)
kb.registeredPocs[pocname] = jsonPoc
def registerPyPoc(path):
_, moduleName = filepathParser(path)
module = changeToPyImportType(path)
try:
__import__(module, fromlist=["*"])
except ImportError:
# TODO 需要再搞一下
addSysPath(_)
__import__(moduleName, fromlist=["*"])
def registerPyPoc(pocDict):
pocname = pocDict.keys()[0]
_, moduleName = filepathParser(pocname)
try:
importer = StringImporter(moduleName, pocDict[pocname])
importer.load_module(moduleName)
except ImportError, ex:
errMsg = "%s register failed \"%s\"" % (moduleName, str(ex))
logger.log(CUSTOM_LOGGING.ERROR, errMsg)
def registerPyPoc(pocDict):
pocname = pocDict.keys()[0]
_, moduleName = filepathParser(pocname)
try:
importer = StringImporter(moduleName, pocDict[pocname])
importer.load_module(moduleName)
except ImportError, ex:
errMsg = "%s register failed \"%s\"" % (moduleName, str(ex))
logger.log(CUSTOM_LOGGING.ERROR, errMsg)
def do_search(self, args):
graph = PrettyTable(["pocId", "PocName"])
graph.align["PocName"] = "m"
graph.padding_width = 1
for k, v in kb.unloadedList.iteritems():
if str(args) in v:
graph.add_row([k, filepathParser(v)[1]])
print graph
pass
def do_search(self, args):
graph = PrettyTable(["pocId", "PocName"])
graph.align["PocName"] = "m"
graph.padding_width = 1
for k, v in kb.unloadedList.iteritems():
if str(args) in v:
graph.add_row([k, filepathParser(v)[1]])
print graph
pass
def avaliable():
graph = PrettyTable(["pocId", "avaliablePocName", "Folder"])
graph.align["PocsName"] = "m"
graph.padding_width = 1
for k, v in kb.unloadedList.iteritems():
path, name = filepathParser(v)
graph.add_row([k, name, os.path.relpath(path, paths.POCSUITE_ROOT_PATH)])
print graph
print
def avaliable():
graph = PrettyTable(["pocId", "avaliablePocName", "Folder"])
graph.align["PocsName"] = "m"
graph.padding_width = 1
for k, v in kb.unloadedList.iteritems():
path, name = filepathParser(v)
graph.add_row([k, name, os.path.relpath(path, paths.POCSUITE_ROOT_PATH)])
print graph
print
def registerPyPoc(path):
_, moduleName = filepathParser(path)
module = changeToPyImportType(path)
try:
__import__(module, fromlist=["*"])
except ImportError:
# TODO 需要再搞一下
addSysPath(_)
try:
__import__(moduleName, fromlist=["*"])
except ImportError, ex:
errMsg = "%s register failed \"%s\"" % (path, str(ex))
logger.log(CUSTOM_LOGGING.ERROR, errMsg)
