utils: Added warnings and checks to images downloader

This commit is contained in:
Derek Kozel 2017-07-14 13:18:10 +01:00 committed by Martin Braun
parent ebfaa4cd1e
commit d630f245e3

View File

@ -248,6 +248,8 @@ def main():
help="Do not clear images directory before extracting new files [default=%default]")
parser.add_option("-v", "--verbose", action="store_true", default=False,
help="Enable verbose output [default=%default]")
parser.add_option("--force-delete", action="store_true", default=False,
help="Delete all files in the target images directory without prompting [default=%default]")
(options, args) = parser.parse_args()
if options.buffer_size <= 0:
print("Invalid buffer size: %s" % (options.buffer_size))
@ -283,6 +285,18 @@ def main():
else:
print("Images destination: %s" % (images_dir))
### Check contradictory arguments
if options.force_delete and options.keep:
print("Error: Keep and force delete options contradict.\n")
parser.print_help()
return 1
### Prevent accidental file deletion
if options.install_location != default_images_dir and options.force_delete == False and options.keep != True:
print("Custom install location specified, defaulting to overwriting only image files.\n"
"Use \'--force-delete\' to clean the target directory first.")
options.keep = True
### Download or copy
downloader = uhd_images_downloader()
try: