added subcontractor on same page ui changes
This commit is contained in:
@@ -1,132 +1,3 @@
|
|||||||
from flask import Blueprint, render_template, request, send_file, flash
|
|
||||||
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.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.utils.helpers import login_required
|
|
||||||
|
|
||||||
import pandas as pd
|
|
||||||
import io
|
|
||||||
from enum import Enum
|
|
||||||
|
|
||||||
# --- 1. DEFINE BLUEPRINT FIRST (Prevents NameError) ---
|
|
||||||
file_report_bp = Blueprint("file_report", __name__, url_prefix="/file")
|
|
||||||
|
|
||||||
class BillType(Enum):
|
|
||||||
Client = 1
|
|
||||||
Subcontractor = 2
|
|
||||||
|
|
||||||
# --- 2. DEFINE CLASSES ---
|
|
||||||
class SubcontractorBill:
|
|
||||||
def __init__(self):
|
|
||||||
self.df_tr = pd.DataFrame()
|
|
||||||
self.df_mh = pd.DataFrame()
|
|
||||||
self.df_dc = pd.DataFrame()
|
|
||||||
|
|
||||||
def Fetch(self, RA_Bill_No=None, subcontractor_id=None):
|
|
||||||
# Build dynamic filters based on what is provided
|
|
||||||
filters = {}
|
|
||||||
if subcontractor_id:
|
|
||||||
filters["subcontractor_id"] = subcontractor_id
|
|
||||||
if RA_Bill_No:
|
|
||||||
filters["RA_Bill_No"] = RA_Bill_No
|
|
||||||
|
|
||||||
# Query using the filters
|
|
||||||
trench = TrenchExcavation.query.filter_by(**filters).all()
|
|
||||||
mh = ManholeExcavation.query.filter_by(**filters).all()
|
|
||||||
dc = ManholeDomesticChamber.query.filter_by(**filters).all()
|
|
||||||
|
|
||||||
# Convert to DataFrames
|
|
||||||
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])
|
|
||||||
|
|
||||||
# Standardize columns and clean internal SQLAlchemy state
|
|
||||||
if not self.df_dc.empty and "MH_NO" in self.df_dc.columns:
|
|
||||||
self.df_dc.rename(columns={"MH_NO": "Node_No"}, inplace=True)
|
|
||||||
|
|
||||||
drop_cols = ["id", "created_at", "_sa_instance_state"]
|
|
||||||
for df in [self.df_tr, self.df_mh, self.df_dc]:
|
|
||||||
if not df.empty:
|
|
||||||
df.drop(columns=drop_cols, errors="ignore", inplace=True)
|
|
||||||
|
|
||||||
# --- Updated Route with "Download All" Support ---
|
|
||||||
@file_report_bp.route("/report", methods=["GET", "POST"])
|
|
||||||
@login_required
|
|
||||||
def report_file():
|
|
||||||
subcontractors = Subcontractor.query.all()
|
|
||||||
|
|
||||||
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" # Check for toggle
|
|
||||||
|
|
||||||
if not subcontractor_id:
|
|
||||||
flash("Please select a subcontractor.", "danger")
|
|
||||||
return render_template("report.html", subcontractors=subcontractors)
|
|
||||||
|
|
||||||
subcontractor = Subcontractor.query.get(subcontractor_id)
|
|
||||||
bill_gen = SubcontractorBill()
|
|
||||||
|
|
||||||
if download_all:
|
|
||||||
# Fetch EVERYTHING for this subcontractor
|
|
||||||
bill_gen.Fetch(subcontractor_id=subcontractor_id, RA_Bill_No=None)
|
|
||||||
file_name = f"{subcontractor.subcontractor_name}_ALL_BILLS.xlsx"
|
|
||||||
else:
|
|
||||||
if not ra_bill_no:
|
|
||||||
flash("Please enter an RA Bill Number or select 'Download All'.", "danger")
|
|
||||||
return render_template("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("report.html", subcontractors=subcontractors)
|
|
||||||
|
|
||||||
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")
|
|
||||||
|
|
||||||
output.seek(0)
|
|
||||||
return send_file(output, download_name=file_name, as_attachment=True,
|
|
||||||
mimetype="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")
|
|
||||||
|
|
||||||
return render_template("report.html", subcontractors=subcontractors)
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
class ClientBill:
|
|
||||||
def __init__(self):
|
|
||||||
self.df_tr = pd.DataFrame()
|
|
||||||
self.df_mh = pd.DataFrame()
|
|
||||||
self.df_dc = 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()
|
|
||||||
|
|
||||||
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])
|
|
||||||
|
|
||||||
# Standardize columns for merging
|
|
||||||
if not self.df_dc.empty and "MH_NO" in self.df_dc.columns:
|
|
||||||
self.df_dc.rename(columns={"MH_NO": "Node_No"}, inplace=True)
|
|
||||||
|
|
||||||
drop_cols = ["id", "created_at", "_sa_instance_state"]
|
|
||||||
for df in [self.df_tr, self.df_mh, self.df_dc]:
|
|
||||||
if not df.empty:
|
|
||||||
df.drop(columns=drop_cols, errors="ignore", inplace=True)
|
|
||||||
|
|
||||||
|
|
||||||
import pandas as pd
|
import pandas as pd
|
||||||
import io
|
import io
|
||||||
from flask import Blueprint, render_template, request, send_file, flash
|
from flask import Blueprint, render_template, request, send_file, flash
|
||||||
@@ -140,9 +11,10 @@ from app.models.mh_dc_client_model import ManholeDomesticChamberClient
|
|||||||
from app.utils.helpers import login_required
|
from app.utils.helpers import login_required
|
||||||
|
|
||||||
# --- BLUEPRINT DEFINITION ---
|
# --- BLUEPRINT DEFINITION ---
|
||||||
# Ensure this is unique to avoid conflicts
|
|
||||||
file_report_bp = Blueprint("file_report", __name__, url_prefix="/file")
|
file_report_bp = Blueprint("file_report", __name__, url_prefix="/file")
|
||||||
|
|
||||||
|
# --- DATA WRAPPERS ---
|
||||||
|
|
||||||
class ClientBill:
|
class ClientBill:
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self.df_tr = pd.DataFrame()
|
self.df_tr = pd.DataFrame()
|
||||||
@@ -158,7 +30,6 @@ class ClientBill:
|
|||||||
self.df_mh = pd.DataFrame([c.serialize() for c in mh])
|
self.df_mh = pd.DataFrame([c.serialize() for c in mh])
|
||||||
self.df_dc = pd.DataFrame([c.serialize() for c in dc])
|
self.df_dc = pd.DataFrame([c.serialize() for c in dc])
|
||||||
|
|
||||||
# Standardize columns for merging
|
|
||||||
if not self.df_dc.empty and "MH_NO" in self.df_dc.columns:
|
if not self.df_dc.empty and "MH_NO" in self.df_dc.columns:
|
||||||
self.df_dc.rename(columns={"MH_NO": "Node_No"}, inplace=True)
|
self.df_dc.rename(columns={"MH_NO": "Node_No"}, inplace=True)
|
||||||
|
|
||||||
@@ -173,10 +44,16 @@ class SubcontractorBill:
|
|||||||
self.df_mh = pd.DataFrame()
|
self.df_mh = pd.DataFrame()
|
||||||
self.df_dc = pd.DataFrame()
|
self.df_dc = pd.DataFrame()
|
||||||
|
|
||||||
def Fetch(self, RA_Bill_No):
|
def Fetch(self, RA_Bill_No=None, subcontractor_id=None):
|
||||||
trench = TrenchExcavation.query.filter_by(RA_Bill_No=RA_Bill_No).all()
|
filters = {}
|
||||||
mh = ManholeExcavation.query.filter_by(RA_Bill_No=RA_Bill_No).all()
|
if subcontractor_id:
|
||||||
dc = ManholeDomesticChamber.query.filter_by(RA_Bill_No=RA_Bill_No).all()
|
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()
|
||||||
|
|
||||||
self.df_tr = pd.DataFrame([c.serialize() for c in trench])
|
self.df_tr = pd.DataFrame([c.serialize() for c in trench])
|
||||||
self.df_mh = pd.DataFrame([c.serialize() for c in mh])
|
self.df_mh = pd.DataFrame([c.serialize() for c in mh])
|
||||||
@@ -190,112 +67,174 @@ class SubcontractorBill:
|
|||||||
if not df.empty:
|
if not df.empty:
|
||||||
df.drop(columns=drop_cols, errors="ignore", inplace=True)
|
df.drop(columns=drop_cols, errors="ignore", inplace=True)
|
||||||
|
|
||||||
|
# --- ROUTES ---
|
||||||
|
|
||||||
|
# @file_report_bp.route("/report", methods=["GET", "POST"])
|
||||||
|
# @login_required
|
||||||
|
# def report_file():
|
||||||
|
# subcontractors = Subcontractor.query.all()
|
||||||
|
# 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"
|
||||||
|
|
||||||
|
# if not subcontractor_id:
|
||||||
|
# flash("Please select a subcontractor.", "danger")
|
||||||
|
# return render_template("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("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("report.html", subcontractors=subcontractors)
|
||||||
|
|
||||||
|
# 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")
|
||||||
|
# output.seek(0)
|
||||||
|
# return send_file(output, download_name=file_name, as_attachment=True, mimetype="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")
|
||||||
|
|
||||||
|
# return render_template("report.html", subcontractors=subcontractors)
|
||||||
|
@file_report_bp.route("/report", methods=["GET", "POST"])
|
||||||
|
@login_required
|
||||||
|
def report_file():
|
||||||
|
subcontractors = Subcontractor.query.all()
|
||||||
|
tables = None # Match the variable name used in HTML
|
||||||
|
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("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("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("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")
|
||||||
|
output.seek(0)
|
||||||
|
return send_file(output, download_name=file_name, as_attachment=True)
|
||||||
|
|
||||||
|
# PREVIEW LOGIC: Convert DataFrames to HTML for the 'tables' variable
|
||||||
|
# We add bootstrap classes directly to the pandas output
|
||||||
|
table_classes = "table table-hover table-bordered table-striped"
|
||||||
|
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)
|
||||||
|
}
|
||||||
|
selected_sc_id = subcontractor_id
|
||||||
|
|
||||||
|
return render_template(
|
||||||
|
"report.html",
|
||||||
|
subcontractors=subcontractors,
|
||||||
|
tables=tables, # This now matches your HTML
|
||||||
|
selected_sc_id=selected_sc_id,
|
||||||
|
ra_bill_no=ra_bill_no,
|
||||||
|
download_all=download_all
|
||||||
|
)
|
||||||
|
|
||||||
@file_report_bp.route("/client_vs_subcont", methods=["GET", "POST"])
|
@file_report_bp.route("/client_vs_subcont", methods=["GET", "POST"])
|
||||||
@login_required
|
@login_required
|
||||||
def client_vs_all_subcontractor():
|
def client_vs_all_subcontractor():
|
||||||
<<<<<<< HEAD
|
|
||||||
=======
|
|
||||||
# Initialize empty variables for the template
|
|
||||||
tables = {"tr": None, "mh": None, "dc": None}
|
tables = {"tr": None, "mh": None, "dc": None}
|
||||||
ra_val = ""
|
ra_val = ""
|
||||||
|
|
||||||
>>>>>>> 2636f2e (added client RA bill wise download report)
|
|
||||||
if request.method == "POST":
|
if request.method == "POST":
|
||||||
RA_Bill_No = request.form.get("RA_Bill_No")
|
RA_Bill_No = request.form.get("RA_Bill_No")
|
||||||
ra_val = RA_Bill_No
|
ra_val = RA_Bill_No
|
||||||
|
|
||||||
if not RA_Bill_No:
|
if not RA_Bill_No:
|
||||||
flash("Please enter RA Bill No.", "danger")
|
flash("Please enter RA Bill No.", "danger")
|
||||||
return render_template("generate_comparison_client_vs_subcont.html")
|
return render_template("generate_comparison_client_vs_subcont.html", tables=tables, ra_val=ra_val)
|
||||||
|
|
||||||
# ... (Keep your existing data fetching and processing logic exactly as is) ...
|
|
||||||
clientBill = ClientBill()
|
clientBill = ClientBill()
|
||||||
clientBill.Fetch(RA_Bill_No=RA_Bill_No)
|
clientBill.Fetch(RA_Bill_No=RA_Bill_No)
|
||||||
contractorBill = SubcontractorBill()
|
contractorBill = SubcontractorBill()
|
||||||
contractorBill.Fetch(RA_Bill_No=RA_Bill_No)
|
contractorBill.Fetch(RA_Bill_No=RA_Bill_No)
|
||||||
<<<<<<< HEAD
|
|
||||||
|
|
||||||
# Updated QTY lists to match model fields exactly
|
# --- SAFETY CHECK: Verify data exists before merging ---
|
||||||
qty_cols = [
|
if clientBill.df_tr.empty and clientBill.df_mh.empty:
|
||||||
"Soft_Murum_0_to_1_5_total", "Soft_Murum_1_5_to_3_0_total", "Soft_Murum_3_0_to_4_5_total",
|
flash(f"No Client records found for RA Bill {RA_Bill_No}", "warning")
|
||||||
"Hard_Murum_0_to_1_5_total", "Hard_Murum_1_5_to_3_0_total", "Hard_Murum_3_0_to_4_5_total",
|
return render_template("generate_comparison_client_vs_subcont.html", tables=tables, ra_val=ra_val)
|
||||||
"Soft_Rock_0_to_1_5_total", "Soft_Rock_1_5_to_3_0_total", "Soft_Rock_3_0_to_4_5_total",
|
|
||||||
"Hard_Rock_0_to_1_5_total", "Hard_Rock_1_5_to_3_0_total", "Hard_Rock_3_0_to_4_5_total",
|
|
||||||
"Hard_Rock_4_5_to_6_0_total", "Hard_Rock_6_0_to_7_5_total"
|
|
||||||
]
|
|
||||||
|
|
||||||
mh_dc_qty_cols = [
|
qty_cols = [...] # (Keep your existing list)
|
||||||
"d_0_to_1_5", "d_1_5_to_2_0", "d_2_0_to_2_5", "d_2_5_to_3_0",
|
mh_dc_qty_cols = [...] # (Keep your existing list)
|
||||||
"d_3_0_to_3_5", "d_3_5_to_4_0", "d_4_0_to_4_5", "d_4_5_to_5_0",
|
|
||||||
"d_5_0_to_5_5", "d_5_5_to_6_0", "d_6_0_to_6_5", "Domestic_Chambers"
|
|
||||||
]
|
|
||||||
|
|
||||||
# Aggregate Subcontractor Data safely
|
|
||||||
=======
|
|
||||||
|
|
||||||
# ... (Keep all your qty_cols, mh_dc_qty_cols, and aggregate_df function) ...
|
|
||||||
qty_cols = [...] # (Your existing list)
|
|
||||||
mh_dc_qty_cols = [...] # (Your existing list)
|
|
||||||
|
|
||||||
>>>>>>> 2636f2e (added client RA bill wise download report)
|
|
||||||
def aggregate_df(df, group_cols, sum_cols):
|
def aggregate_df(df, group_cols, sum_cols):
|
||||||
if df.empty: return pd.DataFrame()
|
if df.empty:
|
||||||
|
# Create an empty DF with the correct columns to avoid Merge/Key Errors
|
||||||
|
return pd.DataFrame(columns=group_cols + sum_cols)
|
||||||
existing_cols = [c for c in sum_cols if c in df.columns]
|
existing_cols = [c for c in sum_cols if c in df.columns]
|
||||||
|
# Ensure group_cols exist in the DF
|
||||||
|
for col in group_cols:
|
||||||
|
if col not in df.columns:
|
||||||
|
df[col] = "N/A" # Fill missing join keys
|
||||||
return df.groupby(group_cols, as_index=False)[existing_cols].sum()
|
return df.groupby(group_cols, as_index=False)[existing_cols].sum()
|
||||||
|
|
||||||
|
# Aggregate data
|
||||||
df_sub_tr_grp = aggregate_df(contractorBill.df_tr, ["Location", "MH_NO"], qty_cols)
|
df_sub_tr_grp = aggregate_df(contractorBill.df_tr, ["Location", "MH_NO"], qty_cols)
|
||||||
df_sub_mh_grp = aggregate_df(contractorBill.df_mh, ["Location", "MH_NO"], qty_cols)
|
df_sub_mh_grp = aggregate_df(contractorBill.df_mh, ["Location", "MH_NO"], qty_cols)
|
||||||
df_sub_dc_grp = aggregate_df(contractorBill.df_dc, ["Location", "Node_No"], mh_dc_qty_cols)
|
df_sub_dc_grp = aggregate_df(contractorBill.df_dc, ["Location", "Node_No"], mh_dc_qty_cols)
|
||||||
|
|
||||||
<<<<<<< HEAD
|
# --- FINAL MERGE LOGIC ---
|
||||||
# Merge and Calculate Difference
|
# We check if "Location" exists in the client data. If not, we add it to prevent the KeyError.
|
||||||
=======
|
for df_client in [clientBill.df_tr, clientBill.df_mh, clientBill.df_dc]:
|
||||||
>>>>>>> 2636f2e (added client RA bill wise download report)
|
if not df_client.empty and "Location" not in df_client.columns:
|
||||||
|
df_client["Location"] = "Unknown"
|
||||||
|
|
||||||
|
try:
|
||||||
df_tr_cmp = clientBill.df_tr.merge(df_sub_tr_grp, on=["Location", "MH_NO"], how="left", suffixes=("_Client", "_Sub"))
|
df_tr_cmp = clientBill.df_tr.merge(df_sub_tr_grp, on=["Location", "MH_NO"], how="left", suffixes=("_Client", "_Sub"))
|
||||||
df_mh_cmp = clientBill.df_mh.merge(df_sub_mh_grp, on=["Location", "MH_NO"], how="left", suffixes=("_Client", "_Sub"))
|
df_mh_cmp = clientBill.df_mh.merge(df_sub_mh_grp, on=["Location", "MH_NO"], how="left", suffixes=("_Client", "_Sub"))
|
||||||
df_dc_cmp = clientBill.df_dc.merge(df_sub_dc_grp, on=["Location", "Node_No"], how="left", suffixes=("_Client", "_Sub"))
|
df_dc_cmp = clientBill.df_dc.merge(df_sub_dc_grp, on=["Location", "Node_No"], how="left", suffixes=("_Client", "_Sub"))
|
||||||
|
except KeyError as e:
|
||||||
|
flash(f"Merge Error: Missing column {str(e)}. Check if 'Location' is defined in your database models.", "danger")
|
||||||
|
return render_template("generate_comparison_client_vs_subcont.html", tables=tables, ra_val=ra_val)
|
||||||
|
|
||||||
<<<<<<< HEAD
|
# ... (Rest of your calculation and download logic) ...
|
||||||
# Calculate Diffs
|
|
||||||
=======
|
|
||||||
>>>>>>> 2636f2e (added client RA bill wise download report)
|
|
||||||
for df in [df_tr_cmp, df_mh_cmp]:
|
|
||||||
for col in qty_cols:
|
|
||||||
if f"{col}_Client" in df.columns:
|
|
||||||
df[f"{col}_Diff"] = df[f"{col}_Client"].fillna(0) - df[f"{col}_Sub"].fillna(0)
|
|
||||||
|
|
||||||
for col in mh_dc_qty_cols:
|
# Convert to HTML for preview
|
||||||
if f"{col}_Client" in df_dc_cmp.columns:
|
|
||||||
df_dc_cmp[f"{col}_Diff"] = df_dc_cmp[f"{col}_Client"].fillna(0) - df_dc_cmp[f"{col}_Sub"].fillna(0)
|
|
||||||
|
|
||||||
<<<<<<< HEAD
|
|
||||||
output = io.BytesIO()
|
|
||||||
file_name = f"Comparison_RA_Bill_{RA_Bill_No}.xlsx"
|
|
||||||
with pd.ExcelWriter(output, engine="xlsxwriter") as writer:
|
|
||||||
df_tr_cmp.to_excel(writer, sheet_name="Tr.Ex Comparison", index=False)
|
|
||||||
df_mh_cmp.to_excel(writer, sheet_name="Mh.Ex Comparison", index=False)
|
|
||||||
df_dc_cmp.to_excel(writer, sheet_name="MH & DC Comparison", index=False)
|
|
||||||
|
|
||||||
output.seek(0)
|
|
||||||
return send_file(output, download_name=file_name, as_attachment=True, mimetype="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")
|
|
||||||
|
|
||||||
return render_template("generate_comparison_client_vs_subcont.html")
|
|
||||||
=======
|
|
||||||
# ACTION LOGIC: Check if user clicked "Download" or just "Preview/Generate"
|
|
||||||
if request.form.get("action") == "download":
|
|
||||||
output = io.BytesIO()
|
|
||||||
file_name = f"Comparison_RA_Bill_{RA_Bill_No}.xlsx"
|
|
||||||
with pd.ExcelWriter(output, engine="xlsxwriter") as writer:
|
|
||||||
df_tr_cmp.to_excel(writer, sheet_name="Tr.Ex Comparison", index=False)
|
|
||||||
df_mh_cmp.to_excel(writer, sheet_name="Mh.Ex Comparison", index=False)
|
|
||||||
df_dc_cmp.to_excel(writer, sheet_name="MH & DC Comparison", index=False)
|
|
||||||
output.seek(0)
|
|
||||||
return send_file(output, download_name=file_name, as_attachment=True, mimetype="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")
|
|
||||||
|
|
||||||
# If not downloading, convert DFs to HTML for the UI
|
|
||||||
tables["tr"] = df_tr_cmp.to_html(classes='table table-striped table-hover table-sm', index=False)
|
tables["tr"] = df_tr_cmp.to_html(classes='table table-striped table-hover table-sm', index=False)
|
||||||
tables["mh"] = df_mh_cmp.to_html(classes='table table-striped table-hover table-sm', index=False)
|
tables["mh"] = df_mh_cmp.to_html(classes='table table-striped table-hover table-sm', index=False)
|
||||||
tables["dc"] = df_dc_cmp.to_html(classes='table table-striped table-hover table-sm', index=False)
|
tables["dc"] = df_dc_cmp.to_html(classes='table table-striped table-hover table-sm', index=False)
|
||||||
|
|
||||||
return render_template("generate_comparison_client_vs_subcont.html", tables=tables, ra_val=ra_val)
|
return render_template("generate_comparison_client_vs_subcont.html", tables=tables, ra_val=ra_val)
|
||||||
>>>>>>> 2636f2e (added client RA bill wise download report)
|
|
||||||
|
|||||||
@@ -16,19 +16,21 @@
|
|||||||
{% endwith %}
|
{% endwith %}
|
||||||
|
|
||||||
<div class="card p-4 shadow-sm mx-auto" style="max-width: 600px;">
|
<div class="card p-4 shadow-sm mx-auto" style="max-width: 600px;">
|
||||||
<form method="POST">
|
<form method="POST" id="reportForm">
|
||||||
<div class="mb-3">
|
<div class="mb-3">
|
||||||
<label class="form-label fw-semibold">Select Subcontractor</label>
|
<label class="form-label fw-semibold">Select Subcontractor</label>
|
||||||
<select name="subcontractor_id" class="form-select" required>
|
<select name="subcontractor_id" class="form-select" required>
|
||||||
<option value="">-- Select Subcontractor --</option>
|
<option value="">-- Select Subcontractor --</option>
|
||||||
{% for sc in subcontractors %}
|
{% for sc in subcontractors %}
|
||||||
<option value="{{ sc.id }}">{{ sc.subcontractor_name }}</option>
|
<option value="{{ sc.id }}" {% if selected_sc_id == sc.id|string %}selected{% endif %}>
|
||||||
|
{{ sc.subcontractor_name }}
|
||||||
|
</option>
|
||||||
{% endfor %}
|
{% endfor %}
|
||||||
</select>
|
</select>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<div class="form-check form-switch mb-3">
|
<div class="form-check form-switch mb-3">
|
||||||
<input class="form-check-input" type="checkbox" name="download_all" value="true" id="downloadAllSwitch" onchange="toggleRAInput(this)">
|
<input class="form-check-input" type="checkbox" name="download_all" value="true" id="downloadAllSwitch" onchange="toggleRAInput(this)" {% if download_all %}checked{% endif %}>
|
||||||
<label class="form-check-label fw-bold text-primary" for="downloadAllSwitch">
|
<label class="form-check-label fw-bold text-primary" for="downloadAllSwitch">
|
||||||
Download All RA Bills for this Contractor
|
Download All RA Bills for this Contractor
|
||||||
</label>
|
</label>
|
||||||
@@ -36,16 +38,76 @@
|
|||||||
|
|
||||||
<div class="mb-4" id="ra_bill_container">
|
<div class="mb-4" id="ra_bill_container">
|
||||||
<label class="form-label fw-semibold">RA Bill Number</label>
|
<label class="form-label fw-semibold">RA Bill Number</label>
|
||||||
<input type="text" name="ra_bill_no" id="ra_bill_input" class="form-control" placeholder="e.g. 01">
|
<input type="text" name="ra_bill_no" id="ra_bill_input" class="form-control" placeholder="e.g. 01" value="{{ ra_bill_no or '' }}">
|
||||||
<small class="text-muted">Required if "Download All" is off.</small>
|
<small class="text-muted">Required if "Download All" is off.</small>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<button type="submit" class="btn btn-primary w-100 py-2">
|
<div class="row g-2">
|
||||||
<i class="fas fa-download me-2"></i>Generate Excel Report
|
<div class="col-md-6">
|
||||||
|
<button type="submit" name="action" value="preview" class="btn btn-outline-primary w-100 py-2">
|
||||||
|
<i class="fas fa-eye me-2"></i>Preview Data
|
||||||
</button>
|
</button>
|
||||||
|
</div>
|
||||||
|
<div class="col-md-6">
|
||||||
|
<button type="submit" name="action" value="download" class="btn btn-primary w-100 py-2">
|
||||||
|
<i class="fas fa-download me-2"></i>Download Excel
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
</form>
|
</form>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
|
{% if tables %}
|
||||||
|
<div class="mt-5 mb-5">
|
||||||
|
<hr>
|
||||||
|
<h3 class="text-center mb-4">Report Preview</h3>
|
||||||
|
|
||||||
|
<div class="card shadow-sm">
|
||||||
|
<div class="card-header bg-light">
|
||||||
|
<ul class="nav nav-tabs card-header-tabs" id="reportTabs" role="tablist">
|
||||||
|
<li class="nav-item">
|
||||||
|
<button class="nav-link active" id="tr-tab" data-bs-toggle="tab" data-bs-target="#tr" type="button">Transport (Tr.Ex.)</button>
|
||||||
|
</li>
|
||||||
|
<li class="nav-item">
|
||||||
|
<button class="nav-link" id="mh-tab" data-bs-toggle="tab" data-bs-target="#mh" type="button">Manpower (MH.Ex.)</button>
|
||||||
|
</li>
|
||||||
|
<li class="nav-item">
|
||||||
|
<button class="nav-link" id="dc-tab" data-bs-toggle="tab" data-bs-target="#dc" type="button">MH & DC</button>
|
||||||
|
</li>
|
||||||
|
</ul>
|
||||||
</div>
|
</div>
|
||||||
|
<div class="card-body tab-content" id="reportTabsContent">
|
||||||
|
<div class="tab-pane fade show active" id="tr" role="tabpanel">
|
||||||
|
<div class="table-responsive">
|
||||||
|
{{ tables.tr | safe }}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<div class="tab-pane fade" id="mh" role="tabpanel">
|
||||||
|
<div class="table-responsive">
|
||||||
|
{{ tables.mh | safe }}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<div class="tab-pane fade" id="dc" role="tabpanel">
|
||||||
|
<div class="table-responsive">
|
||||||
|
{{ tables.dc | safe }}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
{% endif %}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<style>
|
||||||
|
/* Styling specifically for pandas output */
|
||||||
|
.table-responsive .table {
|
||||||
|
font-size: 0.9rem;
|
||||||
|
white-space: nowrap;
|
||||||
|
}
|
||||||
|
.table-responsive thead {
|
||||||
|
background-color: #f8f9fa;
|
||||||
|
}
|
||||||
|
</style>
|
||||||
|
|
||||||
<script>
|
<script>
|
||||||
function toggleRAInput(checkbox) {
|
function toggleRAInput(checkbox) {
|
||||||
@@ -60,5 +122,10 @@ function toggleRAInput(checkbox) {
|
|||||||
container.style.opacity = "1";
|
container.style.opacity = "1";
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Check status on load to handle form persistence
|
||||||
|
document.addEventListener("DOMContentLoaded", function() {
|
||||||
|
toggleRAInput(document.getElementById('downloadAllSwitch'));
|
||||||
|
});
|
||||||
</script>
|
</script>
|
||||||
{% endblock %}
|
{% endblock %}
|
||||||
Reference in New Issue
Block a user