Clean up of code...

app/methods/post_methods.py

@@ -109,7 +109,8 @@ def load_post_obj(
 
 # ### BEGIN ### API Post Methods ### load_post_obj_list() ###
 def load_post_obj_list(
-        account_id: int|str,
+        account_id: int|str|None = None,
+        user_id: int|str|None = None,
         limit: int = 1000,
         model_as_dict: bool = False,
         enabled: str = 'enabled', # enabled, disabled, all
@@ -120,11 +121,17 @@ def load_post_obj_list(
     log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
     log.debug(locals())
 
-    if account_id := redis_lookup_id_random(record_id_random=account_id, table_name='account'): pass
-    else: return False
-
     data = {}
-    data['account_id'] = account_id
+    if account_id:
+        if account_id := redis_lookup_id_random(record_id_random=account_id, table_name='account'): pass
+        else: return False
+        data['account_id'] = account_id
+        sql_obj_type_id = f'`tbl`.account_id = :account_id'
+    elif user_id:
+        if user_id := redis_lookup_id_random(record_id_random=user_id, table_name='user'): pass
+        else: return False
+        data['user_id'] = user_id
+        sql_obj_type_id = f'`tbl`.user_id = :user_id'
 
     if enabled in ['enabled', 'disabled', 'all']:
         if enabled == 'enabled':
@@ -146,8 +153,9 @@ def load_post_obj_list(
     sql = f"""
         SELECT `tbl`.id AS 'post_id', `tbl`.id_random AS 'post_id_random'
         FROM `post` AS `tbl`
-        WHERE `tbl`.account_id = :account_id
-        {sql_enabled}
+        WHERE
+            {sql_obj_type_id}
+            {sql_enabled}
         ORDER BY `tbl`.created_on DESC, `tbl`.updated_on DESC
         {sql_limit};
         """
@@ -176,7 +184,6 @@ def load_post_obj_list(
         log.setLevel(logging.DEBUG) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
         log.debug(post_rec_li_result)
         post_result_li = []
-    log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL
 
     return post_result_li
 # ### END ### API Post Methods ### load_post_obj_list() ###