models.py

from django.db import models

# Create your models here.
class Exploit(models.Model):
    # Model fields are subject to change
    source = models.CharField(max_length=4096)
    source_url = models.URLField(max_length=4096, null=True, blank=True)
    name = models.CharField(max_length=4096)
    cve_id = models.CharField(max_length=4096, null=True, blank=True)
    date_published = models.DateField(null=True, blank=True)
    author = models.CharField(max_length=4096, null=True, blank=True)
    description = models.TextField(null=True, blank=True)
    example_file = models.FileField(null=True, blank=True)
    download_failed = models.BooleanField(default=False)
    is_repo = models.BooleanField(default=False)
    ignore = models.BooleanField(default=False)
    fixed = models.BooleanField(default=False)
    date_created = models.DateTimeField(auto_now_add=True, blank=True, null=True)
    # Below attributes were added for the model output from the process_html.py
    # All of the attributes are text fields just to be safe (better than being sorry).
    # Vendor
    vendor = models.TextField(null=True, blank=True)
    # Product
    product = models.TextField(null=True, blank=True)
    # Vulnerability type
    vul_type = models.TextField(null=True, blank=True)
    # Risk
    risk = models.TextField(null=True, blank=True)
    # Pub Dates (not to be confused with the date_published attribute)
    pub_dates = models.TextField(null=True, blank=True)
    # Version
    version = models.TextField(null=True, blank=True)
    # Remot/Local
    remote_local = models.TextField(null=True, blank=True)
    # Host Information
    host_info = models.TextField(null=True, blank=True)
    # Proof of Concept
    poc = models.TextField(null=True, blank=True)
    # Steps to reproduce
    reproduce = models.TextField(null=True, blank=True)
    # Impact
    impact = models.TextField(null=True, blank=True)

    def __str__(self):
        return self.name