diff options
Diffstat (limited to 'lib/orm/models.py')
| -rw-r--r-- | lib/orm/models.py | 25 |
1 files changed, 9 insertions, 16 deletions
diff --git a/lib/orm/models.py b/lib/orm/models.py index 8d95c319..bca7d427 100644 --- a/lib/orm/models.py +++ b/lib/orm/models.py @@ -22,13 +22,7 @@ from __future__ import unicode_literals from django.db import models, IntegrityError, DataError -from django.db.models import F, Q, Sum, Count -from django.db import IntegrityError, transaction -from django.utils import timezone -from django.utils.encoding import force_bytes -from django.contrib.auth.models import Group, AnonymousUser - -from django.urls import reverse +from django.db import transaction from django.core import validators from django.conf import settings @@ -39,9 +33,8 @@ from users.models import SrtUser import sys import os import re -import itertools from signal import SIGUSR1 -from datetime import timedelta, datetime +from datetime import datetime import json import logging @@ -53,7 +46,7 @@ from srtgui.api import _log # Sqlite support if 'sqlite' in settings.DATABASES['default']['ENGINE']: - from django.db import transaction, OperationalError + from django.db import OperationalError from time import sleep _base_save = models.Model.save @@ -103,7 +96,6 @@ if 'sqlite' in settings.DATABASES['default']['ENGINE']: # end of HACK class GitURLValidator(validators.URLValidator): - import re regex = re.compile( r'^(?:ssh|git|http|ftp)s?://' # http:// or https:// r'(?:(?:[A-Z0-9](?:[A-Z0-9-]{0,61}[A-Z0-9])?\.)+(?:[A-Z]{2,6}\.?|[A-Z0-9-]{2,}\.?)|' # domain... @@ -127,7 +119,7 @@ class SrtSetting(models.Model): helptext = models.TextField() value = models.CharField(max_length=255) - def __unicode__(self): + def __str__(self): return "Setting %s = %s" % (self.name, self.value) class HelpText(models.Model): @@ -350,12 +342,12 @@ class CveDetail(): def get_cpe_list(self): cpe_array = [] for cpe in self.cpe_list.split('|'): - cpe_array.append(cpe.split(',')) + cpe_array.append(cpe.split(',')) return cpe_array def get_ref_list(self): ref_array = [] for ref in self.ref_list.split('|'): - ref_array.append(ref.split('\t')) + ref_array.append(ref.split('\t')) return ref_array # Local full Cve class, based on "Cve" @@ -668,6 +660,7 @@ class Vulnerability(models.Model): @staticmethod def new_vulnerability_name(): # get next vulnerability name atomically + # FIXME ??? if True: current_vulnerability_index,create = SrtSetting.objects.get_or_create(name='current_vulnerability_index') if create: @@ -1012,9 +1005,9 @@ class NotifyCategories(models.Model): def invalidate_cache(**kwargs): from django.core.cache import cache try: - cache.clear() + cache.clear() except Exception as e: - logger.warning("Problem with cache backend: Failed to clear cache: %s" % e) + logger.warning("Problem with cache backend: Failed to clear cache: %s" % e) def signal_runbuilds(): """Send SIGUSR1 to runbuilds process""" |