Add basedir variable that can be modified in deployment

diff --git a/server.py b/server.py
index 6b4350e24548db38f7940efc0c82e5016e3454fd..e2d20389af59d998ce8f76f8c10ea63be8df99b2 100644 (file)
--- 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())