Merge pull request #60 from djedi/export-fix

Fix export permission error in docker

Author: m0ngr31

app/routes.py

@@ -1,4 +1,6 @@
+import os
 import zipfile
+
 from app import app, db, argon2
 from app.models import User, Note, Meta, aes_encrypt, aes_encrypt_old
 from flask import render_template, request, jsonify, abort, send_file
@@ -406,15 +408,19 @@ def export():
   if not user:
     abort(400)
 
-  zf = zipfile.ZipFile('export.zip', mode='w')
+  zip_location = app.config['EXPORT_FILE']
+  zf = zipfile.ZipFile(zip_location, mode='w')
+  os.chmod(zip_location, 0o755)
   notes = user.notes
   for note in notes:
     ret_note = note.serialize
     zf.writestr(ret_note['title'] + '.md', ret_note['data'], zipfile.ZIP_DEFLATED)
     print(ret_note)
   zf.close()
 
-  return send_file('../export.zip', as_attachment=True)
+  rval = send_file(zip_location, as_attachment=True)
+  os.remove(zip_location)
+  return rval
 
 
 @app.route('/', defaults={'path': ''})

config.py

@@ -9,4 +9,5 @@ class Config(object):
   PREVENT_SIGNUPS = os.environ.get('PREVENT_SIGNUPS', False)
   SQLALCHEMY_DATABASE_URI = os.environ.get('DATABASE_URI') or 'sqlite:///' + os.path.join(basedir + '/config', 'app.db')
   SQLALCHEMY_TRACK_MODIFICATIONS = False
-  JWT_ACCESS_TOKEN_EXPIRES = datetime.timedelta(days=7)
+  JWT_ACCESS_TOKEN_EXPIRES = datetime.timedelta(days=7)
+  EXPORT_FILE = os.path.join(basedir, 'config', 'export.zip')