seed.py

#!/usr/bin/python3

import json
import os

from classes import *
from mongoengine import *
from utils import *

configfile = 'options.json'

if not os.path.isfile(configfile):
  print('Could not find ' + configfile + ' aborting!')
  sys.exit()

with open(configfile) as config_file:
  config = json.load(config_file)

db = connect(config['dbname'], host=config['dbhost'])
db.drop_database(config['dbname'])

mitrelink='https://cve.mitre.org/cgi-bin/cvename.cgi?name='

f = open('cves.txt')
while True:
  x = f.readline().rstrip()
  if not x: break
  CVE(cve_name=x).save()
  Links(cve_id=CVE.objects.get(cve_name=x), link=mitrelink+x).save()

f = open('statuses.txt')
while True:
  x = f.readline().rstrip()
  if not x: break
  Status(short_id=x.split('|')[0], text=x.split('|')[1]).save()

getKernelTableFromGithub()

f = open('patches.txt')
while True:
  x = f.readline().rstrip()
  if not x: break
  if x[0] != "#":
    k = x.split('|')[0]
    j = json.loads(x.split('|')[1])
    if "lge-gproj" in k or "lge-mako" in k or "lge-p880" in k:
      k = k.replace("lge-", "lge-kernel-")
    else:
      k = "android_kernel_" + k
    try:
      kernel_id = Kernel.objects.get(repo_name=k).id
      for c in j:
        try:
          cve_id = CVE.objects.get(cve_name=c).id
          status_id = Status.objects.get(short_id=j[c]).id
          Patches.objects(cve=cve_id, kernel=kernel_id).update(status=status_id)
        except:
          print("Couldn't determine id for " + c)
    except:
      print("Couldn't determine id for " + k)