99 lines
2.8 KiB
Python
99 lines
2.8 KiB
Python
from flask import Flask, render_template, request, redirect, send_from_directory, jsonify
|
|
from config import app, db, bcrypt
|
|
from models import Contact, User, Orders
|
|
from werkzeug.utils import secure_filename
|
|
from werkzeug.exceptions import RequestEntityTooLarge
|
|
import os
|
|
|
|
|
|
@app.route('/')
|
|
def index():
|
|
files = os.listdir(app.config['UPLOAD_DIRECTORY'])
|
|
images = []
|
|
|
|
for file in files:
|
|
if os.path.splitext(file)[1].lower() in app.config['ALLOWED_EXTENSIONS']:
|
|
images.append(file)
|
|
return render_template('index.html', images=images)
|
|
|
|
@app.route('/upload', methods=["POST"])
|
|
def upload():
|
|
try:
|
|
file = request.files['file']
|
|
file_extension = os.path.splitext(file.filename)[1] # [0] is the filename
|
|
|
|
# save the file if it exists
|
|
if file:
|
|
if file_extension not in app.config['ALLOWED_EXTENSIONS']:
|
|
return 'File is not an image'
|
|
file.save(os.path.join(
|
|
app.config['UPLOAD_DIRECTORY'],
|
|
secure_filename(file.filename)
|
|
))
|
|
except RequestEntityTooLarge: # except Exception would catch all errors instead of specified
|
|
return 'File is larger than the limit of 16mb.'
|
|
|
|
return redirect('/')
|
|
|
|
@app.route("/mail", methods=["POST"])
|
|
def mail():
|
|
email = request.json["email"]
|
|
password = request.json["password"]
|
|
order = request.json["order"]
|
|
|
|
|
|
if email is None:
|
|
return jsonify({"error": "Unauthorized"}), 401
|
|
|
|
if not bcrypt.check_password_hash(user.password, password):
|
|
return jsonify({"error": "Unauthorized"}), 401
|
|
|
|
# Cache session with redis
|
|
session["user_id"] = user.id
|
|
|
|
return jsonify({
|
|
"id": user.id,
|
|
"email": user.email
|
|
})
|
|
|
|
|
|
# Handle Images
|
|
@app.route('/serve-image/<filename>', methods=['GET'])
|
|
def serve_image(filename):
|
|
return send_from_directory(app.config['UPLOAD_DIRECTORY'], filename)
|
|
|
|
|
|
# Register User
|
|
@app.route("/new-order", methods=["POST"])
|
|
def register_user():
|
|
email = request.json["email"]
|
|
order = request.json["order"]
|
|
|
|
new_order= Orders(email=email, order=order)
|
|
|
|
# Write to db
|
|
try:
|
|
db.session.add(new_order) # add to db staging area
|
|
db.session.commit() # add to db and catch any errors that can happen with writing to db
|
|
except Exception as e:
|
|
return (jsonify({"message": str(e)}), 400)
|
|
|
|
|
|
return jsonify({
|
|
"email": email,
|
|
"order confirmed": order,
|
|
})
|
|
# Get orders
|
|
@app.route("/orders", methods=["GET"])
|
|
def orders():
|
|
orders = Orders.query.all()
|
|
json_orders = list(map(lambda x : x.to_json(), orders))
|
|
return jsonify({"orders": json_orders})
|
|
|
|
# Main
|
|
if __name__ == "__main__":
|
|
# Initialize database connection
|
|
with app.app_context():
|
|
db.create_all()
|
|
|
|
app.run(debug=True, port=5080) |