Beispiel #1
0
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
Beispiel #2
0
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)