def __test(): db = Database() db.db_init() dbsession = db.db_session() testCVE = Spider(dbsession) result, e = testCVE.get_vulns() if not result: print(e)
from lib.common.banner import banner from config.config import * import config.config as config from lib.database import Database from lib.Modles import Vuln from sqlalchemy import and_, or_ import ipdb import logging # 原计划实现对Python版本判断,但疑似python2在运行前便对整个脚本进行了语法校验,如此导致大量错误。 # if not check_python_version(3): # sys.exit() db = Database() db.db_init() def spider(spider_plugin): global db # plugin_path = os.path.join(configs.get('plugin_dir'), spider_plugin) try: print() # 换行 print("[*] Loading spider plugin :: < %s >" % spider_plugin, end="") plugin = importlib.import_module("lib.plugins.%s"%spider_plugin[:-3]) print("\tSuccess.") Spider = getattr(plugin, "Spider") # print(Spider) except Exception as e: print("[*] Something wrong with run < %s >"%spider_plugin) print("[*] Error - %s" % str(e))