205 lines
8.3 KiB
Python
205 lines
8.3 KiB
Python
import pandas as pd
|
|
import io
|
|
from flask import Blueprint, render_template, request, send_file, flash
|
|
from app.utils.helpers import login_required
|
|
|
|
from app.models.subcontractor_model import Subcontractor
|
|
|
|
from app.models.manhole_excavation_model import ManholeExcavation
|
|
from app.models.trench_excavation_model import TrenchExcavation
|
|
from app.models.manhole_domestic_chamber_model import ManholeDomesticChamber
|
|
from app.models.laying_model import Laying
|
|
|
|
from app.models.mh_ex_client_model import ManholeExcavationClient
|
|
from app.models.tr_ex_client_model import TrenchExcavationClient
|
|
from app.models.mh_dc_client_model import ManholeDomesticChamberClient
|
|
from app.models.laying_client_model import LayingClient
|
|
|
|
|
|
# --- BLUEPRINT DEFINITION ---
|
|
file_report_bp = Blueprint("file_report", __name__, url_prefix="/file")
|
|
|
|
# --- Client class ---
|
|
class ClientBill:
|
|
def __init__(self):
|
|
self.df_tr = pd.DataFrame()
|
|
self.df_mh = pd.DataFrame()
|
|
self.df_dc = pd.DataFrame()
|
|
self.df_laying = pd.DataFrame()
|
|
|
|
def Fetch(self, RA_Bill_No):
|
|
trench = TrenchExcavationClient.query.filter_by(RA_Bill_No=RA_Bill_No).all()
|
|
mh = ManholeExcavationClient.query.filter_by(RA_Bill_No=RA_Bill_No).all()
|
|
dc = ManholeDomesticChamberClient.query.filter_by(RA_Bill_No=RA_Bill_No).all()
|
|
lay = LayingClient.query.filter_by(RA_Bill_No=RA_Bill_No).all()
|
|
|
|
self.df_tr = pd.DataFrame([c.serialize() for c in trench])
|
|
self.df_mh = pd.DataFrame([c.serialize() for c in mh])
|
|
self.df_dc = pd.DataFrame([c.serialize() for c in dc])
|
|
self.df_laying = pd.DataFrame([c.serialize() for c in lay])
|
|
|
|
drop_cols = ["id", "created_at", "_sa_instance_state"]
|
|
for df in [self.df_tr, self.df_mh, self.df_dc, self.df_laying]:
|
|
if not df.empty:
|
|
df.drop(columns=drop_cols, errors="ignore", inplace=True)
|
|
|
|
# --- Subcontractor class ---
|
|
class SubcontractorBill:
|
|
def __init__(self):
|
|
self.df_tr = pd.DataFrame()
|
|
self.df_mh = pd.DataFrame()
|
|
self.df_dc = pd.DataFrame()
|
|
self.df_laying = pd.DataFrame()
|
|
|
|
def Fetch(self, RA_Bill_No=None, subcontractor_id=None):
|
|
filters = {}
|
|
if subcontractor_id:
|
|
filters["subcontractor_id"] = subcontractor_id
|
|
if RA_Bill_No:
|
|
filters["RA_Bill_No"] = RA_Bill_No
|
|
|
|
trench = TrenchExcavation.query.filter_by(**filters).all()
|
|
mh = ManholeExcavation.query.filter_by(**filters).all()
|
|
dc = ManholeDomesticChamber.query.filter_by(**filters).all()
|
|
lay = Laying.query.filter_by(**filters).all()
|
|
|
|
self.df_tr = pd.DataFrame([c.serialize() for c in trench])
|
|
self.df_mh = pd.DataFrame([c.serialize() for c in mh])
|
|
self.df_dc = pd.DataFrame([c.serialize() for c in dc])
|
|
self.df_laying = pd.DataFrame([c.serialize() for c in lay])
|
|
|
|
drop_cols = ["id", "created_at", "_sa_instance_state"]
|
|
for df in [self.df_tr, self.df_mh, self.df_dc, self.df_laying]:
|
|
if not df.empty:
|
|
df.drop(columns=drop_cols, errors="ignore", inplace=True)
|
|
|
|
|
|
# --- subcontractor report only ---
|
|
@file_report_bp.route("/Subcontractor_report", methods=["GET", "POST"])
|
|
@login_required
|
|
def report_file():
|
|
subcontractors = Subcontractor.query.all()
|
|
tables = None
|
|
selected_sc_id = None
|
|
ra_bill_no = None
|
|
download_all = False
|
|
|
|
if request.method == "POST":
|
|
subcontractor_id = request.form.get("subcontractor_id")
|
|
ra_bill_no = request.form.get("ra_bill_no")
|
|
download_all = request.form.get("download_all") == "true"
|
|
action = request.form.get("action")
|
|
|
|
if not subcontractor_id:
|
|
flash("Please select a subcontractor.", "danger")
|
|
return render_template("subcontractor_report.html", subcontractors=subcontractors)
|
|
|
|
subcontractor = Subcontractor.query.get(subcontractor_id)
|
|
bill_gen = SubcontractorBill()
|
|
|
|
if download_all:
|
|
bill_gen.Fetch(subcontractor_id=subcontractor_id)
|
|
file_name = f"{subcontractor.subcontractor_name}_ALL_BILLS.xlsx"
|
|
else:
|
|
if not ra_bill_no:
|
|
flash("Please enter an RA Bill Number.", "danger")
|
|
return render_template("subcontractor_report.html", subcontractors=subcontractors)
|
|
bill_gen.Fetch(RA_Bill_No=ra_bill_no, subcontractor_id=subcontractor_id)
|
|
file_name = f"{subcontractor.subcontractor_name}_RA_{ra_bill_no}_Report.xlsx"
|
|
|
|
if bill_gen.df_tr.empty and bill_gen.df_mh.empty and bill_gen.df_dc.empty:
|
|
flash("No data found for this selection.", "warning")
|
|
return render_template("subcontractor_report.html", subcontractors=subcontractors)
|
|
|
|
# If download is clicked, return file immediately
|
|
if action == "download":
|
|
output = io.BytesIO()
|
|
with pd.ExcelWriter(output, engine="xlsxwriter") as writer:
|
|
bill_gen.df_tr.to_excel(writer, index=False, sheet_name="Tr.Ex.")
|
|
bill_gen.df_mh.to_excel(writer, index=False, sheet_name="MH.Ex.")
|
|
bill_gen.df_dc.to_excel(writer, index=False, sheet_name="MH & DC")
|
|
bill_gen.df_laying.to_excel(writer, index=False, sheet_name="Laying")
|
|
output.seek(0)
|
|
return send_file(output, download_name=file_name, as_attachment=True)
|
|
|
|
# We add bootstrap classes directly to the pandas output
|
|
table_classes = "table table-bordered table-striped table-hover table-sm mb-0"
|
|
tables = {
|
|
"tr": bill_gen.df_tr.to_html(classes=table_classes, index=False),
|
|
"mh": bill_gen.df_mh.to_html(classes=table_classes, index=False),
|
|
"dc": bill_gen.df_dc.to_html(classes=table_classes, index=False),
|
|
"laying": bill_gen.df_laying.to_html(classes=table_classes, index=False)
|
|
}
|
|
selected_sc_id = subcontractor_id
|
|
|
|
return render_template(
|
|
"subcontractor_report.html",
|
|
subcontractors=subcontractors,
|
|
tables=tables,
|
|
selected_sc_id=selected_sc_id,
|
|
ra_bill_no=ra_bill_no,
|
|
download_all=download_all
|
|
)
|
|
|
|
|
|
# --- CLIENT REPORT (PREVIEW + DOWNLOAD) ---
|
|
@file_report_bp.route("/client_report", methods=["GET", "POST"])
|
|
@login_required
|
|
def client_report():
|
|
|
|
tables = {"tr": None, "mh": None, "dc": None, "laying": None}
|
|
ra_val = ""
|
|
|
|
if request.method == "POST":
|
|
|
|
# ⚠ MUST match HTML name
|
|
RA_Bill_No = request.form.get("RA_Bill_No")
|
|
action = request.form.get("action")
|
|
ra_val = RA_Bill_No
|
|
|
|
if not RA_Bill_No:
|
|
flash("Please enter RA Bill No.", "danger")
|
|
return render_template("client_report.html", tables=tables, ra_val=ra_val)
|
|
|
|
# -------- FETCH CLIENT DATA --------
|
|
bill_gen = ClientBill()
|
|
bill_gen.Fetch(RA_Bill_No)
|
|
|
|
# If no data
|
|
if (
|
|
bill_gen.df_tr.empty and
|
|
bill_gen.df_mh.empty and
|
|
bill_gen.df_dc.empty and
|
|
bill_gen.df_laying.empty
|
|
):
|
|
flash(f"No Client records found for RA Bill {RA_Bill_No}", "warning")
|
|
return render_template("client_report.html", tables=tables, ra_val=ra_val)
|
|
|
|
# -------- DOWNLOAD --------
|
|
if action == "download":
|
|
|
|
output = io.BytesIO()
|
|
|
|
with pd.ExcelWriter(output, engine="xlsxwriter") as writer:
|
|
bill_gen.df_tr.to_excel(writer, index=False, sheet_name="Trench")
|
|
bill_gen.df_mh.to_excel(writer, index=False, sheet_name="MH")
|
|
bill_gen.df_dc.to_excel(writer, index=False, sheet_name="MH & DC")
|
|
bill_gen.df_laying.to_excel(writer, index=False, sheet_name="Laying")
|
|
|
|
output.seek(0)
|
|
|
|
return send_file(
|
|
output,
|
|
download_name=f"Client_RA_{RA_Bill_No}_Report.xlsx",
|
|
as_attachment=True
|
|
)
|
|
|
|
# -------- PREVIEW --------
|
|
table_class = "table table-bordered table-striped table-hover table-sm"
|
|
|
|
tables["tr"] = bill_gen.df_tr.to_html(classes=table_class, index=False)
|
|
tables["mh"] = bill_gen.df_mh.to_html(classes=table_class, index=False)
|
|
tables["dc"] = bill_gen.df_dc.to_html(classes=table_class, index=False)
|
|
tables["laying"] = bill_gen.df_laying.to_html(classes=table_class, index=False)
|
|
|
|
return render_template("client_report.html", tables=tables, ra_val=ra_val) |