Commit a7286371 authored by PidgeyL's avatar PidgeyL
Browse files

bugfix via4

parent 6424ac88
Loading
Loading
Loading
Loading
+4 −2
Original line number Diff line number Diff line
@@ -38,5 +38,7 @@ setIndex('mgmt_whitelist', 'id')
setIndex('mgmt_blacklist', 'id')
setIndex('capec', 'related_weakness')

for index in dbLayer.getInfo('via4').get('searchables', []):
via4 = dbLayer.getInfo('via4')
if via4:
    for index in via4.get('searchables', []):
        setIndex('via4', index, False)
+6 −3
Original line number Diff line number Diff line
@@ -15,6 +15,8 @@ import sys
runPath = os.path.dirname(os.path.realpath(__file__))
sys.path.append(os.path.join(runPath, ".."))

from dateutil.parser import parse as parse_datetime

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

@@ -36,9 +38,10 @@ except:
    sys.exit("Cannot open url %s. Bad URL or not connected to the internet?"%(Configuration.getFeedURL("via4")))

# check modification date
i=db.getInfo("via4")
lastmodified = parse_datetime(r.headers['last-modified'], ignoretz=True)
i=db.getLastModified("via4")
if i is not None:
    if r.headers['last-modified'] == i['last-modified']:
    if lastmodified == i:
        print("Not modified")
        sys.exit(0)

@@ -50,4 +53,4 @@ db.setColInfo('via4', 'sources', data['metadata']['sources'])
db.setColInfo('via4', 'searchables', data['metadata']['searchables'])

#update database info after successful program-run
db.setColUpdate('via4', r.headers['last-modified'])
db.setColUpdate('via4', lastmodified)