LBRYCommunity/lbry-sdk
1
0
Fork 1
Code Issues Pull requests Projects Releases 282 Packages Wiki Activity

fix get_blobs_for_stream

Browse source
This commit is contained in:
Jack Robison 2019-02-14 18:16:16 -05:00
parent 39737c790f
commit f1b60e3ef2
No known key found for this signature in database
GPG key ID: DF25C68FE0239BB2
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
lbrynet/extras/daemon/storage.py
View file

@ -421,11 +421,12 @@ class SQLiteStorage(SQLiteMixin):
return self.db.run(_store_stream) return self.db.run(_store_stream)
def get_blobs_for_stream(self, stream_hash, only_completed=False): def get_blobs_for_stream(self, stream_hash, only_completed=False) -> typing.Awaitable[typing.List[BlobInfo]]:
def _get_blobs_for_stream(transaction): def _get_blobs_for_stream(transaction):
crypt_blob_infos = [] crypt_blob_infos = []
stream_blobs = transaction.execute( stream_blobs = transaction.execute(
"select blob_hash, position, iv from stream_blob where stream_hash=?", (stream_hash, ) "select blob_hash, position, iv from stream_blob where stream_hash=? "
"order by position asc", (stream_hash, )
).fetchall() ).fetchall()
if only_completed: if only_completed:
lengths = transaction.execute( lengths = transaction.execute(
@ -447,7 +448,8 @@ class SQLiteStorage(SQLiteMixin):
for blob_hash, position, iv in stream_blobs: for blob_hash, position, iv in stream_blobs:
blob_length = blob_length_dict.get(blob_hash, 0) blob_length = blob_length_dict.get(blob_hash, 0)
crypt_blob_infos.append(BlobInfo(position, blob_length, iv, blob_hash)) crypt_blob_infos.append(BlobInfo(position, blob_length, iv, blob_hash))
crypt_blob_infos = sorted(crypt_blob_infos, key=lambda info: info.blob_num) if not blob_hash:
break
return crypt_blob_infos return crypt_blob_infos
return self.db.run(_get_blobs_for_stream) return self.db.run(_get_blobs_for_stream)