129 lines
3.3 KiB
Python
129 lines
3.3 KiB
Python
import json
|
|
from http import HTTPStatus
|
|
from typing import Annotated
|
|
|
|
from aws_lambda_powertools.event_handler.api_gateway import (
|
|
Response,
|
|
Router,
|
|
)
|
|
from aws_lambda_powertools.event_handler.exceptions import (
|
|
BadRequestError as PowertoolsBadRequestError,
|
|
)
|
|
from elasticsearch import Elasticsearch
|
|
from layercake.dynamodb import (
|
|
ComposeKey,
|
|
DynamoDBCollection,
|
|
DynamoDBPersistenceLayer,
|
|
KeyPair,
|
|
MissingError,
|
|
PartitionKey,
|
|
PrefixKey,
|
|
)
|
|
from pydantic import UUID4, BaseModel, StringConstraints
|
|
|
|
import cognito
|
|
import elastic
|
|
from boto3clients import dynamodb_client, idp_client
|
|
from middlewares import AuditLogMiddleware
|
|
from models import User
|
|
from settings import ELASTIC_CONN, USER_POOOL_ID, USER_TABLE
|
|
|
|
|
|
class BadRequestError(MissingError, PowertoolsBadRequestError): ...
|
|
|
|
|
|
router = Router()
|
|
user_layer = DynamoDBPersistenceLayer(USER_TABLE, dynamodb_client)
|
|
user_collect = DynamoDBCollection(user_layer, exception_cls=BadRequestError)
|
|
elastic_client = Elasticsearch(**ELASTIC_CONN)
|
|
|
|
|
|
@router.get('/', compress=True, tags=['User'], summary='Get users')
|
|
def get_users():
|
|
event = router.current_event
|
|
query = event.get_query_string_value('query', '{}')
|
|
page_size = event.get_query_string_value('page_size', '25')
|
|
|
|
return elastic.search(
|
|
index=USER_TABLE,
|
|
page_size=int(page_size),
|
|
query=json.loads(query),
|
|
elastic_client=elastic_client,
|
|
)
|
|
|
|
|
|
@router.post(
|
|
'/',
|
|
compress=True,
|
|
tags=['User'],
|
|
summary='Create user',
|
|
middlewares=[AuditLogMiddleware('USER_ADD', user_collect)],
|
|
)
|
|
def post_user(payload: User):
|
|
return Response(status_code=HTTPStatus.CREATED)
|
|
|
|
|
|
class NewPassword(BaseModel):
|
|
cognito_sub: UUID4
|
|
new_password: Annotated[str, StringConstraints(min_length=6)]
|
|
|
|
|
|
@router.patch('/<id>', compress=True, tags=['User'])
|
|
def patch_newpassword(id: str, payload: NewPassword):
|
|
return Response(status_code=HTTPStatus.OK)
|
|
|
|
|
|
@router.get('/<id>', compress=True, tags=['User'], summary='Get user')
|
|
def get_user(id: str):
|
|
return user_collect.get_item(KeyPair(id, '0'))
|
|
|
|
|
|
@router.get('/<id>/idp', compress=True, include_in_schema=False)
|
|
def get_idp(id: str):
|
|
return cognito.admin_get_user(
|
|
sub=id,
|
|
user_pool_id=USER_POOOL_ID,
|
|
idp_client=idp_client,
|
|
)
|
|
|
|
|
|
@router.get(
|
|
'/<id>/emails',
|
|
compress=True,
|
|
tags=['User'],
|
|
summary='Get user emails',
|
|
)
|
|
def get_emails(id: str):
|
|
return user_collect.query(
|
|
KeyPair(id, PrefixKey('emails')),
|
|
start_key=router.current_event.get_query_string_value('start_key', None),
|
|
)
|
|
|
|
|
|
@router.get(
|
|
'/<id>/logs',
|
|
compress=True,
|
|
tags=['User'],
|
|
summary='Get user logs',
|
|
)
|
|
def get_logs(id: str):
|
|
return user_collect.query(
|
|
# Post-migration: uncomment to enable PartitionKey with a composite key (id with `logs` prefix).
|
|
# PartitionKey(ComposeKey(id, prefix='logs')),
|
|
PartitionKey(ComposeKey(id, prefix='log', delimiter=':')),
|
|
start_key=router.current_event.get_query_string_value('start_key', None),
|
|
)
|
|
|
|
|
|
@router.get(
|
|
'/<id>/orgs',
|
|
compress=True,
|
|
tags=['User'],
|
|
summary='Get user orgs',
|
|
)
|
|
def get_orgs(id: str):
|
|
return user_collect.query(
|
|
KeyPair(id, PrefixKey('orgs')),
|
|
start_key=router.current_event.get_query_string_value('start_key', None),
|
|
)
|