From 339f735bcd2bdf4e9c56f4cb8082ce53fc90caf7 Mon Sep 17 00:00:00 2001
From: Rob Austein <sra@hactrn.net>
Date: Sun, 24 Mar 2013 21:40:57 +0000
Subject: First pass at cleaning up exception handling in rpkic and zookeeper.
 See #485.

svn path=/trunk/; revision=5205
---
 rpkid/rpki/irdb/zookeeper.py | 45 +++++++++++---------------------------------
 1 file changed, 11 insertions(+), 34 deletions(-)

(limited to 'rpkid/rpki/irdb/zookeeper.py')

diff --git a/rpkid/rpki/irdb/zookeeper.py b/rpkid/rpki/irdb/zookeeper.py
index d4fd59d0..b9d44c57 100644
--- a/rpkid/rpki/irdb/zookeeper.py
+++ b/rpkid/rpki/irdb/zookeeper.py
@@ -60,7 +60,8 @@ rootd_section  = "rootd"
 
 # A whole lot of exceptions
 
-class MissingHandle(Exception):         "Missing handle"
+class HandleNotSet(Exception):          "Handle not set."
+class MissingHandle(Exception):         "Missing handle."
 class CouldntTalkToDaemon(Exception):   "Couldn't talk to daemon."
 class BadXMLMessage(Exception):         "Bad XML message."
 class PastExpiration(Exception):        "Expiration date has already passed."
@@ -253,11 +254,9 @@ class Zookeeper(object):
     Get ResourceHolderCA object associated with current handle.
     """
 
-    assert self.handle is not None
-    try:
-      return rpki.irdb.ResourceHolderCA.objects.get(handle = self.handle)
-    except rpki.irdb.ResourceHolderCA.DoesNotExist:
-      return None
+    if self.handle is None:
+      raise HandleNotSet
+    return rpki.irdb.ResourceHolderCA.objects.get(handle = self.handle)
 
 
   @property
@@ -266,10 +265,7 @@ class Zookeeper(object):
     Get ServerCA object.
     """
 
-    try:
-      return rpki.irdb.ServerCA.objects.get()
-    except rpki.irdb.ServerCA.DoesNotExist:
-      return None
+    return rpki.irdb.ServerCA.objects.get()
 
 
   @django.db.transaction.commit_on_success
@@ -517,11 +513,7 @@ class Zookeeper(object):
     Delete a child of this RPKI entity.
     """
 
-    assert child_handle is not None
-    try:
-      self.resource_ca.children.get(handle = child_handle).delete()
-    except rpki.irdb.Child.DoesNotExist:
-      self.log("No such child \"%s\"" % child_handle)
+    self.resource_ca.children.get(handle = child_handle).delete()
 
 
   @django.db.transaction.commit_on_success
@@ -593,11 +585,7 @@ class Zookeeper(object):
     Delete a parent of this RPKI entity.
     """
 
-    assert parent_handle is not None
-    try:
-      self.resource_ca.parents.get(handle = parent_handle).delete()
-    except rpki.irdb.Parent.DoesNotExist:
-      self.log("No such parent \"%s\"" % parent_handle)
+    self.resource_ca.parents.get(handle = parent_handle).delete()
 
 
   @django.db.transaction.commit_on_success
@@ -606,10 +594,7 @@ class Zookeeper(object):
     Delete rootd associated with this RPKI entity.
     """
 
-    try:
-      self.resource_ca.rootd.delete()
-    except rpki.irdb.Rootd.DoesNotExist:
-      self.log("No associated rootd")
+    self.resource_ca.rootd.delete()
 
 
   @django.db.transaction.commit_on_success
@@ -715,11 +700,7 @@ class Zookeeper(object):
     Delete a publication client of this RPKI entity.
     """
 
-    assert client_handle is not None
-    try:
-      self.server_ca.clients.get(handle = client_handle).delete()
-    except rpki.irdb.Client.DoesNotExist:
-      self.log("No such client \"%s\"" % client_handle)
+    self.server_ca.clients.get(handle = client_handle).delete()
 
 
   @django.db.transaction.commit_on_success
@@ -766,11 +747,7 @@ class Zookeeper(object):
     Delete a repository of this RPKI entity.
     """
 
-    assert repository_handle is not None
-    try:
-      self.resource_ca.repositories.get(handle = repository_handle).delete()
-    except rpki.irdb.Repository.DoesNotExist:
-      self.log("No such repository \"%s\"" % repository_handle)
+    self.resource_ca.repositories.get(handle = repository_handle).delete()
 
 
   @django.db.transaction.commit_on_success
-- 
cgit v1.2.3