add cancel

konviva-events/app/events/cancel.py

@@ -0,0 +1,30 @@
+from aws_lambda_powertools import Logger
+from aws_lambda_powertools.utilities.data_classes import (
+    EventBridgeEvent,
+    event_source,
+)
+from aws_lambda_powertools.utilities.typing import LambdaContext
+from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair
+
+import konviva
+from boto3clients import dynamodb_client
+from config import ENROLLMENT_TABLE
+
+logger = Logger(__name__)
+enrollment_layer = DynamoDBPersistenceLayer(ENROLLMENT_TABLE, dynamodb_client)
+
+
+@event_source(data_class=EventBridgeEvent)
+@logger.inject_lambda_context
+def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
+    new_image = event.detail['new_image']
+    data = enrollment_layer.get_item(KeyPair(new_image['id'], 'konviva'))
+
+    try:
+        result = konviva.cancel_enrollment(data['enrollment_id'])
+    except Exception as exc:
+        logger.exception(exc)
+        return False
+    else:
+        logger.info('Enrollment canceled', result=result)
+        return True

konviva-events/app/konviva.py

@@ -31,7 +31,7 @@ def create_user(
     name: str,
     email: str,
     cpf: str | None,
-) -> dict:
+) -> int:
     url = urlparse(KONVIVA_API_URL)._replace(
         path='/action/api/integrarUsuario',
         query='sendMail=false',
@@ -58,9 +58,10 @@ def create_user(
         },
     )
     r.raise_for_status()
+    data = r.json()
 
     # Because Konviva does not return the proper HTTP status code
-    if err := glom(r.json(), 'errors', default=None):
+    if err := glom(data, 'errors', default=None):
         err = err[0] if isinstance(err, list) else err
 
         if err == 'Login já existente':
@@ -68,7 +69,7 @@ def create_user(
         else:
             raise KonvivaError(err)
 
-    return r.json().get('IDUsuario')
+    return int(data.get('IDUsuario'))
 
 
 def update_user(id: str, **kwargs) -> dict:
@@ -86,12 +87,13 @@ def update_user(id: str, **kwargs) -> dict:
         },
     )
     r.raise_for_status()
+    data = r.json()
 
-    if err := glom(r.json(), 'errors', default=None):
+    if err := glom(data, 'errors', default=None):
         err = err[0] if isinstance(err, list) else err
         raise KonvivaError(err)
 
-    return r.json()
+    return data
 
 
 def get_users_by_email(email: str) -> list[dict]:
@@ -104,29 +106,48 @@ def get_users_by_email(email: str) -> list[dict]:
         headers=headers,
     )
     r.raise_for_status()
+    data = r.json()
 
-    if glom(r.json(), '0.errors', default=None):
+    if glom(data, '0.errors', default=None):
         return []
 
-    return r.json()
+    return data
 
 
-def enroll(user_id: str, class_id: str) -> str:
+def _post_enrollment(json: dict) -> dict:
     url = urlparse(KONVIVA_API_URL)._replace(path='/action/api/integrarMatricula')
 
     r = requests.post(
         url=url.geturl(),
         headers=headers,
-        json={
-            'IDUsuario': str(user_id),
-            'IDTurma': str(class_id),
-            'StatusMatricula': 'MATRICULADO',
-        },
+        json=json,
     )
     r.raise_for_status()
+    data = r.json()
 
-    if err := glom(r.json(), 'errors', default=None):
+    if err := glom(data, 'errors', default=None):
         err = err[0] if isinstance(err, list) else err
         raise KonvivaError(err)
 
-    return r.json().get('IDMatricula')
+    return data
+
+
+def enroll(user_id: str, class_id: str) -> int:
+    r = _post_enrollment(
+        {
+            'IDUsuario': int(user_id),
+            'IDTurma': int(class_id),
+            'StatusMatricula': 'MATRICULADO',
+        }
+    )
+
+    return int(r.get('IDMatricula'))  # type: ignore
+
+
+def cancel_enrollment(id: str) -> dict:
+    return _post_enrollment(
+        {
+            'IDMatricula': str(id),
+            'StatusMatricula': 'CANCELADO',
+        }
+    )