Update app.py

app.py

@@ -1,72 +1,91 @@
-from flask import Flask, render_template, request, redirect, url_for, send_file
+from flask import Flask, render_template, request, redirect, session, url_for
 from supabase_client import supabase
-from datetime import datetime
-import pdfkit
 import os
-from io import BytesIO
 
 app = Flask(__name__)
+app.secret_key = os.urandom(24)
 
-pdfkit_config = pdfkit.configuration(wkhtmltopdf="/usr/bin/wkhtmltopdf")
+# -------- Autenticação --------
+@app.route('/login', methods=['GET', 'POST'])
+def login():
+    if request.method == 'POST':
+        email = request.form['email']
+        password = request.form['password']
+        result = supabase.auth.sign_in_with_password({'email': email, 'password': password})
+        if result.user:
+            session['user'] = result.user.id
+            return redirect('/dashboard')
+        else:
+            return "Login inválido"
+    return render_template('login.html')
 
-@app.route("/")
-def listar_orcamentos():
-    response = supabase.table("orcamentos").select("*").execute()
+@app.route('/logout')
+def logout():
+    session.clear()
+    return redirect('/login')
+
+def login_required(f):
+    def wrap(*args, **kwargs):
+        if 'user' in session:
+            return f(*args, **kwargs)
+        else:
+            return redirect('/login')
+    wrap.__name__ = f.__name__
+    return wrap
+
+# -------- Dashboard --------
+@app.route('/dashboard')
+@login_required
+def dashboard():
+    user_id = session['user']
+    response = supabase.table('orcamentos').select('*').eq('user_id', user_id).execute()
     orcamentos = response.data
-    return render_template("list.html", orcamentos=orcamentos)
+    return render_template('dashboard.html', orcamentos=orcamentos)
 
-@app.route("/novo", methods=["GET", "POST"])
-def novo_orcamento():
-    if request.method == "POST":
-        data = {
-            "cliente": request.form["cliente"],
-            "endereco": request.form["endereco"],
-            "area_total": float(request.form["area_total"]),
-            "quartos": int(request.form["quartos"]),
-            "banheiros": int(request.form["banheiros"]),
-            "inicio": request.form["inicio"],
-            "detalhes": request.form["detalhes"],
-            "total": float(request.form["total"]),
-            "criado_em": datetime.now().isoformat()
+# -------- Criar Orçamento --------
+@app.route('/orcamentos/novo', methods=['GET', 'POST'])
+@login_required
+def criar_orcamento():
+    if request.method == 'POST':
+        dados = {
+            'user_id': session['user'],
+            'cliente': request.form['cliente'],
+            'endereco': request.form['endereco'],
+            'valor_total': request.form['valor_total']
         }
-        supabase.table("orcamentos").insert(data).execute()
-        return redirect(url_for("listar_orcamentos"))
-    return render_template("form.html", orcamento=None)
+        supabase.table('orcamentos').insert(dados).execute()
+        return redirect('/dashboard')
+    return render_template('create_orcamento.html')
 
-@app.route("/editar/<int:id>", methods=["GET", "POST"])
-def editar_orcamento(id):
-    orcamento = supabase.table("orcamentos").select("*").eq("id", id).single().execute().data
-    if request.method == "POST":
-        data = {
-            "cliente": request.form["cliente"],
-            "endereco": request.form["endereco"],
-            "area_total": float(request.form["area_total"]),
-            "quartos": int(request.form["quartos"]),
-            "banheiros": int(request.form["banheiros"]),
-            "inicio": request.form["inicio"],
-            "detalhes": request.form["detalhes"],
-            "total": float(request.form["total"])
+# -------- Editar Orçamento --------
+@app.route('/orcamentos/<int:orcamento_id>/editar', methods=['GET', 'POST'])
+@login_required
+def editar_orcamento(orcamento_id):
+    if request.method == 'POST':
+        update_data = {
+            'cliente': request.form['cliente'],
+            'endereco': request.form['endereco'],
+            'valor_total': request.form['valor_total']
         }
-        supabase.table("orcamentos").update(data).eq("id", id).execute()
-        return redirect(url_for("listar_orcamentos"))
-    return render_template("form.html", orcamento=orcamento)
+        supabase.table('orcamentos').update(update_data).eq('id', orcamento_id).execute()
+        return redirect('/dashboard')
+    orcamento = supabase.table('orcamentos').select('*').eq('id', orcamento_id).single().execute().data
+    return render_template('edit_orcamento.html', orcamento=orcamento)
 
-@app.route("/excluir/<int:id>")
-def excluir_orcamento(id):
-    supabase.table("orcamentos").delete().eq("id", id).execute()
-    return redirect(url_for("listar_orcamentos"))
+# -------- Ver Orçamento --------
+@app.route('/orcamentos/<int:orcamento_id>')
+@login_required
+def visualizar_orcamento(orcamento_id):
+    orcamento = supabase.table('orcamentos').select('*').eq('id', orcamento_id).single().execute().data
+    return render_template('view_orcamento.html', orcamento=orcamento)
 
-@app.route("/visualizar/<int:id>")
-def visualizar_orcamento(id):
-    orcamento = supabase.table("orcamentos").select("*").eq("id", id).single().execute().data
-    return render_template("relatorio.html", o=orcamento)
+# -------- Excluir Orçamento --------
+@app.route('/orcamentos/<int:orcamento_id>/excluir', methods=['POST'])
+@login_required
+def excluir_orcamento(orcamento_id):
+    supabase.table('orcamentos').delete().eq('id', orcamento_id).execute()
+    return redirect('/dashboard')
 
-@app.route("/pdf/<int:id>")
-def gerar_pdf(id):
-    o = supabase.table("orcamentos").select("*").eq("id", id).single().execute().data
-    html = render_template("contrato.html", o=o)
-    pdf = pdfkit.from_string(html, False, configuration=pdfkit_config)
-    return send_file(BytesIO(pdf), as_attachment=True, download_name=f"Contrato_{o['cliente']}.pdf")
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860)
 
-if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=int(os.environ.get("PORT", 7860)))