From bf86e740bd1f025af3158ffdaaa05c25d7a2a37a Mon Sep 17 00:00:00 2001
From: jaseg <git@jaseg.net>
Date: Wed, 26 Jun 2019 16:36:20 +0900
Subject: Improve encrypt command line interface, add progress bar

---
 server.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'server.py')

diff --git a/server.py b/server.py
index 694b1ac..4d8e4dc 100644
--- a/server.py
+++ b/server.py
@@ -8,6 +8,7 @@ from flask import Flask, abort, request, Response
 import filecrypt
 
 app = Flask(__name__)
+app.config.from_envvar('SECURE_DOWNLOAD_SETTINGS')
 BASE64_RE = re.compile('^[A-Za-z0-9+-_]+=*$')
 
 @app.route('/<file_id>/<token>/<filename>')
@@ -17,7 +18,7 @@ def download(file_id, token, filename):
     if not BASE64_RE.match(token) or len(token) != filecrypt.TOKEN_LENGTH:
         abort(400, 'Invalid token format')
 
-    path = f'{file_id}.enc'
+    path = f'{app.config["SERVE_PATH"]}/{file_id}.enc'
     if not os.path.isfile(path):
         abort(403) # forbidden
 
-- 
cgit