github/corenetworks
0
0
Fork 0
mirror of https://github.com/thegeeklab/corenetworks.git synced 2024-11-24 10:00:39 +00:00
Code Issues Releases Activity

fix formatting and tests

Browse Source
This commit is contained in:
Robert Kaussow 2020-05-11 22:50:12 +02:00
parent a332423c45
commit 6393f7a4c7
2 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
corenetworks/client.py
View File

@ -211,13 +211,20 @@ class CoreNetworks():
records = self.records(zone, params={"name": params["name"]}) records = self.records(zone, params={"name": params["name"]})
for r in records: for r in records:
if params["type"] == "CNAME" or r["type"] == "CNAME": if params["type"] == "CNAME" and r["type"] != "CNAME":
raise CorenetworksError( raise CorenetworksError(
"A record with the same name already exist ({name}). " "A record with the same name already exist ({name}). "
"CNAME records cannot use the same name with other records.".format( "CNAME records cannot use the same name with other records.".format(
name=r["name"] name=r["name"]
) )
) )
if params["type"] != "CNAME" and r["type"] == "CNAME":
raise CorenetworksError(
"A CNAME record with the same name already exist ({name}). "
"CNAME records cannot use the same name with other records.".format(
name=r["name"]
)
)
curr = copy.deepcopy(params) curr = copy.deepcopy(params)
curr.pop("ttl") curr.pop("ttl")

6
corenetworks/test/fixtures/callback.py vendored
View File

@ -31,7 +31,11 @@ def records_get_callback(request, context):
if query_raw: if query_raw:
query = dict((k.replace("[]", ""), v) for k, v in query_raw.items()) query = dict((k.replace("[]", ""), v) for k, v in query_raw.items())
res = [d for d in records if d["type"].lower() in query["type"]] if query.get("type"):
res = [d for d in records if d["type"].lower() in query["type"]]
else:
res = []
return "{}".format(json.dumps(res)) return "{}".format(json.dumps(res))
else: else:
return "{}".format(json.dumps(records)) return "{}".format(json.dumps(records))