fix

http-api/app/routes/enrollments/__init__.py

@@ -0,0 +1,99 @@
+import json
+
+from aws_lambda_powertools.event_handler.api_gateway import Router
+from elasticsearch import Elasticsearch
+from layercake.dynamodb import (
+    DynamoDBCollection,
+    DynamoDBPersistenceLayer,
+    KeyPair,
+    SortKey,
+    TransactKey,
+)
+from pydantic import UUID4, BaseModel
+
+from boto3clients import dynamodb_client
+from conf import ELASTIC_CONN, ENROLLMENT_TABLE, USER_TABLE
+import elastic
+from middlewares.audit_log_middleware import AuditLogMiddleware
+from middlewares.authentication_middleware import User
+from rules.enrollment import set_status_as_canceled
+
+from .vacancies import router as vacancies
+
+__all__ = ['vacancies']
+
+
+router = Router()
+elastic_client = Elasticsearch(**ELASTIC_CONN)
+enrollment_layer = DynamoDBPersistenceLayer(ENROLLMENT_TABLE, dynamodb_client)
+enrollment_collect = DynamoDBCollection(enrollment_layer)
+user_layer = DynamoDBPersistenceLayer(USER_TABLE, dynamodb_client)
+user_collect = DynamoDBCollection(user_layer)
+
+
+@router.get('/', compress=True, tags=['Enrollment'])
+def get_enrollments():
+    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=ENROLLMENT_TABLE,
+        page_size=int(page_size),
+        query=json.loads(query),
+        elastic_client=elastic_client,
+    )
+
+
+@router.get('/<id>', compress=True, tags=['Enrollment'])
+def get_enrollment(id: str):
+    return enrollment_collect.get_items(
+        TransactKey(id)
+        + SortKey('0')
+        + SortKey('started_date')
+        + SortKey('finished_date')
+        + SortKey('failed_date')
+        + SortKey('canceled_date')
+        + SortKey('archived_date')
+        + SortKey('cancel_policy')
+        + SortKey('parent_vacancy', path_spec='vacancy')
+        + SortKey('lock', path_spec='hash')
+        + SortKey('author')
+        + SortKey('tenant')
+        + SortKey('cert')
+    )
+
+
+class Cancel(BaseModel):
+    id: UUID4 | str
+    lock_hash: str
+    course: dict = {}
+    vacancy: dict = {}
+
+
+@router.patch(
+    '/<id>/cancel',
+    compress=True,
+    tags=['Enrollment'],
+    middlewares=[
+        AuditLogMiddleware('ENROLLMENT_CANCEL', user_collect, ('id', 'course'))
+    ],
+)
+def cancel(id: str, payload: Cancel):
+    user: User = router.context['user']
+
+    set_status_as_canceled(
+        id,
+        lock_hash=payload.lock_hash,
+        author=user.model_dump(),  # type: ignore
+        course=payload.course,  # type: ignore
+        vacancy_key=KeyPair.parse_obj(payload.vacancy),
+        persistence_layer=enrollment_layer,
+    )
+
+    return payload
+
+
+@router.post('/', compress=True, tags=['Enrollment'])
+def enroll():
+    return {}

http-api/app/routes/enrollments/vacancies.py

@@ -0,0 +1,37 @@
+from aws_lambda_powertools.event_handler.api_gateway import Router
+from layercake.dynamodb import (
+    ComposeKey,
+    DynamoDBCollection,
+    DynamoDBPersistenceLayer,
+    PartitionKey,
+)
+
+from boto3clients import dynamodb_client
+from conf import (
+    ENROLLMENT_TABLE,
+    USER_TABLE,
+)
+from middlewares import Tenant, TenantMiddleware
+
+router = Router()
+
+enrollment_layer = DynamoDBPersistenceLayer(ENROLLMENT_TABLE, dynamodb_client)
+user_layer = DynamoDBPersistenceLayer(USER_TABLE, dynamodb_client)
+user_collect = DynamoDBCollection(user_layer)
+enrollment_collect = DynamoDBCollection(enrollment_layer)
+
+
+@router.get(
+    '/vacancies',
+    compress=True,
+    tags=['Enrollment'],
+    middlewares=[
+        TenantMiddleware(user_collect),
+    ],
+)
+def get_vacancies():
+    tenant: Tenant = router.context['tenant']
+
+    return enrollment_collect.query(
+        PartitionKey(ComposeKey(str(tenant.id), prefix='vacancies'))
+    )