Function insert_data

sometask.py:255–276

通用数据插入函数

(table_name, data)

def insert_data(table_name, data):
    """通用数据插入函数"""
    if not data:
        logger.error(f"{table_name} data fetch failed")
        return
    cursor = None
    try:
        start = time.monotonic()
        cursor = conn.cursor()
        if "data" in data:
            data = data["data"]
        logger.info(f'Inserting "{table_name}"...')
        cursor.execute(
            f'INSERT INTO "{table_name}" (data, insert_time) VALUES (%s, %s)',
            (json.dumps(data), int(time.time()))
        )
        logger.info(f'"{table_name}" inserted in {time.monotonic() - start:.2f}s')
    except Exception as err:
        logger.exception(f"Error inserting into {table_name}: {err}")
    finally:
        if cursor:
            cursor.close()


if __name__ == "__main__":