Update app.py

app.py

@@ -1,59 +1,72 @@
-from flask import Flask, render_template, request, redirect, url_for
-from supabase import create_client, Client
+from flask import Flask, render_template, request, redirect, url_for, send_file
+from supabase_client import supabase
 from datetime import datetime
+import pdfkit
 import os
+from io import BytesIO
 
 app = Flask(__name__)
 
-# Configuração Supabase
-SUPABASE_URL = os.getenv("SUPABASE_URL")
-SUPABASE_KEY = os.getenv("SUPABASE_KEY")
-supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
+pdfkit_config = pdfkit.configuration(wkhtmltopdf="/usr/bin/wkhtmltopdf")
 
-# --- Rota: Listagem de orçamentos ---
 @app.route("/")
-def index():
-    response = supabase.table("orcamentos").select("*").order("id", desc=True).execute()
+def listar_orcamentos():
+    response = supabase.table("orcamentos").select("*").execute()
     orcamentos = response.data
-    return render_template("index.html", orcamentos=orcamentos)
+    return render_template("list.html", orcamentos=orcamentos)
 
-# --- Rota: Novo Orçamento ---
 @app.route("/novo", methods=["GET", "POST"])
-def novo():
+def novo_orcamento():
     if request.method == "POST":
-        dados = {
+        data = {
             "cliente": request.form["cliente"],
             "endereco": request.form["endereco"],
-            "area_total": request.form["area_total"],
-            "quartos": request.form["quartos"],
-            "banheiros": request.form["banheiros"],
-            "total": calcular_total(request.form),  # Função que calcula o total
-            "data_criacao": datetime.now().isoformat()
+            "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()
         }
-        supabase.table("orcamentos").insert(dados).execute()
-        return redirect(url_for('index'))
-    return render_template("novo.html")
-
-# --- Rota: Detalhe ---
-@app.route("/orcamento/<int:id>")
-def orcamento(id):
-    response = supabase.table("orcamentos").select("*").eq("id", id).single().execute()
-    orcamento = response.data
-    return render_template("relatorio.html", orcamento=orcamento)
-
-# --- Rota: Excluir ---
-@app.route("/delete/<int:id>")
-def delete(id):
+        supabase.table("orcamentos").insert(data).execute()
+        return redirect(url_for("listar_orcamentos"))
+    return render_template("form.html", orcamento=None)
+
+@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"])
+        }
+        supabase.table("orcamentos").update(data).eq("id", id).execute()
+        return redirect(url_for("listar_orcamentos"))
+    return render_template("form.html", orcamento=orcamento)
+
+@app.route("/excluir/<int:id>")
+def excluir_orcamento(id):
     supabase.table("orcamentos").delete().eq("id", id).execute()
-    return redirect(url_for('index'))
+    return redirect(url_for("listar_orcamentos"))
+
+@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)
 
-# --- Função de cálculo (exemplo simplificado) ---
-def calcular_total(form):
-    area = float(form.get("area_total", 0))
-    quartos = int(form.get("quartos", 0))
-    banheiros = int(form.get("banheiros", 0))
-    base_m2 = 500
-    return area * base_m2 + quartos * 3000 + banheiros * 4000
+@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)
+    app.run(host="0.0.0.0", port=int(os.environ.get("PORT", 7860)))