MySQLdbでupdateのSQLを実行した時、更新した件数を取得することはできないのでしょうか？

Question

python 2.7系を使用しております。

pythonのMySQLdbライブラリを用いて、updateのSQLを実行するプログラムを組んでおります。

    cursor.execute("UPDATE categories set status = 'active'")
    res = connect.commit()
    print res # この時に、などに件数を取得できればとおもったのですが、返り値がありませんでした。

MySQLのCLIでUPDATE文を実行するときは以下のように、何件ヒットして、何件更新されたかが表示されます。

Query OK, 1 row affected (0.02 sec)
Rows matched: 1  Changed: 1  Warnings: 0

こういったことをPythonのプログラム上でもできればと思いました。

宜しくお願いします。

Answer 1

上記のコードの流れでなら、

• cursor.execute()でINSERT/UPDATEを実行した場合に返り値に作用された行数がくるので、

それを使う。

• connect.affected_rows()を使う。（KiyoshiMotokiさんの回答+BJCRobotさんのコメント）

の2方式でできます。

ただし、commit()をしてしまうと変更行数の情報がconnectオブジェクトからも消えるので、commit()前に実施する必要があるみたいです。

試せる環境があったので、試してみたらこうなりました。

• Python 2.7.9
• MySQL-python

>>> import MySQLdb
>>> connect = MySQLdb.connect(db="test",user="root")
>>> cursor = connect.cursor()
>>> cursor.execute("UPDATE categories set status = 'active'")
1L
>>> connect.affected_rows()
1L
>>> res = connect.commit()
>>> res
>>> connect.affected_rows()
0L
>>>
>>>
>>> cursor = connect.cursor()
>>> cursor.execute("insert into categories values (NULL, 'active2')")
1L
>>> connect.affected_rows()
1L
>>> res = connect.commit()
>>> res
>>> connect.affected_rows()
0L

Answer 2

未検証の回答で恐縮ですが、
マニュアルを読む限りcursor.rowcountで取得できるようです。
https://www.python.org/dev/peps/pep-0249/#rowcount

Comments

[Mr_Roboto]

＞ PythonのMySQLdbライブラリを用いて
って話ですよね？ ^_^；

MySQL直接呼ぶならこっちかと
23.8.7.1 mysql_affected_rows()
https://dev.mysql.com/doc/refman/5.6/ja/mysql-affected-rows.html

[KiyoshiMotoki]

BJCRobot様

ご指摘ありがとうございます。
リンクを訂正させていただきました。

以下のページ等で
「MySQLdbにも rowcount アトリビュートが存在しそうだ」
という当たりをつけていたのですが、異なるAPIのマニュアルを参照してしまっておりました (^^;
https://bytes.com/topic/python/answers/638593-mysql-rows-affected