From fb2f3bcc2a6ec010e1ffc33b78833234f57da11a Mon Sep 17 00:00:00 2001
From: jaseg <git@jaseg.net>
Date: Tue, 25 Jun 2019 14:22:13 +0900
Subject: Verify download filenames in URL

---
 encrypt.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

(limited to 'encrypt.py')

diff --git a/encrypt.py b/encrypt.py
index 80cedf3..e813d36 100644
--- a/encrypt.py
+++ b/encrypt.py
@@ -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}')
     
-- 
cgit