Add class to handle SOA records and parse the content.

diff --git a/backend.py b/backend.py
index 9f9e85ba3b51b5c0f1aac3989087099bb3fcee72..4acdd204f93076af277ab0be2ccda8692e20bdb0 100644 (file)
--- a/backend.py
+++ b/backend.py
@@ -105,12 +105,32 @@ class Domain(object):
                records = []
 
                """Fetch all records from domain, line by line and add them to the previous created empty list."""
-               for row in self.db.query("SELECT id FROM records WHERE domain_id = ?", self.id):
-                       record = Record(self, row.id) 
+               for row in self.db.query("SELECT id, type FROM records WHERE domain_id = ?", self.id):
+                       if row.type == "SOA":
+                               record = SOARecord(self, row.id)
+                       elif row.type == "A":
+                               record = ARecord(self, row.id)
+                       else:
+                               record = Record(self, row.id) 
                        records.append(record)
 
                return records
 
+       # Get records by a specified type.
+       def get_records_by_type(self, type):
+               records = []
+               for record in self.records:
+                       if record.type == type:
+                               records.append(record)
+
+               return records
+
+       # Quick function to get the first SOA record from the domain.
+       @property
+       def SOA(self):
+               records = self.get_records_by_type("SOA")
+               if records:
+                       return records[0]
 
 
 # Create class for domain records.
@@ -169,8 +189,63 @@ class Record(object):
                row = self.db.get("SELECT ordername FROM records WHERE id = ?" , self.id)
                return row.ordername
 
-       # Gain all information about records authification.
+       # Gain all information about records authentication.
        @property
-       def authification(self):
+       def authentication(self):
                row = self.db.get("SELECT auth FROM records WHERE id = ?" , self.id)
                return row.auth
+
+
+# Create an own class to deal with "SOA" records.
+"""Use splitt() to generate a list of the original content string from the database, and return the requested entries.""" 
+class SOARecord(Record):
+       def __init__(self, domain, record_id):
+               Record.__init__(self, domain, record_id)
+
+               self.soa_attrs = self.content.split()
+
+               # Check if the content from database is valid (It contains all 7 required information).
+               if not len(self.soa_attrs) == 7:
+                       #XXX Add something like an error message or log output.
+                       pass                    
+
+       # Primary NS - the domain name of the name server that was the original source of the data.     
+       @property
+       def mname(self):
+               return self.soa_attrs[0]
+
+       # E-mail address of the person which is responsible for this domain.
+       @property
+       def email(self):
+               return self.soa_attrs[1]
+
+       # The serial which increases allways after a change on the domain has been made.
+       @property
+       def serial(self):
+               return self.soa_attrs[2]
+
+       # The number of seconds between the time that a secondary name server gets a copy of the domain.
+       @property
+       def refresh(self):
+               return self.soa_attrs[3]
+
+       # The number of seconds during the next refresh attempt if the previous fails.
+       @property
+       def retry(self):
+               return self.soa_attrs[4]
+
+       # The number of seconds that lets the secondary name server(s) know how long they can hold the information.
+       @property
+       def expire(self):
+               return self.soa_attrs[5]
+
+       # The number of seconds that the records in the domain are valid.
+       @property
+       def minimum(self):
+               return self.soa_attrs[6]
+
+
+
+class ARecord(Record):
+       pass
+