feat: add to_table to insert_into_select

Author: NightMarcher

README.md

@@ -131,11 +131,12 @@ await AccountMgr.insert_into_select(
         "name": RawSQL("CONCAT(LEFT(name, 26), ' [NEW]')"),
         "extend": {},
     },
+    to_table="account_bak",
 )
 ```
 Generate sql and execute
 ```sql
-    INSERT INTO account
+    INSERT INTO account_bak
         (gender, locale, active, name, extend)
     SELECT gender, locale, False active, CONCAT(LEFT(name, 26), ' [NEW]') name, '{}' extend
     FROM account

examples/service/routers/account.py

@@ -83,7 +83,6 @@ async def bulk_init_view():
                 }
             }
         })
-    # logger.debug(dicts)
     ok = await AccountMgr.bulk_create_from_dicts(
         dicts,
         batch_size=5,
@@ -144,7 +143,6 @@ async def bulk_upsert_view():
             "locale": locale.value,
             "extend": {},
         })
-    logger.debug(dicts)
     row_cnt = await AccountMgr.upsert_on_duplicated(
         dicts,
         insert_fields=["id", "gender", "name", "locale", "extend"],
@@ -165,6 +163,7 @@ async def bulk_clone_view(
             "name": RawSQL("CONCAT(LEFT(name, 26), ' [NEW]')"),
             "extend": {},
         },
+        to_table="account_bak",
     )
     return {"ok": ok}
 

fastapi_esql/orm/base_manager.py

@@ -119,12 +119,14 @@ async def insert_into_select(
         wheres: Union[str, Q, Dict[str, Any], List[Q]],
         remain_fields: List[str],
         assign_field_dict: Dict[str, Any],
+        to_table: Optional[str] = None,
     ):
         sql = SQLizer.insert_into_select(
             cls.table,
             wheres,
             remain_fields,
             assign_field_dict,
+            to_table,
             cls.model,
         )
         return await CursorHandler.exec_if_ok(sql, cls.rw_conn, logger)

fastapi_esql/utils/cursor_handler.py

@@ -16,7 +16,7 @@ async def fetch_dicts(
         try:
             return await conn.execute_query_dict(sql)
         except Exception as e:
-            logger.exception(e)
+            logger.exception(f"{e} SQL=>{sql}")
             return None
 
     @classmethod
@@ -30,7 +30,7 @@ async def sum_row_cnt(
             row_cnt, _ = await conn.execute_query(sql)
             return row_cnt
         except Exception as e:
-            logger.exception(e)
+            logger.exception(f"{e} SQL=>{sql}")
             return None
 
     @classmethod
@@ -44,5 +44,5 @@ async def exec_if_ok(
             await conn.execute_script(sql)
             return True
         except Exception as e:
-            logger.exception(e)
+            logger.exception(f"{e} SQL=>{sql}")
             return False

fastapi_esql/utils/sqlizer.py

@@ -159,7 +159,6 @@ def upsert_on_duplicated(
             f"({', '.join(cls._sqlize_value(d.get(f)) for f in insert_fields)})"
             for d in dicts
         ]
-        # logger.debug(values)
         new_table = f"`new_{table}`"
         upserts = [f"{field}={new_table}.{field}" for field in upsert_fields]
 
@@ -180,6 +179,7 @@ def insert_into_select(
         wheres: Union[str, Q, Dict[str, Any], List[Q]],
         remain_fields: List[str],
         assign_field_dict: Dict[str, Any],
+        to_table: Optional[str] = None,
         model: Optional[Model] = None,
     ) -> Optional[str]:
         if not all([table, wheres] or not any([remain_fields, assign_field_dict])):
@@ -190,10 +190,9 @@ def insert_into_select(
         for k, v in assign_field_dict.items():
             fields.append(k)
             assign_fields.append(f"{cls._sqlize_value(v)} {k}")
-        # logger.debug(assign_fields)
 
         sql = f"""
-    INSERT INTO {table}
+    INSERT INTO {to_table or table}
         ({", ".join(fields)})
     SELECT {", ".join(remain_fields + assign_fields)}
     FROM {table}