我想制作自己的MagicCursor课程。我希望它继承sqlite3.Cursor()的所有方法和属性。但在他的情况下看起来并不那么容易。
通常我们会创建一个这样的游标:
import sqlite3
conn = sqlite3.connect('test.db')
crs = conn.cursor()
因为两者之间存在联系。所以我认为我必须定义自己的Connection类,然后定义我自己的Cursor。
以下是我要实施的内容:
import sqlite3
conn = MyConnect('test.db') ## this returns me a connect object, inherit from sqlite3.Connection
crs = conn.cursor() ## this returns me a MyCursor class,
## which inherit from sqlite3.Cursor() and has my customized method
这是我的代码,但失败了。
class MyConnect(sqlite3.Connection):
def cursor(self):
return MyCursor(self)
class MyCursor(sqlite3.cursor):
def __init__(self, connect):
self = connect.cursor()
def mymethod1(self, argv)
...... return ...... ## what ever I defined
任何人都知道如何实现它?
答案 0 :(得分:3)
import sqlite3
class MyConnect(sqlite3.Connection):
def cursor(self):
return super(MyConnect, self).cursor(MyCursor)
class MyCursor(sqlite3.Cursor):
def mymethod1(self, argv):
print('Got here')
conn = sqlite3.connect(':memory:', factory=MyConnect)
print(conn)
# <__main__.MyConnect object at 0x90db66c>
cursor = conn.cursor()
print(cursor)
# <__main__.MyCursor object at 0x90d19dc>
cursor.mymethod1('foo')
# Got here
在此示例中,super(MyConnect, self).cursor(MyCursor)
调用
sqlite3.Connection.cursor(MyCursor)
。
sqlite3.Connection.cursor接受可选参数cursorClass
。提供时,这是用于实例化游标的类。