Allow getting zone records of specific type (#5)

ludeeus · Oct 6, 2020 · 522a103 · 522a103
1 parent db2c3c4
commit 522a103
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pycfdns/__init__.py b/pycfdns/__init__.py
@@ -46,11 +46,14 @@ async def get_zone_id(self):
             raise CloudflareZoneException("Could not get zone ID") from error
         return zone_id
 
-    async def get_zone_records(self, zone_id):
+    async def get_zone_records(self, zone_id, record_type = None):
         """Get the records of a zone."""
         records = []
 
         endpoint = f"{zone_id}/dns_records&per_page=100"
+        if record_type:
+            endpoint += f"&type={record_type}"
+
         url = BASE_URL.format(endpoint)
         data = await self.api.get_json(url)
         data = data["result"]