add postgres

konviva-events/app/boto3clients.py

@@ -1,13 +1,19 @@
 import os
+from typing import TYPE_CHECKING
 
 import boto3
 
+if TYPE_CHECKING:
+    from mypy_boto3_dynamodb.client import DynamoDBClient
+else:
+    DynamoDBClient = object
 
-def get_dynamodb_client():
+
+def get_dynamodb_client() -> DynamoDBClient:
     if os.getenv('AWS_LAMBDA_FUNCTION_NAME'):
         return boto3.client('dynamodb')
 
     return boto3.client('dynamodb', endpoint_url='http://127.0.0.1:8000')
 
 
-dynamodb_client = get_dynamodb_client()
+dynamodb_client: DynamoDBClient = get_dynamodb_client()

konviva-events/app/events/create_user.py

@@ -39,7 +39,7 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
         user_id = glom(result, '0.IDUsuario')
 
         if not result:
-            raise UserNotFoundError()
+            raise UserNotFoundError('User not found')
     except Exception:
         raise
 
@@ -47,7 +47,7 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
         transact.update(
             key=KeyPair(new_image['id'], '0'),
             update_expr='SET metadata__konviva_user_id = :user_id, \
-                        updated_at = :updated_at',
+                             updated_at = :updated_at',
             expr_attr_values={
                 ':user_id': user_id,
                 ':updated_at': now_,

konviva-events/app/events/enroll.py

@@ -41,6 +41,7 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
                 ':enrollment_id': enrollment_id,
             },
         )
+        # To relate with the enrollment
         transact.put(
             item={
                 'id': 'konviva',

konviva-events/app/konviva.py

@@ -26,7 +26,7 @@ class EmailAlreadyExistsError(KonvivaError):
     pass
 
 
-def _raise_konviva_errors_if_any(data: dict = {}) -> None:
+def _raise_errors_if_any(data: dict = {}) -> None:
     """Raises specific exceptions if Konviva API returned errors in the response body"""
     errors = glom(data, 'errors', default=None)
     if not errors:
@@ -72,7 +72,7 @@ def create_user(
     )
     r.raise_for_status()
     data = r.json()
-    _raise_konviva_errors_if_any(data)
+    _raise_errors_if_any(data)
 
     return int(data.get('IDUsuario'))
 
@@ -93,7 +93,7 @@ def update_user(id: str, **kwargs) -> dict:
     )
     r.raise_for_status()
     data = r.json()
-    _raise_konviva_errors_if_any(data)
+    _raise_errors_if_any(data)
 
     return data
 
@@ -126,7 +126,7 @@ def _post_enrollment(json: dict) -> dict:
     )
     r.raise_for_status()
     data = r.json()
-    _raise_konviva_errors_if_any(data)
+    _raise_errors_if_any(data)
 
     return data