diff --git a/lib_af_api/voter.py b/lib_af_api/voter.py index b4526db..205ae26 100644 --- a/lib_af_api/voter.py +++ b/lib_af_api/voter.py @@ -21,7 +21,7 @@ class Voter(): self.jwt = jwt - def get_all(self, args = {}): + def get_all(self, args = None): """ get Get the voter @@ -37,9 +37,10 @@ class Voter(): url_args = "" prefix = "" - for k in args: - url_args += f"{prefix}{k}={args[k]}" - prefix = "&" + if isinstance(args, dict): + for k in args: + url_args += f"{prefix}{k}={args[k]}" + prefix = "&" # Run the GET response = requests.get( @@ -151,3 +152,33 @@ class Voter(): # Log error and retunr null logging.error("[AFVOTER CREATE] [%s]", response.text) return None + + + def delete(self, voter_key): + """ + delete Delete the Voter + + :param voter_key: The Voter key to delete with API + """ + + # Set the headers for the request + headers = { + 'Content-Type': 'application/json', + 'Accept': 'application/json', + 'Authorization': f'Bearer {self.jwt}' + } + + # Run the GET + response = requests.delete( + f'{self.base_url}/api/v1/voters/{voter_key}', + headers=headers, + timeout=300 + ) + + # Good response. Return Voter + if response.ok: + return response.json() + + # Log error and retunr null + logging.error("[AFVOTER DELETE] [%s]", response.text) + return None diff --git a/setup.py b/setup.py index 8fb3b70..f178167 100644 --- a/setup.py +++ b/setup.py @@ -9,7 +9,7 @@ with open('README.md', 'r') as f: setup( name='lib_af_api', - version='0.1.3', + version='0.1.4', author='', author_email='', description='',