fix: graceful domain_set as str

Fediseer · Aug 9, 2023 · a570a2c · a570a2c
1 parent 8ac2db4
commit a570a2c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/pythonseer/censure.py b/pythonseer/censure.py
@@ -86,6 +86,9 @@ def get_given(
             domain_csv = self._requestor.home_domain
             if not domain_csv:
                 raise Exception("Must provide a domain or login to GET /censures_given/ endpoint")
+        # Handle sending the domain name as str gracefully
+        elif type(domain_set) is str:
+            domain_csv = domain_set
         else:
             domain_csv = ','.join(domain_set)
         endpoint =  f"/censures_given/{domain_csv}{format.get_query('?')}"

diff --git a/pythonseer/endorsement.py b/pythonseer/endorsement.py
@@ -86,6 +86,9 @@ def get_given(
             domain_csv = self._requestor.home_domain
             if not domain_csv:
                 raise Exception("Must provide a domain or login to GET /approvals/ endpoint")
+        # Handle sending the domain name as str gracefully
+        elif type(domain_set) is str:
+            domain_csv = domain_set
         else:
             domain_csv = ','.join(domain_set)
         endpoint =  f"/approvals/{domain_csv}{format.get_query('?')}"