Indeed disposition sync api

api/auth/__init__.py

@@ -0,0 +1,5 @@
+# HTK Imports
+from htk.api.auth.requests import HTTPBearerAuth
+
+
+__all__ = [HTTPBearerAuth]

api/auth/requests.py

@@ -0,0 +1,11 @@
+# Third Party (PyPI) Imports
+from requests.auth import AuthBase
+
+
+class HTTPBearerAuth(AuthBase):
+    def __init__(self, token):
+        self.token = token
+
+    def __call__(self, r):
+        r.headers['Authorization'] = f'Bearer {self.token}'
+        return r

lib/indeed/__init__.py

@@ -0,0 +1,5 @@
+# HTK Imports
+from htk.lib.indeed.api import IndeedDispositionSyncAPI
+
+
+__all__ = [IndeedDispositionSyncAPI]

lib/indeed/api.py

@@ -0,0 +1,164 @@
+# Python Standard Library Imports
+import time
+
+# Third Party (PyPI) Imports
+import requests
+import rollbar
+from requests.auth import HTTPBasicAuth
+
+# HTK Imports
+from htk.api.auth import HTTPBearerAuth
+from htk.lib.indeed.constants import (
+    INDEED_DISPOSITION_AUTH_TOKEN_URL,
+    INDEED_DISPOSITION_GRAPHQL_API_URL,
+)
+from htk.utils import chunks
+
+
+class IndeedDispositionSyncAPI(object):
+    """API to post the disposition status of applications received through Indeed
+
+    Indeed uses GraphQL API to update disposition status of applications
+    Ref: https://docs.indeed.com/disposition-sync-api/
+    """
+
+    def __init__(self, client_id, client_secret, dispositions):
+        self.client_id = client_id
+        self.client_secret = client_secret
+        self.dispositions = dispositions
+        self.token_expires_at = 0
+        self.access_token = None
+
+    def request_post(self, url, **kwargs):
+        response = requests.post(url, **kwargs)
+        status_code = response.status_code
+        response_json = response.json()
+
+        return response_json, status_code
+
+    def generate_access_token(self):
+        """
+        Generate access token using app credentials
+
+        Generated token is valid for one hour(3600 seconds)
+        Ref: https://docs.indeed.com/getstarted/register-app-and-call-apis#step-5-refresh-your-access-token
+        """
+        auth = HTTPBasicAuth(self.client_id, self.client_secret)
+        headers = {
+            'content-type': 'application/x-www-form-urlencoded',
+        }
+        data = {
+            'grant_type': 'client_credentials',
+            'scope': 'employer_access',
+        }
+        response_json, status_code = self.request_post(
+            INDEED_DISPOSITION_AUTH_TOKEN_URL,
+            auth=auth,
+            headers=headers,
+            data=data
+        )
+
+        if status_code == 200:
+            self.access_token = response_json.get('access_token')
+            self.token_expires_at = time.time() + int(
+                response_json.get('expires_in')
+            )
+        else:
+            self.access_token = None
+            self.token_expires_at = None
+            rollbar._report_message(
+                'Indeed disposition sync access token generation failed',
+                extra_data={
+                    'status_code': status_code,
+                    'response': response_json,
+                },
+            )
+
+    def get_access_token(self):
+        """
+        Returns access token
+
+        Access token is valid for one hour(3600 seconds)
+        Re-generate the access token when it expires
+        """
+        if self.token_expires_at <= time.time():
+            self.generate_access_token()
+        return self.access_token
+
+    def handle_response(self, response_json, status_code):
+        if status_code == 200:
+            if 'data' in response_json:
+                partner_disposition = response_json.get('data').get(
+                    'partnerDisposition'
+                )
+                partner_disposition_sent = partner_disposition.get('send')
+                rollbar.report_message(
+                    'Indeed disposition status update succeeded',
+                    level='info',
+                    extra_data={
+                        'total_dispositions_succeeded': partner_disposition_sent.get(
+                            'numberGoodDispositions'
+                        ),
+                        'response_json': response_json,
+                    },
+                )
+
+                failed_dispositions = partner_disposition_sent.get(
+                    'failedDispositions'
+                )
+                if failed_dispositions:
+                    rollbar.report_message(
+                        'Indeed disposition status update failed',
+                        extra_data={
+                            'failed_dispositions': failed_dispositions,
+                            'response_json': response_json,
+                        },
+                    )
+            if 'errors' in response_json:
+                rollbar.report_message(
+                    'Error updating Indeed disposition status',
+                    extra_data={
+                        'errors': response_json.get('errors'),
+                        'response_json': response_json,
+                    },
+                )
+        else:
+            rollbar.report_message(
+                'Request to update Indeed disposition status rejected',
+                extra_data={
+                    'status_code': status_code,
+                    'response': response_json,
+                },
+            )
+
+    def sync(self):
+        for chunk in chunks(self.dispositions, 25):
+            mutation = """
+            mutation Send($chunk: SendPartnerDispositionInput!) {
+                partnerDisposition {
+                    send(input: $chunk) {
+                        numberGoodDispositions
+                        failedDispositions {
+                            identifiedBy {
+                                indeedApplyID
+                            }
+                            rationale
+                        }
+                    }
+                }
+            }
+            """
+            variables = {
+                'chunk': {
+                    "dispositions": chunk,
+                }
+            }
+            access_token = self.get_access_token()
+            auth = HTTPBearerAuth(access_token)
+
+            response_json, status_code = self.request_post(
+                INDEED_DISPOSITION_GRAPHQL_API_URL,
+                auth=auth,
+                json={'query': mutation, 'variables': variables},
+            )
+            self.handle_response(response_json, status_code)

lib/indeed/constants.py

@@ -0,0 +1,2 @@
+INDEED_DISPOSITION_AUTH_TOKEN_URL = 'https://apis.indeed.com/oauth/v2/tokens'
+INDEED_DISPOSITION_GRAPHQL_API_URL = 'https://apis.indeed.com/graphql'