Python 执行原始 SQL 时获取更新查询 (Mysql) 的结果

发布于09月15日

我在Django项目上工作，需要在另一个远程数据库(不是Django DB)中应用更改.

这些变化是不同步的，并与Celery 一起推出.但在执行这些查询时，我需要知道更新是否很好地计算了受影响的行数，否则我必须采取纠正措施. 到目前为止，我还没有找到任何方法来做到这一点.我试了cusr.fetchone()和curs.fetchall()，但结果是None.当I为print(curs.execute())时，即使找到乘积，结果也是0.This thread表示如果完成更新，则应为1. 即使是official documentation也不会谈论这个.

因此，就目前而言，我找到的唯一解决办法是执行第一个SELECT查询，如果是rowcount < 1，我将引发Exception.不是很好...

有没有办法做得更好？注:我用的是django.db.backends.mysql和pymysql

推荐答案

如果不是您的方法，我建议您在提交和判断之后立即判断受影响的行.

from django.db import connection

# Initialize affected_rows outside the context manager
affected_rows = None

# Assuming you have a SQL update query
update_query = "UPDATE your_table SET your_column = your_value WHERE your_condition"

# Get a cursor
with connection.cursor() as cursor:
    # Execute the update query
    cursor.execute(update_query)
    
    # Commit the transaction
    cursor.connection.commit()
    
    # Get the number of affected rows
    affected_rows = cursor.rowcount

# Check if the update was successful
if affected_rows is not None and affected_rows > 0:
    # Update was successful, take appropriate action
    pass
else:
    # No rows were affected, handle the error or raise an Exception
    raise Exception("No rows were affected by the update query.")

按顺序try 此操作并查找受影响的行