def main(): opts = parseArg() if opts.search: search(opts.search, opts.exclude) elif opts.analyze: statis(opts.analyze) elif opts.init: initDB(Config.dbOpts, opts.year) elif opts.update: db = DB(Config.dbOpts) datas = [updateValue(opts.update)] # print(datas) db.addCVEData(datas) return
def initDB(dbOpts, year): minYear = 0 maxYear = 3000 year = year.split(",") if len(year) == 1: minYear = int(year[0]) elif len(year) == 2: minYear = int(year[0]) maxYear = int(year[1]) db = DB(dbOpts) datas = [] for kd in getValue(minYear=minYear, maxYear=maxYear): # for kd in getValue(minYear=2020, minNumber=4, maxNumber=9): # print(kd) datas.append(kd) if len(datas) > 1000: print("insert upto %s" % kd[0]) db.addCVEData(datas) datas = [] db.addCVEData(datas)