251 lines
9.5 KiB
Python
251 lines
9.5 KiB
Python
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 ..log import *
|
|
from app.config import settings
|
|
from app.db_sql import *
|
|
|
|
from .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_domain_obj(
|
|
obj: Site_Domain_Base,
|
|
x_account_id: str = Header(...),
|
|
return_obj: Optional[bool] = True,
|
|
by_alias: Optional[bool] = True,
|
|
exclude_unset: Optional[bool] = True,
|
|
exclude_none: Optional[bool] = True,
|
|
response: Response = Response,
|
|
):
|
|
log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
|
|
log.debug(locals())
|
|
|
|
obj_type = 'site_domain'
|
|
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_domain_obj(
|
|
obj: Site_Domain_Base,
|
|
obj_id: str = Query(..., min_length=1, max_length=22),
|
|
x_account_id: str = Header(..., ),
|
|
return_obj: bool = True,
|
|
by_alias: bool = True,
|
|
exclude_unset: bool = True,
|
|
exclude_none: bool = True,
|
|
response: Response = Response,
|
|
):
|
|
log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
|
|
log.debug(locals())
|
|
|
|
obj_type = 'site_domain'
|
|
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('/lookup_fqdn', response_model=Resp_Body_Base)
|
|
async def lookup_site_domain_obj(
|
|
fqdn: str,
|
|
access_key: Optional[str] = Query(None, min_length=4, max_length=50),
|
|
referrer: Optional[str] = Query(None, min_length=8, max_length=150),
|
|
referer: Optional[str] = Query(None, min_length=8, max_length=150),
|
|
x_account_id: str = Header(...),
|
|
by_alias: bool = True,
|
|
exclude_unset: bool = True,
|
|
exclude_none: bool = True,
|
|
response: Response = Response,
|
|
):
|
|
log.setLevel(logging.INFO) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
|
|
log.debug(locals())
|
|
|
|
data = {}
|
|
data['fqdn'] = fqdn
|
|
data['domain_access_key'] = access_key
|
|
if referrer: # This should work...
|
|
data['required_referrer'] = referrer
|
|
data['required_referer'] = referrer
|
|
elif referer: # This should not be needed...
|
|
referrer = referer
|
|
data['required_referrer'] = referer
|
|
data['required_refer'] = referer
|
|
log.info(f'Looking up site domain with FQDN: {fqdn}; Access Key {access_key}; Referrer: {referrer}')
|
|
log.debug(data)
|
|
|
|
if access_key and referrer:
|
|
sql_access_key_referrer = """
|
|
AND site_domain.domain_access_key = :domain_access_key
|
|
AND site_domain.required_referrer = :required_referrer
|
|
"""
|
|
elif access_key:
|
|
sql_access_key_referrer = """
|
|
AND site_domain.domain_access_key = :domain_access_key
|
|
AND (site_domain.required_referrer IS NULL OR site_domain.required_referrer = '')
|
|
"""
|
|
# elif referrer:
|
|
# sql_access_key_referrer = """
|
|
# AND site_domain.required_referrer = :required_referrer
|
|
# """
|
|
else:
|
|
sql_access_key_referrer = """
|
|
AND (site_domain.domain_access_key IS NULL OR site_domain.domain_access_key = '')
|
|
AND (site_domain.required_referrer IS NULL OR site_domain.required_referrer = '')
|
|
"""
|
|
|
|
sql = f"""
|
|
SELECT *, site_enable_from AS 'enable_from', site_enable_to AS 'enable_to'
|
|
FROM `v_site_domain` AS site_domain
|
|
WHERE site_domain.fqdn = :fqdn
|
|
{sql_access_key_referrer}
|
|
;
|
|
"""
|
|
log.debug(sql)
|
|
|
|
# NOTE: This need to broken out into a methods function for site and or site_domain
|
|
if site_domain_obj_result := sql_select(data=data, sql=sql):
|
|
if isinstance(site_domain_obj_result, dict):
|
|
log.info('Got a site domain result')
|
|
try:
|
|
site_obj = Site_Base(**site_domain_obj_result).dict(by_alias=by_alias, exclude_unset=exclude_unset, exclude_none=exclude_none)
|
|
log.debug(site_obj)
|
|
except ValidationError as e:
|
|
log.error(e.json())
|
|
try:
|
|
site_domain_obj = Site_Domain_Base(**site_domain_obj_result).dict(by_alias=by_alias, exclude_unset=exclude_unset, exclude_none=exclude_none)
|
|
log.debug(site_domain_obj)
|
|
except ValidationError as e:
|
|
log.error(e.json())
|
|
|
|
site_domain_obj['site'] = site_obj
|
|
log.debug(site_domain_obj)
|
|
|
|
return mk_resp(data=site_domain_obj, response=response)
|
|
elif isinstance(site_domain_obj_result, list):
|
|
log.info('Got multiple site domain results')
|
|
log.warning(f'More than one site domain records were returned. This was unexpected and needs to be corrected. FQDN: {fqdn}')
|
|
site_domain_obj_li = site_domain_obj_result
|
|
return mk_resp(data=site_domain_obj_li, response=response)
|
|
else:
|
|
log.error(f'Unexpected results for the site domain records was returned. FQDN: {fqdn}; Access Key {access_key}; Referrer: {referrer}')
|
|
return mk_resp(data=False, status_code=500, status_message=f'Unexpected results for the site domain records was returned. FQDN: {fqdn}; Access Key {access_key}; Referrer: {referrer}', response=response)
|
|
else:
|
|
log.info('No site domain results')
|
|
log.debug(site_domain_obj_result)
|
|
return mk_resp(data=False, status_code=404, response=response)
|
|
|
|
|
|
@router.get('/list', response_model=Resp_Body_Base)
|
|
async def get_site_domain_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_domain'
|
|
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('/{site_domain_id}', response_model=Resp_Body_Base)
|
|
async def get_site_domain_obj(
|
|
site_domain_id: str = Query(..., min_length=11, max_length=22),
|
|
x_account_id: str = Header(...),
|
|
by_alias: Optional[bool] = True,
|
|
exclude_unset: Optional[bool] = True,
|
|
exclude_none: Optional[bool] = True,
|
|
response: Response = Response,
|
|
):
|
|
log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
|
|
log.debug(locals())
|
|
|
|
if site_domain_id := redis_lookup_id_random(record_id_random=site_domain_id, table_name='site_domain'): pass
|
|
else: return mk_resp(data=None, status_code=404, response=response)
|
|
|
|
data = { 'site_domain_id': site_domain_id }
|
|
|
|
sql = f"""
|
|
SELECT *, site_enable_from AS 'enable_from', site_enable_to AS 'enable_to'
|
|
FROM `v_site_domain` AS site_domain
|
|
WHERE site_domain.id = :site_domain_id
|
|
;
|
|
"""
|
|
log.debug(sql)
|
|
|
|
# NOTE: This need to broken out into a methods function for site and or site_domain
|
|
if site_domain_obj_result := sql_select(data=data, sql=sql):
|
|
try:
|
|
site_obj = Site_Base(**site_domain_obj_result).dict(by_alias=by_alias, exclude_unset=exclude_unset, exclude_none=exclude_none)
|
|
log.debug(site_obj)
|
|
except ValidationError as e:
|
|
log.error(e.json())
|
|
try:
|
|
site_domain_obj = Site_Domain_Base(**site_domain_obj_result).dict(by_alias=by_alias, exclude_unset=exclude_unset, exclude_none=exclude_none)
|
|
log.debug(site_domain_obj)
|
|
except ValidationError as e:
|
|
log.error(e.json())
|
|
|
|
site_domain_obj['site'] = site_obj
|
|
log.debug(site_domain_obj)
|
|
|
|
return mk_resp(data=site_domain_obj, response=response)
|
|
else:
|
|
log.debug(site_domain_obj_result)
|
|
return mk_resp(data=False, status_code=404, response=response)
|
|
|
|
|
|
@router.delete('/{obj_id}', response_model=Resp_Body_Base)
|
|
async def delete_site_domain_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_domain'
|
|
result = delete_obj_template(
|
|
obj_type=obj_type,
|
|
obj_id=obj_id,
|
|
)
|
|
return result
|