88 lines
2.4 KiB
Python
88 lines
2.4 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 layercake.dynamodb import (
|
|
ComposeKey,
|
|
DynamoDBCollection,
|
|
DynamoDBPersistenceLayer,
|
|
KeyPair,
|
|
PartitionKey,
|
|
)
|
|
from pydantic import UUID4, BaseModel, StringConstraints
|
|
|
|
import elastic
|
|
from models import User
|
|
from settings import ELASTIC_CONN, USER_TABLE
|
|
|
|
router = Router()
|
|
dynamodb_client = boto3.client('dynamodb')
|
|
user_layer = DynamoDBPersistenceLayer(USER_TABLE, dynamodb_client)
|
|
collect = DynamoDBCollection(user_layer)
|
|
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')
|
|
def post_user(payload: User):
|
|
return Response(status_code=HTTPStatus.CREATED)
|
|
|
|
|
|
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):
|
|
start_key = router.current_event.get_query_string_value('start_key', None)
|
|
|
|
return collect.get_items(
|
|
key=KeyPair(id, 'emails'),
|
|
start_key=start_key,
|
|
)
|
|
|
|
|
|
@router.get('/<id>/logs', compress=True, tags=['User'], summary='Get user logs')
|
|
def get_logs(id: str):
|
|
start_key = router.current_event.get_query_string_value('start_key', None)
|
|
|
|
return collect.get_items(
|
|
key=PartitionKey(ComposeKey(id, prefix='log', delimiter=':')),
|
|
start_key=start_key,
|
|
)
|
|
|
|
|
|
@router.get('/<id>/orgs', compress=True, tags=['User'], summary='Get user orgs')
|
|
def get_orgs(id: str):
|
|
start_key = router.current_event.get_query_string_value('start_key', None)
|
|
|
|
return collect.get_items(
|
|
key=KeyPair(id, 'orgs'),
|
|
start_key=start_key,
|
|
)
|