Madiator2011/lbry-sdk
1
0
Fork 0
forked from LBRYCommunity/lbry-sdk
Code Pull requests Activity

pylint

Browse source
This commit is contained in:
Victor Shyba 2018-10-18 17:25:24 -03:00 committed by Lex Berezhny
parent 3b021af1bd
commit c497e9884a
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
torba/basedatabase.py
View file

@ -33,7 +33,7 @@ class AIOSQLite:
self.executor.shutdown(wait=True)
conn = self.connection
self.connection = None
return asyncio._get_running_loop().call_later(0.01, __close, conn)
return asyncio.get_event_loop_policy().get_event_loop().call_later(0.01, __close, conn)
def executescript(self, script: str) -> Awaitable:
return wrap_future(self.executor.submit(self.connection.executescript, script))
@ -48,13 +48,13 @@ class AIOSQLite:
parameters = parameters if parameters is not None else []
return self.run(lambda conn, sql, parameters: conn.execute(sql, parameters), sql, parameters)
def run(self, fn: Callable[[sqlite3.Connection, Any], Any], *args, **kwargs) -> Awaitable:
return wrap_future(self.executor.submit(self.__run_transaction, fn, *args, **kwargs))
def run(self, fun: Callable[[sqlite3.Connection, Any, Any], Any], *args, **kwargs) -> Awaitable:
return wrap_future(self.executor.submit(self.__run_transaction, fun, *args, **kwargs))
def __run_transaction(self, fn: Callable[[sqlite3.Connection, Any], Any], *args, **kwargs):
def __run_transaction(self, fun: Callable[[sqlite3.Connection, Any, Any], Any], *args, **kwargs):
self.connection.execute('begin')
try:
fn(self.connection, *args, **kwargs)
fun(self.connection, *args, **kwargs) # type: ignore
except (Exception, OSError):
self.connection.rollback()
raise