finish register

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