aboutsummaryrefslogtreecommitdiff
path: root/rpki
diff options
context:
space:
mode:
Diffstat (limited to 'rpki')
-rw-r--r--rpki/gui/cacheview/util.py2
-rw-r--r--rpki/gui/script_util.py5
2 files changed, 6 insertions, 1 deletions
diff --git a/rpki/gui/cacheview/util.py b/rpki/gui/cacheview/util.py
index 3ff2094c..00298b2c 100644
--- a/rpki/gui/cacheview/util.py
+++ b/rpki/gui/cacheview/util.py
@@ -154,7 +154,7 @@ model_class = {
def save_status(repo, vs):
timestamp = datetime.fromXMLtime(vs.timestamp).to_sql()
status = LABEL_CACHE[vs.status]
- g = models.generations_dict[vs.generation] if vs.generation else None
+ g = models.generations_dict.get(vs.generation)
repo.statuses.create(generation=g, timestamp=timestamp, status=status)
# if this object is in our interest set, update with the current validation
diff --git a/rpki/gui/script_util.py b/rpki/gui/script_util.py
index 31e40821..17e5608c 100644
--- a/rpki/gui/script_util.py
+++ b/rpki/gui/script_util.py
@@ -28,3 +28,8 @@ def setup():
# If this doesn't work, try changing it to "rpki.django_settings.gui".
os.environ.update(DJANGO_SETTINGS_MODULE = "rpki.django_settings.irdb")
+ #os.environ.update(DJANGO_SETTINGS_MODULE = "rpki.django_settings.gui")
+
+ # Initialize Django.
+ import django
+ django.setup()