2

データベース接続用のコンテキストマネージャーを作成しました。外部クラス（cx_oracle.Cursorの「execute」）のメソッドをオーバーライドして、クエリをデータベースに渡す前に無関係なバインド変数を削除したいと思います（そうでない場合は、DatabaseErrorが発生します）。そもそもなぜそこにいるのかという問題は直交しています。

以下は機能しますが、ここで_ getattribute _を正しく使用しているかどうか疑問に思っています（これまで実際に使用する言い訳はありませんでした）。

私は「すべての耳」でもあります...目...この外部ライブラリからこのメソッドをオーバーライドする方法について誰かがより良い提案を持っている場合。

ありがとう。私はPython2.7を使用しています

import cx_Oracle
from contextlib import contextmanager

class Cursor(cx_Oracle.Cursor):
    """A wrapper for cx_Oracle cursors that will drop extraneous bind variables passed to in"""

    def __init__(self, curs):
        self.curs = curs

    def execute(self, sql, *args, **kwargs):
        params = {}
        print("in execute, args[0] = [%s]" % str(args[0]))
        if len(args) == 1 and isinstance(args[0], dict):
            for bv in args[0].keys():
                if ':%s' % bv in sql:
                    params[bv] = args[0][bv]
            print('params = %s' % str(params))
        else:
            return self.curs.execute(sql, *args, **kwargs)

        return self.curs.execute(sql, params)


    def __getattribute__(self, name):
        if name == 'execute':
            return object.__getattribute__(self, name)
        elif name == 'curs':
            return object.__getattribute__(self, 'curs')
        else:
            curs = object.__getattribute__(self, 'curs')
            return cx_Oracle.Cursor.__getattribute__(curs, name)


@contextmanager
def db_conn():

    pool = cx_Oracle.SessionPool("user", "pwd", "database", min=2, max=10, increment=1, threaded=True)
    conn = pool.acquire()

    try:
        yield Cursor(conn.cursor())
    except:
        conn.rollback()
        raise
    else:
        conn.commit()
    finally:
        pool.release(conn)


if __name__ == '__main__':

    with db_conn() as curs:
        curs.execute('select * from dual where 1 = :a', {'a':1, 'b':2})
        print(curs.fetchall())
4

1 に答える 1

1

__getattr__ではなくメソッドを使用する__getattribute__場合、独自のクラスで名前を特別に大文字にする必要はありません。__getattr__通常の手段で属性が見つからない場合にのみ調べます。

何かのようなもの：

def __getattr__(self, name):
    return getattr(self.curs, name)

class Cursor(cx_Oracle.Cursor):に変更する必要がありますclass Cursor(object):。カーソルはデータベース接続のメソッドによって作成されるため、サブクラス化によってやろうとしていることを行うのは難しいです。そのcursorメソッドがカスタム カーソル インスタンスを返すように、SessionPool をサブクラス化することもできますが、おそらくその努力をする価値はありません。

于 2011-04-19T20:45:03.233 に答える