You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
This repo is archived. You can view files and clone it, but cannot push or open issues/pull-requests.
suchwow/suchwow/routes/post.py

58 lines
2.2 KiB

from os import path
from flask import render_template, Blueprint, request, session, flash
from flask import send_from_directory, redirect, url_for, current_app
from werkzeug.utils import secure_filename
from suchwow.models import Post
from suchwow.utils.decorators import login_required
from suchwow.utils.helpers import allowed_file
bp = Blueprint("post", "post")
@bp.route("/post/<id>")
def read(id):
if Post.filter(id=id):
post = Post.get(Post.id == id)
return render_template("post/read.html", post=post)
else:
return "no meme there brah"
@bp.route("/post/create", methods=["GET", "POST"])
@login_required
def create():
if request.method == "POST":
post_title = request.form.get("title")
# check if the post request has the file part
if "file" not in request.files:
flash("You didn't upload a caliente meme, bro! You're fuckin up!")
return redirect(request.url)
file = request.files["file"]
# if user does not select file, browser also
# submit an empty part without filename
if file.filename == "":
flash("You didn't upload a caliente meme, bro! You're fuckin up!")
return redirect(request.url)
if post_title is "":
flash("You didn't give your meme a spicy title, bro! You're fuckin up!")
return redirect(request.url)
if file and allowed_file(file.filename):
filename = secure_filename(file.filename)
save_path = path.join(current_app.config["UPLOAD_FOLDER"], filename)
file.save(save_path)
# gen wallet
post = Post(
title=post_title,
text=request.form.get("text", ""),
submitter=session["auth"]["preferred_username"],
image_name=filename,
account_index=0,
address_index=0
)
post.save()
return redirect(url_for("post.read", id=post.id))
return render_template("post/create.html")
@bp.route("/uploads/<path:filename>")
def uploaded_file(filename):
return send_from_directory(current_app.config["UPLOAD_FOLDER"], filename)