85 lines
2.5 KiB
Python
85 lines
2.5 KiB
Python
import json
|
|
from http import HTTPStatus
|
|
from typing import Annotated
|
|
|
|
import boto3
|
|
from aws_lambda_powertools.event_handler.api_gateway import (
|
|
Response,
|
|
Router,
|
|
)
|
|
from elasticsearch import Elasticsearch
|
|
from pydantic import UUID4, BaseModel, StringConstraints
|
|
from layercake.dynamodb import DynamoDBPersistenceLayer
|
|
from dynamodb import KeyLoc, get_records
|
|
|
|
import elastic
|
|
from http_models import RecordResponse, SearchResponse
|
|
from settings import ELASTIC_CONN, USER_TABLE
|
|
|
|
router = Router()
|
|
dynamodb_client = boto3.client('dynamodb')
|
|
user_layer = DynamoDBPersistenceLayer(USER_TABLE, dynamodb_client)
|
|
elastic_client = Elasticsearch(**ELASTIC_CONN)
|
|
|
|
|
|
@router.get('/', compress=True, tags=['User'], summary='Get users')
|
|
def get_users() -> SearchResponse:
|
|
event = router.current_event
|
|
query = event.get_query_string_value('query', '{}')
|
|
page_size = event.get_query_string_value('page_size', '25')
|
|
|
|
return elastic.search( # type: ignore
|
|
index=USER_TABLE,
|
|
page_size=int(page_size),
|
|
query=json.loads(query),
|
|
elastic_client=elastic_client,
|
|
)
|
|
|
|
|
|
class ResetPasswordPayload(BaseModel):
|
|
cognito_sub: UUID4
|
|
new_password: Annotated[str, StringConstraints(min_length=6)]
|
|
|
|
|
|
@router.patch('/<id>', compress=True, tags=['User'])
|
|
def patch_reset(id: str, payload: ResetPasswordPayload):
|
|
return Response(status_code=HTTPStatus.OK)
|
|
|
|
|
|
@router.get('/<id>/emails', compress=True, tags=['User'], summary='Get user emails')
|
|
def get_emails(id: str) -> RecordResponse:
|
|
start_key = router.current_event.get_query_string_value('start_key', None)
|
|
|
|
return get_records( # type: ignore
|
|
pk=id,
|
|
prefix='emails',
|
|
start_key=start_key,
|
|
persistence_layer=user_layer,
|
|
)
|
|
|
|
|
|
@router.get('/<id>/logs', compress=True, tags=['User'], summary='Get user logs')
|
|
def get_logs(id: str) -> RecordResponse:
|
|
start_key = router.current_event.get_query_string_value('start_key', None)
|
|
|
|
return get_records( # type: ignore
|
|
pk=id,
|
|
prefix='log',
|
|
delimiter=':',
|
|
keyloc_prefix=KeyLoc.PK,
|
|
start_key=start_key,
|
|
persistence_layer=user_layer,
|
|
)
|
|
|
|
|
|
@router.get('/<id>/orgs', compress=True, tags=['User'], summary='Get user orgs')
|
|
def get_orgs(id: str) -> RecordResponse:
|
|
start_key = router.current_event.get_query_string_value('start_key', None)
|
|
|
|
return get_records( # type: ignore
|
|
pk=id,
|
|
prefix='orgs',
|
|
start_key=start_key,
|
|
persistence_layer=user_layer,
|
|
)
|