forked from lza_menace/suchwow
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.
118 lines
4.2 KiB
118 lines
4.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 secrets import token_urlsafe
|
|
from suchwow import wownero
|
|
from suchwow.models import Post, Comment
|
|
from suchwow.utils.decorators import login_required, profile_required
|
|
from suchwow.utils.helpers import allowed_file
|
|
|
|
|
|
bp = Blueprint("post", "post")
|
|
|
|
@bp.route("/posts/top")
|
|
def top():
|
|
top_posts = {}
|
|
posts = Post.select()
|
|
for post in posts:
|
|
transfers = []
|
|
incoming = wownero.Wallet().incoming_transfers(post.account_index)
|
|
if "transfers" in incoming:
|
|
for xfer in incoming["transfers"]:
|
|
transfers.append(wownero.from_atomic(xfer["amount"]))
|
|
total = sum(transfers)
|
|
if total > 0:
|
|
top_posts[float(total)] = post
|
|
return render_template("post/top.html", posts=sorted(top_posts.items(), reverse=True))
|
|
|
|
@bp.route("/post/<id>")
|
|
def read(id):
|
|
if Post.filter(id=id):
|
|
wallet = wownero.Wallet()
|
|
post = Post.get(id=id)
|
|
comments = Comment.select().where(Comment.post==post.id)
|
|
if wallet.connected:
|
|
address = wallet.get_address(account=post.account_index)
|
|
transfers = wallet.transfers(account=post.account_index)
|
|
else:
|
|
address = "?"
|
|
transfers = "?"
|
|
return render_template(
|
|
"post/read.html",
|
|
post=post,
|
|
address=address,
|
|
comments=comments,
|
|
transfers=transfers
|
|
)
|
|
else:
|
|
flash("No meme there, brah")
|
|
return redirect(url_for("index"))
|
|
|
|
@bp.route("/post/create", methods=["GET", "POST"])
|
|
@login_required
|
|
@profile_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 = "{}-{}".format(
|
|
token_urlsafe(12),
|
|
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)
|
|
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("/post/<id>/delete")
|
|
@login_required
|
|
def delete(id):
|
|
filtered = Post.filter(id=id)
|
|
user = session["auth"]["preferred_username"]
|
|
if filtered:
|
|
post = filtered.first()
|
|
if user == post.submitter:
|
|
post.delete_instance()
|
|
flash("Deleted that shit, brah!")
|
|
return redirect(url_for("index"))
|
|
else:
|
|
flash("You can't delete a meme you don't own, brah")
|
|
return redirect(url_for("post.read", id=post.id))
|
|
else:
|
|
flash("No meme there, brah")
|
|
return redirect(url_for("index"))
|
|
|
|
@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)
|