Don't crash if, get_domain() gets a non existing domain name.

[people/stevee/pypdns.git] / backend.py

diff --git a/backend.py b/backend.py
index 4acdd204f93076af277ab0be2ccda8692e20bdb0..3e04d39df5bb93b0c6f6f63130d426481b75f2d6 100644 (file)
--- a/backend.py
+++ b/backend.py
@@ -47,9 +47,12 @@ class DNS(object):
        # Get a domain by it's name.
        def get_domain(self, name):
                row = self.db.get("SELECT id FROM domains WHERE name = ?", name)
-               domain = Domain(self, row.id)
 
-               return domain
+               # Only do anything, if there is an existing domain.
+               if row:
+                       domain = Domain(self, row.id)
+
+                       return domain
 
 # Create Domain class.
 """Use query method from database module to get all requested information about our domain."""