Work membership and general clean up

app/routers/membership.py

@@ -12,6 +12,7 @@ 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 ..models.membership_model import Membership_Base
+from ..models.membership_methods import load_membership_obj
 from ..models.response_model import *
 
 
@@ -90,6 +91,84 @@ async def get_membership_obj_li(
     return result
 
 
+# Look up is only for account, person, or user records
+@router.get('/lookup', response_model=Resp_Body_Base)
+async def lookup_membership_obj(
+        for_obj_id: Union[int,str],
+        for_obj_type: str = Query(..., min_length=2, max_length=50),
+        x_account_id: str = Header(...),
+        inc_membership_profile: bool = True,
+        inc_membership_cfg: bool = True,
+        inc_extended_profile: bool = True,
+        by_alias: bool = True,
+        exclude_unset: bool = True,
+        ):
+    log.setLevel(logging.DEBUG) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
+    log.debug(locals())
+
+    obj_type = 'membership'
+    base_name = Membership_Base
+
+    if for_obj_id := redis_lookup_id_random(record_id_random=for_obj_id, table_name=for_obj_type): pass
+    else: return mk_resp(data=False, status_code=404) # Not Found
+    log.setLevel(logging.DEBUG) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
+
+    data = {}
+    as_list = False
+    if for_obj_type == 'account' and for_obj_id:
+        data['account_id'] = for_obj_id
+        sql_where_for_obj_type = """`membership`.account_id = :account_id"""
+        sql_limit = ''
+        as_list = True
+    elif for_obj_type == 'person' and for_obj_id:
+        data['person_id'] = for_obj_id
+        sql_where_for_obj_type = """`membership`.person_id = :person_id"""
+        sql_limit = 'LIMIT 1'
+    elif for_obj_type == 'user' and for_obj_id:
+        data['user_id'] = for_obj_id
+        sql_where_for_obj_type = """`membership`.user_id = :user_id"""
+        sql_limit = 'LIMIT 1'
+    else:
+        log.debug(f'Object type={for_obj_type}; Object ID={for_obj_id}')
+        return mk_resp(data=False, status_code=400) # Bad Request
+
+    sql = f"""
+        SELECT id AS 'membership_id', id_random AS 'membership_id_random'
+        FROM `membership` AS `membership`
+        WHERE {sql_where_for_obj_type}
+        {sql_limit}
+        """
+
+    # This will return a list if selecting by account ID
+    membership_obj_result = sql_select(data=data, sql=sql, as_list=as_list)
+    if isinstance(membership_obj_result, dict):
+        membership_id = membership_obj_result.get('membership_id', None)
+        membership_obj = load_membership_obj(
+            membership_id=membership_id,
+            inc_membership_profile=inc_membership_profile,
+            inc_membership_cfg=inc_membership_cfg,
+            inc_extended_profile=inc_extended_profile
+            ).dict(by_alias=by_alias, exclude_unset=exclude_unset)
+        data = membership_obj
+    elif isinstance(membership_obj_result, list):
+        membership_obj_li = []
+        for membership_obj in membership_obj_result:
+            membership_id = membership_obj.get('membership_id', None)
+            membership_obj_li.append(
+                load_membership_obj(
+                    membership_id=membership_id,
+                    inc_membership_profile=inc_membership_profile,
+                    inc_membership_cfg=inc_membership_cfg,
+                    inc_extended_profile=inc_extended_profile
+                    ).dict(by_alias=by_alias, exclude_unset=exclude_unset)
+                )
+        data = membership_obj_li
+    else:
+        log.debug(membership_obj_result)
+        return mk_resp(data=None, status_code=404) # Not Found
+    return mk_resp(data=data)
+
+
 @router.get('/{obj_id}', response_model=Resp_Body_Base)
 async def get_membership_obj(
         obj_id: str = Query(..., min_length=1, max_length=22),