From 53e500a56377ae2b8f92343c2ba4814bab707098 Mon Sep 17 00:00:00 2001
From: Rick Ross <rick.ross@allianceforge.net>
Date: Mon, 17 Jun 2024 10:34:37 -0700
Subject: [PATCH] Added User class to the library

---
 lib_af_api/__init__.py |   1 +
 lib_af_api/user.py     | 155 +++++++++++++++++++++++++++++++++++++++++
 2 files changed, 156 insertions(+)
 create mode 100644 lib_af_api/user.py

diff --git a/lib_af_api/__init__.py b/lib_af_api/__init__.py
index b5ab504..b03b595 100644
--- a/lib_af_api/__init__.py
+++ b/lib_af_api/__init__.py
@@ -11,3 +11,4 @@ from lib_af_api.microsite import Microsite
 from lib_af_api.token import Token
 from lib_af_api.voterset import VoterSet
 from lib_af_api.voter import Voter
+from lib_af_api.user import User
diff --git a/lib_af_api/user.py b/lib_af_api/user.py
new file mode 100644
index 0000000..f6f481c
--- /dev/null
+++ b/lib_af_api/user.py
@@ -0,0 +1,155 @@
+"""
+User Class object
+"""
+
+import logging
+import requests
+
+
+class User():
+    """User class"""
+
+    def __init__(self, base_url, jwt):
+        """
+        __init__ Init User
+
+        :param base_url: Base URL for User API
+        :param jwt: JWT Authentication for User API
+        """
+
+        self.base_url = base_url
+        self.jwt = jwt
+
+
+    def get_all(self, args = None):
+        """
+        get_all Get the users
+
+        :param args: Dict of args to add as URL params
+        """
+
+        # Set the headers for the request
+        headers = {
+            'Content-Type': 'application/json',
+            'Accept': 'application/json',
+            'Authorization': f'Bearer {self.jwt}'
+        }
+
+        if not isinstance(args, dict):
+            args = {}
+
+        # Run the GET
+        response = requests.get(
+            f'{self.base_url}/api/v1/users',
+            params=args,
+            headers=headers,
+            timeout=300
+        )
+
+        # Good response. Return Events
+        if response.ok:
+            return response.json()
+
+        # Log error and return null
+        logging.error("[AFUSER GET ALL] [%s]", response.text)
+        return None
+
+
+    def get(self, user_key):
+        """
+        get Get the user
+
+        :param user_key: The User key to get 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.get(
+            f'{self.base_url}/api/v1/users/{user_key}',
+            headers=headers,
+            timeout=300
+        )
+
+        # Good response. Return User
+        if response.ok:
+            return response.json()
+
+        # Log error and return null
+        logging.error("[AFUSER GET] [%s]", response.text)
+        return None
+
+
+    def patch(self, user_key, patch_data):
+        """
+        patch Patch the User
+
+        :param user_key: The User key to update with API
+        :param patch_data: The data to update the User
+        """
+
+        # Set the headers for the request
+        headers = {
+            'Content-Type': 'application/json',
+            'Accept': 'application/json',
+            'Authorization': f'Bearer {self.jwt}'
+        }
+
+        # Set the key to the patch data
+        patch_data['key'] = user_key
+
+        # Run the GET
+        response = requests.patch(
+            f'{self.base_url}/api/v1/users/{user_key}',
+            headers=headers,
+            timeout=300,
+            json=patch_data
+        )
+
+        # Good response. Return User
+        if response.ok:
+            return response.json()
+
+        # Log error and return null
+        logging.error("[AFUSER UPDATE] [%s]", response.text)
+        return None
+
+
+    def change_role(self, user_key, role_type, role_key, action):
+        """
+        patch Patch the User Roles
+
+        :param user_key: The User key to update with API
+        :param key_type: The type of key provided (either campaign, business, or company)
+        :param role_key: The Campaign / Business key to add to roles
+        :param action: The action to take (either add or remove)
+        """
+
+        # verify parameters are not null
+        if user_key is None or role_key is None:
+            return None
+        if role_type not in ('campaign', 'business', 'company'):
+            return None
+        if action not in ('add', 'remove'):
+            return None
+
+        new_role = f"{role_type}_key:{role_key}"
+
+        # get the user record
+        user = self.get(user_key)
+
+        # check if role exists
+        if new_role in user['roles']:
+            if action == "remove":
+                user['roles'].remove(new_role)
+        else:
+            if action == "add":
+                user['roles'].append(new_role)
+
+        # patch the user record and return the results
+        return self.patch(user_key, {'roles': user['roles']})