unity-data-services/cumulus_lambda_functions/uds_api/system_admin_api.py at 90773fc507371c22a28310997d9c2de001c9ba16 · unity-sds/unity-data-services · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
from typing import Union

from cumulus_lambda_functions.cumulus_es_setup.es_setup import SetupESIndexAlias
from cumulus_lambda_functions.lib.lambda_logger_generator import LambdaLoggerGenerator
from cumulus_lambda_functions.uds_api.dapa.auth_crud import AuthCrud, AuthDeleteModel, AuthListModel, AuthAddModel
from cumulus_lambda_functions.uds_api.fast_api_utils import FastApiUtils
from cumulus_lambda_functions.uds_api.web_service_constants import WebServiceConstants
from fastapi import APIRouter, HTTPException, Request, Response

LOGGER = LambdaLoggerGenerator.get_logger(__name__, LambdaLoggerGenerator.get_level_from_env())

router = APIRouter(
    prefix=f'/{WebServiceConstants.ADMIN}/system',
    tags=["System Admin Records CRUD (Admins-Only)"],
    responses={404: {"description": "Not found"}},
)


@router.put("/es_setup")
@router.put("/es_setup/")
async def es_setup(request: Request, tenant: Union[str, None]=None, venue: Union[str, None]=None, group_names: Union[str, None]=None):
    LOGGER.debug(f'started es_setup')
    auth_info = FastApiUtils.get_authorization_info(request)
    query_body = {
        'tenant': tenant,
        'venue': venue,
        'ldap_group_names': group_names if group_names is None else [k.strip() for k in group_names.split(',')],
    }
    auth_crud = AuthCrud(auth_info, query_body)
    is_admin_result = auth_crud.is_admin()
    if is_admin_result['statusCode'] != 200:
        raise HTTPException(status_code=is_admin_result['statusCode'], detail=is_admin_result['body'])
    try:
        SetupESIndexAlias().start()
    except Exception as e:
        LOGGER.exception(f'')
        raise HTTPException(status_code=500, detail=str(e))
    return {'message': 'successful'}


@router.put("/maap_daac_config_setup")
@router.put("/maap_daac_config_setup/")
async def maap_daac_config_setup(request: Request, tenant: Union[str, None]=None, venue: Union[str, None]=None, group_names: Union[str, None]=None):
    LOGGER.debug(f'started maap_daac_config_setup')
    auth_info = FastApiUtils.get_authorization_info(request)
    query_body = {
        'tenant': tenant,
        'venue': venue,
        'ldap_group_names': group_names if group_names is None else [k.strip() for k in group_names.split(',')],
    }
    auth_crud = AuthCrud(auth_info, query_body)
    is_admin_result = auth_crud.is_admin()
    if is_admin_result['statusCode'] != 200:
        raise HTTPException(status_code=is_admin_result['statusCode'], detail=is_admin_result['body'])
    try:
        SetupESIndexAlias().setup_maap_daac_index()
    except Exception as e:
        LOGGER.exception(f'')
        raise HTTPException(status_code=500, detail=str(e))
    return {'message': 'successful'}