server.py 1.1 KB

1234567891011121314151617181920212223242526272829303132333435
  1. import os
  2. from flask import Flask, request, jsonify
  3. from werkzeug.utils import secure_filename
  4. from flask_cors import CORS
  5. UPLOAD_FOLDER = 'uploads'
  6. ALLOWED_EXTENSIONS = set(['txt', 'pdf', 'png', 'jpg', 'jpeg', 'gif'])
  7. app = Flask(__name__)
  8. app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
  9. CORS(app)
  10. def allowed_file(filename):
  11. return '.' in filename and \
  12. filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
  13. @app.route('/upload', methods=['POST'])
  14. def upload_file():
  15. if request.method == 'POST':
  16. # check if the post request has the file part
  17. print request.files
  18. if len(request.files) == 0:
  19. return jsonify(
  20. error="No file n request"
  21. ), 400
  22. for fi in request.files:
  23. file = request.files[fi]
  24. if file and allowed_file(file.filename):
  25. filename = secure_filename(file.filename)
  26. file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
  27. return jsonify(
  28. message="ok"
  29. ), 201
  30. if __name__ == '__main__':
  31. app.run(port=3020)