def get_store(cls, self_or_cls, **callargs):
with super(DynamicTableNameMixin,
cls).get_store(self_or_cls, **callargs) as self_or_cls:
driver = DynamicTableNameDriver(
self_or_cls.DYNAMIC_TABLE.format(**callargs),
self_or_cls.DYNAMIC_DB.format(**callargs))
yield proxy(self_or_cls, prop_name="cur", prop=driver)
def get_store(cls, self_or_cls, **callargs):
with super(MyDriverMixin, cls).get_store(self_or_cls,
**callargs) as self_or_cls:
driver = MyDriver()
try:
yield proxy(self_or_cls, prop_name="store", prop=driver)
except Exception as e:
driver.rollback()
raise e
else:
driver.close()
def get_store(cls, self_or_cls, **callargs):
prop_name = "cur"
with super(DynamicTableNameMixin,
cls).get_store(self_or_cls, **callargs) as self_or_cls:
if self_or_cls._need_proxy(prop_name):
driver = DynamicTableNameDriver(
self_or_cls.DYNAMIC_TABLE.format(**callargs),
self_or_cls.DYNAMIC_DB.format(**callargs))
self_or_cls = proxy(self_or_cls,
prop_name=prop_name,
prop=driver)
yield self_or_cls
def get_store(cls, self_or_cls, **callargs):
conn, cur = cls.init_sqlite
with super(SqliteDriverMixin,
cls).get_store(self_or_cls, **callargs) as self_or_cls:
cur = conn.cursor()
if hasattr(self_or_cls, "_need_proxy") \
and self_or_cls._need_proxy("store"):
store = SqliteProxy(
cur, before=[AOP.Hook(execute_before, ["execute"])])
self_or_cls = proxy(self_or_cls, prop_name="store", prop=store)
try:
yield self_or_cls
finally:
conn.commit()
async def get_store(cls, self_or_cls, **callargs):
with super().get_store(self_or_cls, **callargs) as self_or_cls:
yield proxy(self_or_cls, 111, "a")