views.py

from flask import Blueprint, render_template, request, flash, redirect
from models import Post
from datetime import datetime
from db import db
import boto3
import uuid
from dotenv import load_dotenv
import os

ALLOWED_EXTENSIONS = ['jpg', 'jpeg', 'png', 'heic']
# Variáveis boto3
load_dotenv()
s3 = boto3.client("s3", 
                aws_access_key_id=os.getenv('AWS_ACCESS_KEY'),
                aws_secret_access_key=os.getenv('AWS_ACCESS_KEY_SECRET'))
bucket_name = os.getenv('bucket')
regiao = "sa-east-1"

def allowed_files(filename):
    extensao_arq = filename.rsplit('.')[-1].lower()
    if extensao_arq in ALLOWED_EXTENSIONS:
        return True
    else:
        return False
    
def retorna_arquivo(arq):
    # Definição do arquivo
    new_filename = uuid.uuid4().hex + '.' + arq.filename.rsplit('.', 1)[1].lower()
    # Armazena a imagem no bucket aws
    s3.upload_fileobj(arq, bucket_name, new_filename)
    # Fim do tratamento da imagem
    url_img = f"https://{bucket_name}.s3.{regiao}.amazonaws.com/{new_filename}"
    return url_img


views = Blueprint("views", __name__)

@views.route("/sobre", methods=['GET', 'POST'])
def sobre():
    return render_template("sobre.html")

@views.route("/")
@views.route("/inicio", methods=['GET'])
def inicio():
    postes = Post.query.all()
    return render_template("inicio.html", posts=postes)

@views.route('/posts_div', methods=['GET'])
def retorna_posts():
    postes = Post.query.all()
    return render_template('posts_div.html', posts=postes)

@views.route("/adote", methods=['GET'])
def adote():
    return render_template("adote.html")


@views.route("/crie", methods=['GET', 'POST'])
def create_post():
    if request.method == "POST":
        titulo = request.form.get('titulo')
        text = request.form.get('text')
        url_insta = request.form.get('insta_url')
        uploaded_file = request.files["imagem"]

        # Checagem dos inputs
        if 'https://www.instagram.com/' not in url_insta:
            flash('Url do instagram está incorreta, verifique se a sua url começa com https://', category='error')
            return redirect('/crie')
        elif not text:
            flash('Descrição não pode estar vazia', category='error')
            return redirect('/crie')
        else:
            # Inicio do tratamento da imagem
            if not allowed_files(uploaded_file.filename):
                flash("Tipo de arquivo não permitido", category='error')
                return redirect('/crie')

            url_img = retorna_arquivo(uploaded_file)
            post = Post(title=titulo, 
                        text=text, 
                        date_created=datetime.now().strftime('%d/%m/%Y'),
                        foto=url_img,
                        url=url_insta
                        )

            db.session.add(post)
            db.session.commit()
            flash('Postagem criada!', category='success')
            return redirect('/')
    return render_template('criar_post.html')

@views.route("/deletar", methods=['GET', 'POST'])
def deletar():
    postes = Post.query.all()
    if request.method == "POST":
        post_id = request.form.get('post_id')
        delet_post = Post.query.get_or_404(post_id)

        # Deletar imagem com boto3
        filename = delet_post.foto
        url_img = f"https://{bucket_name}.s3.{regiao}.amazonaws.com/"

        new_filename = filename.strip(url_img)
        # Deleta imagem
        s3.delete_object(
            Bucket=bucket_name,
            Key=new_filename
        )
        db.session.delete(delet_post)
        db.session.commit()
        flash('Postagem deletada!', category='success')
        return redirect('/deletar')
    return render_template('deletar.html', posts=postes)

@views.route('/atualizar', methods=['GET', 'POST'])
def atualizar():
    postes = Post.query.all()
    if request.method == "POST":
        post_id = request.form.get('post_id')
        post_att = Post.query.get_or_404(post_id)
        novo_titulo = request.form.get('titulo')
        novo_texto = request.form.get('text')
        nova_foto = request.files["imagem"]
        nova_url_insta = request.form.get('insta_url')

        # Checagem dos inputs
        if 'https://www.instagram.com/' not in nova_url_insta:
            flash('Url do instagram está incorreta, verifique se a sua url começa com https://', category='error')
            return redirect('/atualizar')
        elif not novo_texto:
            flash('Descrição não pode estar vazia', category='error')
            return redirect('/atualizar')
        elif not allowed_files(nova_foto.filename):
                flash("Tipo de arquivo não permitido", category='error')
                return redirect('/atualizar')
        else:
            # Deletar imagem com boto3
            filename = post_att.foto
            url_img = f"https://{bucket_name}.s3.{regiao}.amazonaws.com/"

            new_filename = filename.strip(url_img)
            # Deleta imagem
            s3.delete_object(
                Bucket=bucket_name,
                Key=new_filename
            )

            # adicionar remoção do arquivo antigo e opção de manter outras opções originais
            url_img = retorna_arquivo(nova_foto)
            post_att.title = novo_titulo
            post_att.text = novo_texto
            post_att.date_created = datetime.now().strftime('%d/%m/%Y')
            post_att.foto = url_img
            post_att.url = nova_url_insta

            db.session.commit()
            flash('Postagem atualizada!', category='success')
    return render_template('atualizar.html', posts=postes)

# Cadastra Usuário
# @views.route('/cadastrar', methods=['GET', 'POST'])
# def cadastrar():
#     name = None
#     formulario = UserForm()
#     if formulario.validate_on_submit:
#         user = User.query.filter_by(name=formulario.name.data).first()
#         if formulario.name.data == None:
#             flash("Nome não pode estar vazio", category='error')
#         elif user is None:
#             user = User(name=formulario.name.data)
#             db.session.add(user)
#             db.session.commit()
#             flash("Usuário inserido com sucesso!")
#         elif formulario.name.data == user.name:
#             flash("Nome de usuário já existe", category='error')
#         name = formulario.name.data
#         formulario.name.data=""
#     our_users = User.query.order_by(User.date_added)
#     return render_template('#.html', form=formulario, name=name, our_users=our_users)