Commit ba6f635e authored by PidgeyL's avatar PidgeyL
Browse files

bugfixes

parent 4d9c613a
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -202,7 +202,7 @@ if __name__ == '__main__':
            if r.headers['last-modified'] == i['last-modified']:
                print("Not modified")
                sys.exit(0)
        db.setColUpdate("cve", f.headers['last-modified'])
        db.setColUpdate("cve", r.headers['last-modified'])

        # get your parser on !!
        parser = make_parser()
+4 −3
Original line number Diff line number Diff line
@@ -33,6 +33,7 @@ import tempfile

from lib.ProgressBar import progressbar
from lib.Config import Configuration
import lib.DatabaseLayer as db

argparser = argparse.ArgumentParser(description='populate/update NIST CWE Common Weakness Enumeration database')
argparser.add_argument('-v', action='store_true', help='verbose output')
@@ -82,7 +83,7 @@ try:
except:
    sys.exit("Cannot open url %s. Bad URL or not connected to the internet?"%(cwedict))
lastmodified = f.headers['last-modified']
i = dbLayer.getLastModified('cwe')
i = db.getLastModified('cwe')
if i is not None:
    if lastmodified == i:
        print("Not modified")
@@ -105,7 +106,7 @@ for cwe in progressbar(ch.cwe):
    if args.v:
        print (cwe)
    cweList.append(cwe)
dbLayer.cweBulkUpdate(cweList)
db.cweBulkUpdate(cweList)

#update database info after successful program-run
dbLayer.setColUpdate('cwe', lastmodified)
db.setColUpdate('cwe', lastmodified)