update
This commit is contained in:
@@ -1,4 +1,5 @@
|
||||
import re
|
||||
import urllib.parse as urllib
|
||||
|
||||
OPERATORS = [
|
||||
'!=',
|
||||
@@ -41,7 +42,7 @@ def parse_condition(condition: str) -> dict[str, str] | None:
|
||||
|
||||
|
||||
def parse(s: str) -> list[dict]:
|
||||
filter_expr = re.sub(r'\s+', ' ', s).strip()
|
||||
filter_expr = re.sub(r'\s+', ' ', urllib.unquote(s)).strip()
|
||||
|
||||
if filter_expr == '':
|
||||
return []
|
||||
@@ -59,4 +60,4 @@ def parse(s: str) -> list[dict]:
|
||||
|
||||
|
||||
def encode(conditions: list[dict]):
|
||||
return ' AND '.join(f'{c["attr"]} {c["op"]} {c["value"]}' for c in conditions if c)
|
||||
return ' AND '.join(' '.join(c.values()).strip() for c in conditions if c)
|
||||
|
||||
@@ -34,12 +34,12 @@ def get_orders():
|
||||
sort = event.get_query_string_value('sort', 'create_date:desc')
|
||||
page = int(event.get_query_string_value('page', '1'))
|
||||
hits_per_page = int(event.get_query_string_value('hitsPerPage', '25'))
|
||||
filter_ = meili.parse(parse.unquote(event.get_query_string_value('filter', '')))
|
||||
filter_ = meili.parse(event.get_query_string_value('filter', ''))
|
||||
|
||||
if tenant.id != '*':
|
||||
filter_ = [
|
||||
{
|
||||
'attr': 'tenant',
|
||||
'attr': 'tenant_id',
|
||||
'op': '=',
|
||||
'value': tenant.id,
|
||||
},
|
||||
|
||||
@@ -1,4 +1,4 @@
|
||||
import json
|
||||
import urllib.parse as parse
|
||||
from http import HTTPStatus
|
||||
from typing import Annotated
|
||||
|
||||
@@ -6,7 +6,6 @@ from aws_lambda_powertools.event_handler.api_gateway import Router
|
||||
from aws_lambda_powertools.event_handler.exceptions import (
|
||||
BadRequestError as PowertoolsBadRequestError,
|
||||
)
|
||||
from elasticsearch import Elasticsearch
|
||||
from layercake.dynamodb import (
|
||||
DynamoDBCollection,
|
||||
DynamoDBPersistenceLayer,
|
||||
@@ -15,14 +14,15 @@ from layercake.dynamodb import (
|
||||
TransactKey,
|
||||
)
|
||||
from layercake.extra_types import CpfStr, NameStr
|
||||
from meilisearch import Client as Meilisearch
|
||||
from pydantic import UUID4, BaseModel, StringConstraints
|
||||
|
||||
import cognito
|
||||
import elastic
|
||||
import meili
|
||||
from api_gateway import JSONResponse
|
||||
from boto3clients import dynamodb_client, idp_client
|
||||
from config import ELASTIC_CONN, USER_POOOL_ID, USER_TABLE
|
||||
from middlewares import AuditLogMiddleware
|
||||
from config import MEILISEARCH_API_KEY, MEILISEARCH_HOST, USER_POOOL_ID, USER_TABLE
|
||||
from middlewares import AuditLogMiddleware, Tenant, TenantMiddleware
|
||||
from models import User
|
||||
from rules.user import update_user
|
||||
|
||||
@@ -40,20 +40,45 @@ class BadRequestError(MissingError, PowertoolsBadRequestError):
|
||||
router = Router()
|
||||
user_layer = DynamoDBPersistenceLayer(USER_TABLE, dynamodb_client)
|
||||
user_collect = DynamoDBCollection(user_layer, exc_cls=BadRequestError)
|
||||
elastic_client = Elasticsearch(**ELASTIC_CONN)
|
||||
meili_client = Meilisearch(MEILISEARCH_HOST, MEILISEARCH_API_KEY) # noqa: F821
|
||||
|
||||
|
||||
@router.get('/', compress=True, tags=['User'], summary='Get users')
|
||||
@router.get(
|
||||
'/',
|
||||
compress=True,
|
||||
tags=['User'],
|
||||
middlewares=[
|
||||
TenantMiddleware(user_layer.collection),
|
||||
],
|
||||
summary='Get users',
|
||||
)
|
||||
def get_users():
|
||||
tenant: Tenant = router.context['tenant']
|
||||
event = router.current_event
|
||||
query = event.get_query_string_value('query', '{}')
|
||||
page_size = event.get_query_string_value('page_size', '25')
|
||||
query = parse.unquote(event.get_query_string_value('q', ''))
|
||||
sort = event.get_query_string_value('sort', 'create_date:desc')
|
||||
page = int(event.get_query_string_value('page', '1'))
|
||||
hits_per_page = int(event.get_query_string_value('hitsPerPage', '25'))
|
||||
filter_ = meili.parse(event.get_query_string_value('filter', ''))
|
||||
|
||||
return elastic.search(
|
||||
index=USER_TABLE,
|
||||
page_size=int(page_size),
|
||||
query=json.loads(query),
|
||||
elastic_client=elastic_client,
|
||||
if tenant.id != '*':
|
||||
filter_ = [
|
||||
{
|
||||
'attr': 'tenant_id',
|
||||
'op': '=',
|
||||
'value': tenant.id,
|
||||
},
|
||||
] + filter_
|
||||
|
||||
return meili_client.index(USER_TABLE).search(
|
||||
query,
|
||||
{
|
||||
'sort': [sort],
|
||||
'locales': ['pt'],
|
||||
'page': page,
|
||||
'hitsPerPage': hits_per_page,
|
||||
'filter': meili.encode(filter_),
|
||||
},
|
||||
)
|
||||
|
||||
|
||||
|
||||
@@ -16,8 +16,8 @@ def create_course(
|
||||
transact.put(
|
||||
item={
|
||||
'sk': '0',
|
||||
'tenant': org.id,
|
||||
'create_date': now_,
|
||||
'tenant_id': {org.id},
|
||||
'created_at': now_,
|
||||
**course.model_dump(),
|
||||
}
|
||||
)
|
||||
@@ -27,7 +27,7 @@ def create_course(
|
||||
'sk': 'metadata#tenant',
|
||||
'org_id': org.id,
|
||||
'name': org.name,
|
||||
'create_date': now_,
|
||||
'created_at': now_,
|
||||
}
|
||||
)
|
||||
|
||||
@@ -46,7 +46,7 @@ def update_course(
|
||||
transact.update(
|
||||
key=KeyPair(id, '0'),
|
||||
update_expr='SET #name = :name, access_period = :access_period, \
|
||||
cert = :cert, update_date = :update_date',
|
||||
cert = :cert, updated_at = :updated_at',
|
||||
expr_attr_names={
|
||||
'#name': 'name',
|
||||
},
|
||||
@@ -54,7 +54,7 @@ def update_course(
|
||||
':name': course.name,
|
||||
':cert': course.cert.model_dump() if course.cert else None,
|
||||
':access_period': course.access_period,
|
||||
':update_date': now_,
|
||||
':updated_at': now_,
|
||||
},
|
||||
cond_expr='attribute_exists(sk)',
|
||||
)
|
||||
|
||||
@@ -9,3 +9,17 @@ def test_parse():
|
||||
{'attr': 'payment_date', 'op': '>=', 'value': '2025-07-01'},
|
||||
{'attr': 'status', 'op': '=', 'value': 'PAID'},
|
||||
]
|
||||
|
||||
assert meili.parse('cnpj%20EXISTS') == [
|
||||
{'attr': 'cnpj', 'op': 'EXISTS', 'value': ''}
|
||||
]
|
||||
|
||||
|
||||
def test_encode():
|
||||
r = meili.encode(
|
||||
[
|
||||
{'attr': 'tenant', 'op': '=', 'value': '*'},
|
||||
{'attr': 'cnpj', 'op': 'EXISTS', 'value': ''},
|
||||
]
|
||||
)
|
||||
assert r == 'tenant = * AND cnpj EXISTS'
|
||||
|
||||
Reference in New Issue
Block a user