From def4dc70a33724ce154019552da1b0a814c44684 Mon Sep 17 00:00:00 2001 From: Christoph Egger Date: Tue, 25 Nov 2014 20:20:06 +0100 Subject: [PATCH] Add basedir variable that can be modified in deployment --- server.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/server.py b/server.py index 6b4350e..e2d2038 100644 --- a/server.py +++ b/server.py @@ -4,7 +4,9 @@ import hashlib import pgpdump import glob import base64 +import os.path +basedir = '.' app = Flask(__name__) @app.template_filter('base64') @@ -16,14 +18,14 @@ def add(): keytext = request.form['keytext'] pgp = pgpdump.AsciiData(keytext) fingerprint = pgp.packets().next().fingerprint - with open('data/%s.asc' % fingerprint, 'w') as f: + with open(os.path.join(basedir, 'data/%s.asc' % fingerprint), 'w') as f: f.write(keytext) return Response("OK", mimetype="text/plain") @app.route('/') def index(): result = dict() - for entry in glob.glob('data/*.asc'): + for entry in glob.glob(os.path.join(basedir, 'data/*.asc')): pgp = pgpdump.AsciiData(file(entry).read()) uids = [] photoids = [] @@ -44,7 +46,7 @@ def index(): @app.route('/download/') def download(): result = [] - for entry in glob.glob('data/*.asc'): + for entry in glob.glob(os.path.join(basedir, 'data/*.asc')): with open(entry) as f: result.append(f.read()) -- 2.39.2