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

71 lines
2.6 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 import wownero
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):
wallet = wownero.Wallet()
post = Post.get(id=id)
if wallet.connected:
address = wallet.addresses(account=post.account_index)[0]
else:
address = "?"
return render_template("post/read.html", post=post, address=address)
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_base = path.join(current_app.config["DATA_FOLDER"], "uploads")
save_path = path.join(save_path_base, filename)
file.save(save_path)
# gen wallet
try:
wallet = wownero.Wallet()
account_index = wallet.new_account()
except:
account_index = 0
post = Post(
title=post_title,
text=request.form.get("text", ""),
submitter=session["auth"]["preferred_username"],
image_name=filename,
account_index=account_index,
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):
file_path = path.join(current_app.config["DATA_FOLDER"], "uploads")
return send_from_directory(file_path, filename)