with構文を使うとno attribute 'cursor'と言うエラーが出てしまいます

Question

###該当のソースコード

ファイル名DBConnection

import pymysql

def getConnection():
    return pymysql.connect(
        host='localhost',
        user='root',
        db='データベース名',
        charset='utf8',
        cursorclass=pymysql.cursors.DictCursor
    )

の時

###該当のソースコード

from DBConnection import getConnection

db = getConnection()
with db.cursor() as cursor:

はエラーが出ないのですが

###該当のソースコード

from DBConnection import getConnection

with getConnection() as db:
    cursor = db.cursor()

だと'DictCursor' object has no attribute 'cursor'と言うエラーが出てしまいます。
上記二つは同じことをしているような気がするのですがこれはなぜなのでしょうか？
それとwithをつける分はエラーが出ないcursorメソッドを使う時では無く、エラーが出るconnectメソッドを使う時になるのでしょうか？

Answer 1

ご自身が作成されたgetConnection()関数はPyMYSQLのConnectionオブジェクトを返し、それに対してwith句にかけるとCursor(DictCursor)オブジェクトが得られます。

https://github.com/PyMySQL/PyMySQL/blob/ddbf0e8521d111e2f4ac4deb4ef8d42fad72f159/pymysql/connections.py#L870

従いまして、

with getConnection() as db:
    cursor = db.cursor()

ここのdbにはConnectionオブジェクトではなく、Cursorオブジェクトが入っています。Cursorオブジェクトには、cursorメソッドはありませんので、エラーとなります。

Comments

[退会済みユーザー]

ご回答ありがとうございます。withをつける位置はwith getConnection() as db:のタイミングになるのでしょうか？その場合Cursorオブジェクトには、cursorメソッドが無くエラーになってしまうのですがどのように変えればいいのでしょうか？

[yukkeorg]

そのタイミングで良いかと思います。db変数はCursorオブジェクトなので、db変数を先のcursorとして使えばよいかと思います。（db変数はcursorなど名前を変更しましょう。）

[退会済みユーザー]

できました。ご回答ありがとうございました。