Verify download filenames in URL

This commit is contained in:
jaseg 2019-06-25 14:22:13 +09:00
parent 0ad5efec38
commit fb2f3bcc2a
3 changed files with 9 additions and 6 deletions

View file

@ -14,6 +14,7 @@ if __name__ == '__main__':
print(f'{infile} is not a file or directory, exiting.')
os.exit(2)
file_id, token = encrypt_file(args.infile)
print(f'/{file_id}/{token}/{os.path.basename(args.infile)}')
download_filename = os.path.basename(args.infile)
file_id, token = encrypt_file(args.infile, download_filename)
print(f'/{file_id}/{token}/{download_filename}')