120 lines
3.0 KiB
Python
120 lines
3.0 KiB
Python
import os
|
|
|
|
from flask import (
|
|
Flask,
|
|
render_template,
|
|
jsonify,
|
|
abort,
|
|
send_file,
|
|
request,
|
|
redirect,
|
|
flash,
|
|
)
|
|
from sqlalchemy import func
|
|
from base64 import standard_b64decode, standard_b64encode
|
|
from io import BytesIO
|
|
|
|
from model import db, Post, Category, ImageBase64
|
|
|
|
# Check for environment variable
|
|
env_vars = ["DATABASE_URL", "PASSWORD", "SECRET_KEY"]
|
|
|
|
for env_var in env_vars:
|
|
if not os.getenv(env_var):
|
|
raise RuntimeError(f"{env_var} is not set")
|
|
|
|
app = Flask(__name__)
|
|
|
|
app.config["SQLALCHEMY_DATABASE_URI"] = os.getenv("DATABASE_URL")
|
|
app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False
|
|
app.config["JSON_SORT_KEYS"] = False
|
|
app.secret_key = os.getenv("SECRET_KEY")
|
|
|
|
# Bind db to application
|
|
db.init_app(app)
|
|
|
|
|
|
@app.route("/")
|
|
def index():
|
|
posts = Post.get_posts()
|
|
# print(posts[0].title)
|
|
|
|
return render_template("main/index.html", posts=posts)
|
|
|
|
|
|
@app.route("/<string:category_name>")
|
|
def category(category_name):
|
|
category_item = Category.query.filter(
|
|
func.lower(Category.name) == category_name.replace("_", " ")
|
|
).first()
|
|
|
|
if not category_item:
|
|
return abort(404)
|
|
|
|
posts = Post.get_posts_with_category(category_item.id)
|
|
|
|
return render_template("main/index.html", posts=posts)
|
|
|
|
|
|
@app.route("/contact")
|
|
def contact():
|
|
return render_template("main/contact.html")
|
|
|
|
|
|
@app.route("/api/post/<int:post_id>")
|
|
def get_post(post_id):
|
|
post = Post.query.get(post_id)
|
|
|
|
if post:
|
|
return jsonify({"success": True, "post": post.serialize})
|
|
else:
|
|
return jsonify({"success": False})
|
|
|
|
|
|
@app.route("/images/<string:filename>")
|
|
def get_image(filename):
|
|
image_b64 = ImageBase64.query.filter(ImageBase64.filename == filename).first()
|
|
|
|
if not image_b64:
|
|
return abort(404)
|
|
|
|
image = standard_b64decode(image_b64.data)
|
|
|
|
return send_file(
|
|
BytesIO(image), mimetype=image_b64.mimetype, attachment_filename=filename
|
|
)
|
|
|
|
|
|
@app.route("/adm/uploadfile", methods=["POST", "GET"])
|
|
def file_uploaded():
|
|
if request.method == "POST":
|
|
if not request.form.get("password") == os.getenv("PASSWORD"):
|
|
flash("Wrong Password")
|
|
return redirect("/adm/uploadfile")
|
|
|
|
# check if the post request has the file part
|
|
if "files" not in request.files:
|
|
return redirect("/adm/uploadfile")
|
|
|
|
files = request.files.getlist("files")
|
|
|
|
# if user does not select file, browser also
|
|
# submit an empty part without filename
|
|
for file in files:
|
|
if file.filename == "":
|
|
flash("No files")
|
|
return redirect("/adm/uploadfile")
|
|
data = standard_b64encode(file.read()).decode()
|
|
database_object = ImageBase64(
|
|
filename=file.filename, mimetype=file.mimetype, data=data
|
|
)
|
|
db.session.add(database_object)
|
|
|
|
db.session.commit()
|
|
return render_template("adm/uploadfile.html")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app.app_context().push()
|
|
db.create_all()
|