Как использовать один оператор INSERT INTO для нескольких строк в Python?

В настоящее время я работаю над ботом Discord Python, где я просматриваю список ForumTags и генерирую оператор INSERT INTO SQL для каждого объекта для вставки данных в базу данных MySQL.

Однако я хочу оптимизировать свой код, объединив все эти отдельные операторы INSERT INTO в один запрос, например:

INSERT INTO guild_support_tags (guild_id, tag_id, tag_name, category) VALUES 
(123, 1, "test", "test_category"), (456, 2, "another tag", "test_category2)

Это мой текущий код:

start = time.time()
for tag in forum.available_tags:
    await write_query("INSERT INTO guild_support_tags (guild_id, tag_id, tag_name, category) VALUES "
                      "(:guild_id, :tag_id, :tag_name, :category)", 
                      {"guild_id": interaction.guild_id, "tag_id": tag.id, "tag_name": tag.name, 
                       "category": category})

    print(f"loop done after {time.time() - start}")


# from other file - created by myself to execute MySQL statements
# I would like a solution where this part is untouched. But if it can be improved, its okay.
async def write_query(query: str, params: dict) -> None:
    async with async_session() as session:
        async with session.begin():
            await session.execute(text(query), params)

Возможно, полезно знать: в настоящее время я использую SQLAlchemy с aiomysql и Python3.12 в базе данных MySQL.

Я думаю, вы ищете функцию cursor.executemany().

John Gordon 25.07.2024 22:30
Почему в Python есть оператор "pass"?
Почему в Python есть оператор "pass"?
Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.
Некоторые методы, о которых вы не знали, что они существуют в Python
Некоторые методы, о которых вы не знали, что они существуют в Python
Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...
Основы Python Часть I
Основы Python Часть I
Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?
LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа
LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа
Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:
Оптимизация кода с помощью тернарного оператора Python
Оптимизация кода с помощью тернарного оператора Python
И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это
Советы по эффективной веб-разработке с помощью Python
Советы по эффективной веб-разработке с помощью Python
Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.
0
1
59
1
Перейти к ответу Данный вопрос помечен как решенный

Ответы 1

Ответ принят как подходящий

Я обнаружил, что все еще могу использовать функцию Execute(), и изменив тип аргумента params с dict на dict | list, она работает.

SQLAlchemy обнаружит это и изменит свое поведение, чтобы вставить все необходимые строки. Этот подход позволяет эффективно вставлять без циклов:

# Prepare the data for the query
query_values: list[dict] = [{"guild_id": interaction.guild_id, "tag_id": tag.id, 
                             "tag_name": tag.name, "category": category} 
                            for tag in forum.available_tags]

# Modify the function to allow "list" and "dict" as type for "params"
async def write_query(query: str, params: dict | list) -> None:
    async with async_session() as session:
        async with session.begin():
            await session.execute(text(query), params)

# execute it
await write_query("INSERT INTO guild_support_tags (guild_id, tag_id, tag_name, category) 
                   VALUES (:guild_id, :tag_id, :tag_name, :category)", query_values)

Другие вопросы по теме