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

add head_blob_only to ttfb

Browse source
This commit is contained in:
Victor Shyba 2019-05-01 13:59:45 -03:00
parent 895b3992c9
commit fc88261cc6
1 changed files with 14 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
scripts/time_to_first_byte.py
View file

@ -118,6 +118,7 @@ async def main(uris=None, cmd_args=None):
first_byte = time.time()
first_byte_times.append(first_byte - start)
print(f"{i + 1}/{len(resolvable)} - {first_byte - start} {uri}")
if not cmd_args.head_blob_only:
downloaded, amount_downloaded, blobs_in_stream = await wait_for_done(
conf, uri, cmd_args.stall_download_timeout
)
@ -134,7 +135,7 @@ async def main(uris=None, cmd_args=None):
failed_to_start.append(uri)
if cmd_args.exit_on_error:
return
if cmd_args.delete_after_download:
if cmd_args.delete_after_download or cmd_args.head_blob_only:
await daemon_rpc(conf, 'file_delete', delete_from_download_dir=True, claim_name=parse_lbry_uri(uri).name)
await asyncio.sleep(0.1)
@ -171,4 +172,5 @@ if __name__ == "__main__":
parser.add_argument("--exit_on_error", action='store_true')
parser.add_argument("--stall_download_timeout", default=10, type=int)
parser.add_argument("--delete_after_download", action='store_true')
parser.add_argument("--head_blob_only", action='store_true')
asyncio.run(main(cmd_args=parser.parse_args()))