finish register

id.saladeaula.digital/app/app.py

@@ -44,6 +44,8 @@ def health():
 
 @app.exception_handler(ServiceError)
 def exc_error(exc: ServiceError):
+    logger.exception(exc)
+
     return Response(
         body={
             'type': type(exc).__name__,

id.saladeaula.digital/app/boto3clients.py

@@ -4,9 +4,11 @@ from typing import TYPE_CHECKING
 import boto3
 
 if TYPE_CHECKING:
+    from mypy_boto3_cognito_idp import CognitoIdentityProviderClient
     from mypy_boto3_dynamodb.client import DynamoDBClient
 else:
     DynamoDBClient = object
+    CognitoIdentityProviderClient = object
 
 
 def get_dynamodb_client() -> DynamoDBClient:
@@ -17,3 +19,4 @@ def get_dynamodb_client() -> DynamoDBClient:
 
 
 dynamodb_client: DynamoDBClient = get_dynamodb_client()
+idp_client: CognitoIdentityProviderClient = boto3.client('cognito-idp')

id.saladeaula.digital/app/routes/authentication.py

@@ -2,7 +2,6 @@ from http import HTTPStatus
 from typing import Annotated
 from uuid import uuid4
 
-import boto3
 from aws_lambda_powertools.event_handler import (
     Response,
 )
@@ -17,7 +16,7 @@ from layercake.dateutils import now, ttl
 from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair, SortKey
 from passlib.hash import pbkdf2_sha256
 
-from boto3clients import dynamodb_client
+from boto3clients import dynamodb_client, idp_client
 from config import (
     OAUTH2_TABLE,
     SESSION_EXPIRES_IN,
@@ -25,7 +24,6 @@ from config import (
 
 router = Router()
 dyn = DynamoDBPersistenceLayer(OAUTH2_TABLE, dynamodb_client)
-idp = boto3.client('cognito-idp')
 
 
 class InvalidCredentialsError(UnauthorizedError): ...
@@ -125,7 +123,7 @@ def _get_idp_user(
     ).digest()
 
     try:
-        idp.initiate_auth(
+        idp_client.initiate_auth(
             AuthFlow='USER_PASSWORD_AUTH',
             AuthParameters={
                 'USERNAME': username,
@@ -155,7 +153,9 @@ def new_session(user_id: str) -> str:
     exp = ttl(start_dt=now_, seconds=SESSION_EXPIRES_IN)
 
     with dyn.transact_writer() as transact:
-        transact.delete(key=KeyPair(user_id, 'FAILED_ATTEMPTS'))
+        transact.delete(
+            key=KeyPair(user_id, 'FAILED_ATTEMPTS'),
+        )
         transact.update(
             key=KeyPair(user_id, '0'),
             # Post-migration (users): uncomment the following line

id.saladeaula.digital/app/routes/register.py

@@ -3,9 +3,11 @@ from http import HTTPStatus
 from typing import Annotated
 from uuid import uuid4
 
+from aws_lambda_powertools.event_handler import content_types
 from aws_lambda_powertools.event_handler.api_gateway import Response, Router
 from aws_lambda_powertools.event_handler.exceptions import NotFoundError, ServiceError
 from aws_lambda_powertools.event_handler.openapi.params import Body
+from aws_lambda_powertools.shared.cookies import Cookie
 from layercake.dateutils import now, ttl
 from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair
 from layercake.extra_types import CpfStr, NameStr
@@ -14,7 +16,9 @@ from passlib.hash import pbkdf2_sha256
 from pydantic import UUID4, EmailStr
 
 from boto3clients import dynamodb_client
-from config import OAUTH2_TABLE
+from config import OAUTH2_TABLE, SESSION_EXPIRES_IN
+
+from .authentication import new_session
 
 router = Router()
 dyn = DynamoDBPersistenceLayer(OAUTH2_TABLE, dynamodb_client)
@@ -68,15 +72,36 @@ def register(
         )
 
         return Response(
+            content_type=content_types.APPLICATION_JSON,
             status_code=HTTPStatus.OK,
+            compress=True,
             body=asdict(new_user),
+            cookies=[
+                _cookie(existing['id']),
+            ],
         )
 
     _create_user(user=new_user, password=password)
 
     return Response(
+        content_type=content_types.APPLICATION_JSON,
         status_code=HTTPStatus.CREATED,
+        compress=True,
         body=asdict(new_user),
+        cookies=[
+            _cookie(new_user.id),
+        ],
+    )
+
+
+def _cookie(user_id: str) -> Cookie:
+    return Cookie(
+        name='SID',
+        value=new_session(user_id),
+        http_only=True,
+        secure=True,
+        same_site=None,
+        max_age=SESSION_EXPIRES_IN,
     )
 
 

id.saladeaula.digital/pyproject.toml

@@ -12,7 +12,7 @@ dev = [
     "pytest>=8.4.1",
     "ruff>=0.12.1",
     "pytest-cov>=6.2.1",
-    "boto3-stubs[essential]>=1.40.44",
+    "boto3-stubs[cognito-idp,essential]>=1.40.44",
 ]
 
 

id.saladeaula.digital/tests/routes/test_register.py

@@ -23,6 +23,7 @@ def test_preexisting_user(
                 'cpf': '07879819908',
                 'password': 'Led@Zepellin',
                 'email': 'sergio@somosbeta.com.br',
+                'never_logged': 'true',
             },
         ),
         lambda_context,
@@ -80,6 +81,7 @@ def test_preexisting_user_update_email(
         lambda_context,
     )
     assert r['statusCode'] == HTTPStatus.OK
+    assert 'cookies' in r
 
     user = dynamodb_persistence_layer.collection.get_items(
         TransactKey(

id.saladeaula.digital/uv.lock

@@ -133,6 +133,9 @@ wheels = [
 ]
 
 [package.optional-dependencies]
+cognito-idp = [
+    { name = "mypy-boto3-cognito-idp" },
+]
 essential = [
     { name = "mypy-boto3-cloudformation" },
     { name = "mypy-boto3-dynamodb" },
@@ -423,7 +426,7 @@ dependencies = [
 
 [package.dev-dependencies]
 dev = [
-    { name = "boto3-stubs", extra = ["essential"] },
+    { name = "boto3-stubs", extra = ["cognito-idp", "essential"] },
     { name = "jsonlines" },
     { name = "pytest" },
     { name = "pytest-cov" },
@@ -435,7 +438,7 @@ requires-dist = [{ name = "layercake", directory = "../layercake" }]
 
 [package.metadata.requires-dev]
 dev = [
-    { name = "boto3-stubs", extras = ["essential"], specifier = ">=1.40.44" },
+    { name = "boto3-stubs", extras = ["cognito-idp", "essential"], specifier = ">=1.40.44" },
     { name = "jsonlines", specifier = ">=4.0.0" },
     { name = "pytest", specifier = ">=8.4.1" },
     { name = "pytest-cov", specifier = ">=6.2.1" },
@@ -591,6 +594,15 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/2e/38/12301080cc5004593b8593c0cc7404c13d702ac1c15d4e0ccfacd1f4f416/mypy_boto3_cloudformation-1.40.44-py3-none-any.whl", hash = "sha256:64c8fe58ab7661fbb0bdea07c7375d3ebc3875760140feb6ad8f591a08a22647", size = 69896, upload-time = "2025-10-02T20:31:56.896Z" },
 ]
 
+[[package]]
+name = "mypy-boto3-cognito-idp"
+version = "1.40.60"
+source = { registry = "https://pypi.org/simple" }
+sdist = { url = "https://files.pythonhosted.org/packages/b3/4e/57aeebe4c57a6f3345cffab1c7ae05756c08fef335bc72ce34308a534835/mypy_boto3_cognito_idp-1.40.60.tar.gz", hash = "sha256:4e06656f3954e193e4dff69042bc214db5ef575ebbb606a1b5fcb6dc3d6fd32e", size = 52148, upload-time = "2025-10-27T19:43:24.788Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/bd/d2/b42e487058fea7b43086b62150826a3aa2777f9a7fe0c9ed31e03bb05100/mypy_boto3_cognito_idp-1.40.60-py3-none-any.whl", hash = "sha256:afdb6c81676442d76be4c4bb63378758e13b970f9b366290ac2d1bfeda26f669", size = 57972, upload-time = "2025-10-27T19:43:22.069Z" },
+]
+
 [[package]]
 name = "mypy-boto3-dynamodb"
 version = "1.40.44"