Skip to content

Commit

Permalink
Fixed the lint error
Browse files Browse the repository at this point in the history
  • Loading branch information
athakur-reef committed Oct 17, 2023
1 parent 69ad361 commit 3c2e9af
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 9 deletions.
16 changes: 10 additions & 6 deletions b2/console_tool.py
Original file line number Diff line number Diff line change
Expand Up @@ -1413,16 +1413,18 @@ def _setup_parser(cls, parser):
super()._setup_parser(parser)

def run(self, args):
args.localFileName, mode, allow_seeking = self.get_download_info_from_local_filename(args.localFileName)
local_filename, mode, allow_seeking = self.get_download_info_from_local_filename(
args.localFileName
)

progress_listener = make_progress_listener(args.localFileName, args.noProgress)
progress_listener = make_progress_listener(local_filename, args.noProgress)
encryption_setting = self._get_source_sse_setting(args)
self._set_threads_from_args(args)
downloaded_file = self.api.download_file_by_id(
args.fileId, progress_listener, encryption=encryption_setting
)
self._print_download_info(downloaded_file)
downloaded_file.save_to(args.localFileName, mode, allow_seeking)
downloaded_file.save_to(local_filename, mode, allow_seeking)
self._print('Download finished')
return 0

Expand Down Expand Up @@ -1454,17 +1456,19 @@ def _setup_parser(cls, parser):
super()._setup_parser(parser)

def run(self, args):
args.localFileName, mode, allow_seeking = self.get_download_info_from_local_filename(args.localFileName)
local_filename, mode, allow_seeking = self.get_download_info_from_local_filename(
args.localFileName
)

self._set_threads_from_args(args)
bucket = self.api.get_bucket_by_name(args.bucketName)
progress_listener = make_progress_listener(args.localFileName, args.noProgress)
progress_listener = make_progress_listener(local_filename, args.noProgress)
encryption_setting = self._get_source_sse_setting(args)
downloaded_file = bucket.download_file_by_name(
args.b2FileName, progress_listener, encryption=encryption_setting
)
self._print_download_info(downloaded_file)
downloaded_file.save_to(args.localFileName, mode, allow_seeking)
downloaded_file.save_to(local_filename, mode, allow_seeking)
self._print('Download finished')
return 0

Expand Down
4 changes: 1 addition & 3 deletions test/unit/test_console_tool.py
Original file line number Diff line number Diff line change
Expand Up @@ -1263,9 +1263,7 @@ def test_download_to_non_seekable_file(self):
['upload-file', '--noProgress', 'my-bucket', local_file, 'file.txt'],
remove_version=True,
)
command = [
'download-file-by-name', 'my-bucket', 'file.txt', '/dev/stdout'
]
command = ['download-file-by-name', 'my-bucket', 'file.txt', '/dev/stdout']
self._run_command(command)

# Restore original stdout
Expand Down

0 comments on commit 3c2e9af

Please sign in to comment.