OSIT-AE-API-FastAPI/app/routers/site.py

import datetime
#from datetime import datetime, time, timedelta
from fastapi import APIRouter, Body, Depends, Header, HTTPException, Query, Response, status
from pydantic import BaseModel, EmailStr, Field
from typing import Dict, List, Optional, Set, Union

from app.lib_general import log, logging
from app.config import settings
from app.db_sql import sql_insert, sql_update, sql_insert_or_update, sql_select, sql_delete, redis_lookup_id_random

from app.routers.api_crud import delete_obj_template, get_obj_template, get_obj_li_template, patch_obj_template, post_obj_template

from app.models.response_models import *
from app.models.site_models import Site_Base
from app.models.site_domain_models import Site_Domain_Base


router = APIRouter()


@router.post('', response_model=Resp_Body_Base)
async def post_site_obj(
        obj: Site_Base,
        x_account_id: str = Header(...),
        return_obj: Optional[bool] = True,
        by_alias: Optional[bool] = True,
        exclude_unset: Optional[bool] = True,
        response: Response = Response,
        ):
    log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

    obj_type = 'site'
    obj_data_dict = obj.dict(by_alias=False, exclude_unset=True)
    result = post_obj_template(
        obj_type=obj_type,
        data=obj_data_dict,
        return_obj=True,
        by_alias=True,
        exclude_unset=True,
        )
    return result


@router.patch('/{obj_id}', response_model=Resp_Body_Base)
async def patch_site_obj(
        obj_id: str = Query(..., min_length=1, max_length=22),
        obj: Site_Base = None,
        x_account_id: Optional[str] = Header(..., ),
        return_obj: Optional[bool] = True,
        by_alias: Optional[bool] = True,
        exclude_unset: Optional[bool] = True,
        response: Response = Response,
        ):
    log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

    obj_type = 'site'
    obj_data_dict = obj.dict(by_alias=False, exclude_unset=True)
    obj_data_dict['id'] = redis_lookup_id_random(record_id_random=obj_id, table_name=obj_type)
    obj_data_dict['id_random'] = obj_id
    result = patch_obj_template(
        obj_type=obj_type,
        data=obj_data_dict,
        obj_id=obj_id,
        return_obj=True,
        by_alias=True,
        exclude_unset=True,
        )
    return result


@router.get('/list', response_model=Resp_Body_Base)
async def get_site_obj_li(
        for_obj_type: Optional[str] = Query(None, min_length=2, max_length=50),
        for_obj_id: Optional[str] = Query(None, min_length=1, max_length=22),
        x_account_id: str = Header(...),
        by_alias: Optional[bool] = True,
        exclude_unset: Optional[bool] = True,
        response: Response = Response,
        ):
    log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

    obj_type = 'site'
    result = get_obj_li_template(
        obj_type=obj_type,
        for_obj_type=for_obj_type,
        for_obj_id=for_obj_id,
        by_alias=True,
        exclude_unset=True,
        )
    return result


@router.get('/{obj_id}', response_model=Resp_Body_Base)
async def get_site_obj(
        obj_id: str = Query(..., min_length=1, max_length=22),
        x_account_id: str = Header(...),
        by_alias: Optional[bool] = True,
        exclude_unset: Optional[bool] = True,
        response: Response = Response,
        ):
    log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

    obj_type = 'site'
    result = get_obj_template(
        obj_type=obj_type,
        obj_id=obj_id,
        by_alias=True,
        exclude_unset=True,
        )
    return result


@router.delete('/{obj_id}', response_model=Resp_Body_Base)
async def delete_site_obj(
        obj_id: str = Query(..., min_length=1, max_length=22),
        x_account_id: str = Header(...),
        response: Response = Response,
        ):
    log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
    log.debug(locals())

    obj_type = 'site'
    result = delete_obj_template(
        obj_type=obj_type,
        obj_id=obj_id,
        )
    return result


# @router.get('/domain/lookup_fqdn', response_model=Resp_Body_Base)
# async def get_site_obj(
#         fqdn: str,
#         access_key: Optional[str] = None,
#         referrer: Optional[str] = None,
#         x_account_id: str = Header(...),
#         by_alias: Optional[bool] = True,
#         exclude_unset: Optional[bool] = True,
#         response: Response = Response,
#         ):
#     log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
#     log.debug(locals())

#     data = {}
#     data['fqdn'] = fqdn
#     data['domain_access_key'] = access_key
#     data['required_referrer'] = referrer

#     if access_key:
#         sql_domain_access_key = 'AND site_domain.domain_access_key = :domain_access_key'
#     if referrer:
#         sql_referrer = 'AND site_domain.required_referrer = :required_referrer'

#     sql = """
#         SELECT *
#         FROM `v_site_domain` AS site_domain
#         WHERE site_domain.fqdn = :fqdn
#             {sql_domain_access_key}
#             {sql_referrer}
#         ;
#         """

#     if site_domain_obj_result := sql_select(data=data, sql=sql):
#         try:
#             site_domain_obj = Site_Domain_Base(**site_domain_obj_result)
#             log.debug(order_obj)
#         except ValidationError as e:
#             log.error(e.json())
#         return mk_resp(data=order_obj_li)
#     else:
#         log.debug(order_obj_li_result)
#         return mk_resp(data=False, status_code=404, response=response)