LBRYFoundation/lbry-sdk
1
0
Fork 0
mirror of https://github.com/LBRYFoundation/lbry-sdk.git synced 2025-08-23 17:27:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor add_torrent, lints

Browse source
This commit is contained in:
Victor Shyba 2022-09-23 11:28:32 -03:00
parent 37adc59b37
commit 7c7e18534e
1 changed files with 10 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
lbry/extras/daemon/storage.py
View file

@ -639,11 +639,8 @@ class SQLiteStorage(SQLiteMixin):
async def get_all_torrent_files(self) -> typing.List[typing.Dict]: async def get_all_torrent_files(self) -> typing.List[typing.Dict]:
def _get_all_torrent_files(transaction): def _get_all_torrent_files(transaction):
cursor = transaction.execute("select * from file join torrent on file.bt_infohash=torrent.bt_infohash") cursor = transaction.execute("select * from file join torrent on file.bt_infohash=torrent.bt_infohash")
return [ return map(lambda row: dict(zip(list(map(itemgetter(0), cursor.description)), row)), cursor.fetchall())
{field: value for field, value in zip(list(map(itemgetter(0), cursor.description)), row)} return list(await self.db.run(_get_all_torrent_files))
for row in cursor.fetchall()
]
return await self.db.run(_get_all_torrent_files)
def change_file_status(self, stream_hash: str, new_status: str): def change_file_status(self, stream_hash: str, new_status: str):
log.debug("update file status %s -> %s", stream_hash, new_status) log.debug("update file status %s -> %s", stream_hash, new_status)
@ -885,21 +882,20 @@ class SQLiteStorage(SQLiteMixin):
if stream_hash in self.content_claim_callbacks: if stream_hash in self.content_claim_callbacks:
await self.content_claim_callbacks[stream_hash]() await self.content_claim_callbacks[stream_hash]()
def _save_torrent(self, transaction, bt_infohash, length, name):
transaction.execute(
"insert or replace into torrent values (?, NULL, ?, ?)", (bt_infohash, length, name)
).fetchall()
async def add_torrent(self, bt_infohash, length, name): async def add_torrent(self, bt_infohash, length, name):
return await self.db.run(self._save_torrent, bt_infohash, length, name) def _save_torrent(transaction, bt_infohash, length, name):
transaction.execute(
"insert or replace into torrent values (?, NULL, ?, ?)", (bt_infohash, length, name)
).fetchall()
return await self.db.run(_save_torrent, bt_infohash, length, name)
async def save_torrent_content_claim(self, bt_infohash, claim_outpoint, length, name): async def save_torrent_content_claim(self, bt_infohash, claim_outpoint, length, name):
def _save_torrent(transaction): def _save_torrent_claim(transaction):
self._save_torrent(transaction, bt_infohash, length, name)
transaction.execute( transaction.execute(
"insert or replace into content_claim values (NULL, ?, ?)", (bt_infohash, claim_outpoint) "insert or replace into content_claim values (NULL, ?, ?)", (bt_infohash, claim_outpoint)
).fetchall() ).fetchall()
await self.db.run(_save_torrent) await self.add_torrent(bt_infohash, length, name)
await self.db.run(_save_torrent_claim)
# update corresponding ManagedEncryptedFileDownloader object # update corresponding ManagedEncryptedFileDownloader object
if bt_infohash in self.content_claim_callbacks: if bt_infohash in self.content_claim_callbacks:
await self.content_claim_callbacks[bt_infohash]() await self.content_claim_callbacks[bt_infohash]()