#999 dns record-list can handle zones with a missing Host record (like SRV records). Improved the zone-list command a bit by adding the records count column

Author: allmightyspiff

SoftLayer/CLI/dns/record_list.py

@@ -14,36 +14,28 @@
 @click.argument('zone')
 @click.option('--data', help='Record data, such as an IP address')
 @click.option('--record', help='Host record, such as www')
-@click.option('--ttl',
-              type=click.INT,
+@click.option('--ttl', type=click.INT,
               help='TTL value in seconds, such as 86400')
-@click.option('--type', help='Record type, such as A or CNAME')
+@click.option('--type', 'record_type', help='Record type, such as A or CNAME')
 @environment.pass_env
-def cli(env, zone, data, record, ttl, type):
+def cli(env, zone, data, record, ttl, record_type):
     """List all records in a zone."""
 
     manager = SoftLayer.DNSManager(env.client)
     table = formatting.Table(['id', 'record', 'type', 'ttl', 'data'])
-
-    table.align['ttl'] = 'l'
-    table.align['record'] = 'r'
-    table.align['data'] = 'l'
+    table.align = 'l'
 
     zone_id = helpers.resolve_id(manager.resolve_ids, zone, name='zone')
 
-    records = manager.get_records(zone_id,
-                                  record_type=type,
-                                  host=record,
-                                  ttl=ttl,
-                                  data=data)
+    records = manager.get_records(zone_id, record_type=record_type, host=record, ttl=ttl, data=data)
 
     for the_record in records:
         table.add_row([
-            the_record['id'],
-            the_record['host'],
-            the_record['type'].upper(),
-            the_record['ttl'],
-            the_record['data']
+            the_record.get('id'),
+            the_record.get('host', ''),
+            the_record.get('type').upper(),
+            the_record.get('ttl'),
+            the_record.get('data')
         ])
 
     env.fout(table)

SoftLayer/CLI/dns/zone_list.py

@@ -6,6 +6,7 @@
 import SoftLayer
 from SoftLayer.CLI import environment
 from SoftLayer.CLI import formatting
+from SoftLayer.utils import clean_time
 
 
 @click.command()
@@ -14,17 +15,22 @@ def cli(env):
     """List all zones."""
 
     manager = SoftLayer.DNSManager(env.client)
-    zones = manager.list_zones()
-    table = formatting.Table(['id', 'zone', 'serial', 'updated'])
-    table.align['serial'] = 'c'
-    table.align['updated'] = 'c'
-
+    objectMask = "mask[id,name,serial,updateDate,resourceRecordCount]"
+    zones = manager.list_zones(mask=objectMask)
+    table = formatting.Table(['id', 'zone', 'serial', 'updated', 'records'])
+    table.align = 'l'
     for zone in zones:
+        zone_serial = str(zone.get('serial'))
+        zone_date = zone.get('updateDate', None)
+        if zone_date is None:
+            # The serial is just YYYYMMDD##, and since there is no createDate, just format it like it was one.
+            zone_date = "{}-{}-{}T00:00:00-06:00".format(zone_serial[0:4], zone_serial[4:6], zone_serial[6:8])
         table.add_row([
             zone['id'],
             zone['name'],
-            zone['serial'],
-            zone['updateDate'],
+            zone_serial,
+            clean_time(zone_date),
+            zone.get('resourceRecordCount', 0)
         ])
 
     env.fout(table)

SoftLayer/managers/dns.py

@@ -39,7 +39,10 @@ def list_zones(self, **kwargs):
         :returns: A list of dictionaries representing the matching zones.
 
         """
-        return self.client['Account'].getDomains(**kwargs)
+        if kwargs.get('iter') is None:
+            kwargs['iter'] = True
+        return self.client.call('SoftLayer_Account', 'getDomains', **kwargs)
+        # return self.client['Account'].getDomains(**kwargs)
 
     def get_zone(self, zone_id, records=True):
         """Get a zone and its records.
@@ -181,8 +184,7 @@ def get_record(self, record_id):
         """
         return self.record.getObject(id=record_id)
 
-    def get_records(self, zone_id, ttl=None, data=None, host=None,
-                    record_type=None):
+    def get_records(self, zone_id, ttl=None, data=None, host=None, record_type=None):
         """List, and optionally filter, records within a zone.
 
         :param zone: the zone name in which to search.
@@ -191,8 +193,7 @@ def get_records(self, zone_id, ttl=None, data=None, host=None,
         :param str host: record's host
         :param str record_type: the type of record
 
-        :returns: A list of dictionaries representing the matching records
-                  within the specified zone.
+        :returns: A list of dictionaries representing the matching records within the specified zone.
         """
         _filter = utils.NestedDict()
 
@@ -208,12 +209,9 @@ def get_records(self, zone_id, ttl=None, data=None, host=None,
         if record_type:
             _filter['resourceRecords']['type'] = utils.query_filter(record_type.lower())
 
-        results = self.service.getResourceRecords(
-            id=zone_id,
-            mask='id,expire,domainId,host,minimum,refresh,retry,mxPriority,ttl,type,data,responsiblePerson',
-            filter=_filter.to_dict(),
-        )
-
+        objectMask = 'id,expire,domainId,host,minimum,refresh,retry,mxPriority,ttl,type,data,responsiblePerson'
+        results = self.client.call('SoftLayer_Dns_Domain', 'getResourceRecords', id=zone_id,
+                                    mask=objectMask, filter=_filter.to_dict(), iter=True)
         return results
 
     def edit_record(self, record):

tests/CLI/modules/dns_tests.py

@@ -54,11 +54,8 @@ def test_list_zones(self):
         result = self.run_command(['dns', 'zone-list'])
 
         self.assert_no_fail(result)
-        self.assertEqual(json.loads(result.output),
-                         [{'serial': 2014030728,
-                           'updated': '2014-03-07T13:52:31-06:00',
-                           'id': 12345,
-                           'zone': 'example.com'}])
+        actual_output = json.loads(result.output)
+        self.assertEqual(actual_output[0]['zone'], 'example.com')
 
     def test_list_records(self):
         result = self.run_command(['dns', 'record-list', '1234'])
@@ -261,3 +258,24 @@ def test_import_zone(self):
             self.assertEqual(call.args[0], expected_call)
 
         self.assertIn("Finished", result.output)
+
+    def test_issues_999(self):
+        """Makes sure certain zones with a None host record are pritable"""
+
+        # SRV records can have a None `host` record, or just a plain missing one.
+        fake_records = [
+            {
+                'data': '1.2.3.4',
+                'id': 137416416,
+                'ttl': 900,
+                'type': 'srv'
+            }
+        ]
+        record_mock = self.set_mock('SoftLayer_Dns_Domain', 'getResourceRecords')
+        record_mock.return_value = fake_records
+        result = self.run_command(['dns', 'record-list', '1234'])
+
+        self.assert_no_fail(result)
+        actual_output = json.loads(result.output)[0]
+        self.assertEqual(actual_output['id'], 137416416)
+        self.assertEqual(actual_output['record'], '')