add batch

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

@@ -5,22 +5,19 @@ from elasticsearch import Elasticsearch
 from layercake.dynamodb import (
     DynamoDBCollection,
     DynamoDBPersistenceLayer,
-    KeyPair,
     SortKey,
     TransactKey,
 )
-from pydantic import UUID4, BaseModel
 
+import elastic
 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 .cancel import router as cancel
+from .enroll import router as enroll
 from .vacancies import router as vacancies
 
-__all__ = ['vacancies']
+__all__ = ['vacancies', 'cancel', 'enroll']
 
 
 router = Router()
@@ -62,38 +59,3 @@ def get_enrollment(id: str):
         + 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 {}