client models changes filds are requ.

This commit is contained in:
2025-12-20 16:35:35 +05:30
parent 47df8a2668
commit a7de81be73
9 changed files with 262 additions and 127 deletions

View File

@@ -11,7 +11,7 @@ class ManholeDomesticChamber(db.Model):
subcontractor = db.relationship("Subcontractor", backref="manhole_domestic_chamber_records") subcontractor = db.relationship("Subcontractor", backref="manhole_domestic_chamber_records")
# Basic Fields # Basic Fields
Location = db.Column(db.String(255)) Location = db.Column(db.String(500))
Node_No = db.Column(db.String(100)) Node_No = db.Column(db.String(100))
Depth_of_MH = db.Column(db.Float) Depth_of_MH = db.Column(db.Float)

View File

@@ -11,7 +11,7 @@ class ManholeExcavation(db.Model):
subcontractor = db.relationship("Subcontractor", backref="manhole_records") subcontractor = db.relationship("Subcontractor", backref="manhole_records")
# Basic Fields # Basic Fields
Location = db.Column(db.String(255)) Location = db.Column(db.String(500))
MH_NO = db.Column(db.String(100)) MH_NO = db.Column(db.String(100))
Upto_IL_Depth = db.Column(db.Float) Upto_IL_Depth = db.Column(db.Float)

View File

@@ -6,12 +6,13 @@ class ManholeDomesticChamberClient(db.Model):
id = db.Column(db.Integer, primary_key=True) id = db.Column(db.Integer, primary_key=True)
# Foreign Key to Subcontractor table # Foreign Key to Subcontractor table
subcontractor_id = db.Column(db.Integer, db.ForeignKey("subcontractors.id"), nullable=False) # subcontractor_id = db.Column(db.Integer, db.ForeignKey("subcontractors.id"), nullable=False)
# Relationship for easy access (subcontractor.subcontractor_name) # Relationship for easy access (subcontractor.subcontractor_name)
subcontractor = db.relationship("Subcontractor", backref="mh_dc_records") # subcontractor = db.relationship("Subcontractor", backref="mh_dc_records")
# Basic Fields # Basic Fields
Location = db.Column(db.String(255)) RA_Bill_No=db.Column(db.String(500))
Location = db.Column(db.String(500))
Node_No = db.Column(db.String(100)) Node_No = db.Column(db.String(100))
MH_TOP_LEVEL = db.Column(db.Float) MH_TOP_LEVEL = db.Column(db.Float)
MH_IL_LEVEL = db.Column(db.Float) MH_IL_LEVEL = db.Column(db.Float)

View File

@@ -6,12 +6,13 @@ class ManholeExcavationClient(db.Model):
id = db.Column(db.Integer, primary_key=True) id = db.Column(db.Integer, primary_key=True)
# Foreign Key to Subcontractor table # Foreign Key to Subcontractor table
subcontractor_id = db.Column(db.Integer, db.ForeignKey("subcontractors.id"), nullable=False) # subcontractor_id = db.Column(db.Integer, db.ForeignKey("subcontractors.id"), nullable=False)
# Relationship for easy access (subcontractor.subcontractor_name) # Relationship for easy access (subcontractor.subcontractor_name)
subcontractor = db.relationship("Subcontractor", backref="mh_ex_records") # subcontractor = db.relationship("Subcontractor", backref="mh_ex_records")
# Basic Fields # Basic Fields
Location = db.Column(db.String(255)) RA_Bill_No=db.Column(db.String(500))
Location = db.Column(db.String(500))
MH_NO = db.Column(db.String(100)) MH_NO = db.Column(db.String(100))
GL_m = db.Column(db.Float) GL_m = db.Column(db.Float)
MH_Invert_Level = db.Column(db.Float) MH_Invert_Level = db.Column(db.Float)

View File

@@ -6,12 +6,13 @@ class TrenchExcavationClient(db.Model):
id = db.Column(db.Integer, primary_key=True) id = db.Column(db.Integer, primary_key=True)
# Foreign Key to Subcontractor table # Foreign Key to Subcontractor table
subcontractor_id = db.Column(db.Integer, db.ForeignKey("subcontractors.id"), nullable=False) # subcontractor_id = db.Column(db.Integer, db.ForeignKey("subcontractors.id"), nullable=False)
# Relationship for easy access (subcontractor.subcontractor_name) # Relationship for easy access (subcontractor.subcontractor_name)
subcontractor = db.relationship("Subcontractor", backref="tr_ex_records") # subcontractor = db.relationship("Subcontractor", backref="tr_ex_records")
# Basic Fields # Basic Fields
Location = db.Column(db.String(255)) RA_Bill_No=db.Column(db.String(500))
Location = db.Column(db.String(500))
MH_NO = db.Column(db.String(100)) MH_NO = db.Column(db.String(100))
CC_length = db.Column(db.Float) CC_length = db.Column(db.Float)
Actual_Trench_Length = db.Column(db.Float) Actual_Trench_Length = db.Column(db.Float)

View File

@@ -11,7 +11,7 @@ class TrenchExcavation(db.Model):
subcontractor = db.relationship("Subcontractor", backref="trench_records") subcontractor = db.relationship("Subcontractor", backref="trench_records")
# Basic Fields # Basic Fields
Location = db.Column(db.String(255)) Location = db.Column(db.String(500))
MH_NO = db.Column(db.String(100)) MH_NO = db.Column(db.String(100))
CC_length = db.Column(db.Float) CC_length = db.Column(db.Float)
Invert_Level = db.Column(db.Float) Invert_Level = db.Column(db.Float)

View File

@@ -31,11 +31,12 @@ def client_import_file():
if request.method == "POST": if request.method == "POST":
file = request.files.get("file") file = request.files.get("file")
subcontractor_id = request.form.get("subcontractor_id") # subcontractor_id = request.form.get("subcontractor_id")
file_type = request.form.get("file_type") # file_type = request.form.get("file_type")
RA_Bill_No = request.form.get("RA_Bill_No")
service = FileService() service = FileService()
success, msg = service.handle_file_upload(file, subcontractor_id, file_type) success, msg = service.handle_client_file_upload(file, RA_Bill_No)
flash(msg, "success" if success else "danger") flash(msg, "success" if success else "danger")

View File

@@ -287,139 +287,268 @@ class FileService:
# ---------------------- client ---------------------------------- # ---------------------- client ----------------------------------
def client_trench_excavation(self, df, subcontractor_id): # def handle_client_file_upload(self, file, RA_Bill_No):
df.columns = [str(c).strip() for c in df.columns]
# If the sheet has merged cells -> forward fill Location
if "Location" in df.columns:
df["Location"] = df["Location"].ffill()
df = df.dropna(how="all") # REMOVE empty rows # if not RA_Bill_No:
# Identify missing location rows before insert # return False, "Please Enter RA Bill No ."
missing_loc = df[df["Location"].isna() | (df["Location"].astype(str).str.strip() == "")]
if not missing_loc.empty:
return False, f"Error: Some rows have empty Location. Rows: {missing_loc.index.tolist()}"
saved_count = 0 # if not file or file.filename == "":
# return False, "No file selected."
# if not self.allowed_file(file.filename):
# return False, "Invalid file type! Allowed: CSV, XLSX, XLS"
try: # ensure_upload_folder()
for index, row in df.iterrows():
record_data = {}
# Insert only fields that exist in model
for col in df.columns:
if hasattr(TrenchExcavationClient, col):
value = row[col]
# Normalize empty values
if pd.isna(value) or str(value).strip() in ["", "-", "", "nan", "NaN"]:
value = None
record_data[col] = value # folder = os.path.join(Config.UPLOAD_FOLDER, f"Client_Bill_{RA_Bill_No}")
# os.makedirs(folder, exist_ok=True)
record = TrenchExcavationClient( # filename = secure_filename(file.filename)
subcontractor_id=subcontractor_id, # filepath = os.path.join(folder, filename)
**record_data # file.save(filepath)
)
db.session.add(record) # try:
saved_count += 1 # df_tr_ex = pd.read_excel(filepath, sheet_name ="Tr.Ex.", header=12)
# df_mh_ex = pd.read_excel(filepath, sheet_name="MH Ex.", header=12)
# df_mh_dc = pd.read_excel(filepath, sheet_name="MH & DC", header=11)
db.session.commit() # print("\n=== Uploaded File tr ex ===")
return True, f"Clinnt Tr Ex data saved successfully. Total rows: {saved_count}" # print(df_tr_ex.head())
# print("=============================\n")
except Exception as e: # print("\n=== Uploaded File mh ex ===")
db.session.rollback() # print(df_mh_ex.head())
return False, f"Clinnt Tr Ex Save Failed: {e}" # print("========================================\n")
# print("\n=== Uploaded File MH DC ===")
# print(df_mh_dc.head())
# print("========================================\n")
# Mh Ex save method (ManholeExcavationClient model) # self.client_trench_excavation(df_tr_ex, RA_Bill_No)
def client_manhole_excavation(self, df, subcontractor_id): # self.client_manhole_excavation(df_mh_ex, RA_Bill_No)
# self.client_manhole_domestic_chamber(df_mh_dc, RA_Bill_No)
# return True, "File uploaded successfully."
# except Exception as e:
# return False, f"Processing failed: {e}"
# # Tr Ex save method (TrenchExcavationClient model)
# def client_trench_excavation(self, df, RA_Bill_No):
# df.columns = [str(c).strip() for c in df.columns]
# # If the sheet has merged cells -> forward fill Location
# if "Location" in df.columns:
# df["Location"] = df["Location"].ffill()
# df = df.dropna(how="all") # REMOVE empty rows
# # Identify missing location rows before insert
# missing_loc = df[df["Location"].isna() | (df["Location"].astype(str).str.strip() == "")]
# if not missing_loc.empty:
# return False, f"Error: Some rows have empty Location. Rows: {missing_loc.index.tolist()}"
# saved_count = 0
# try:
# for index, row in df.iterrows():
# record_data = {}
# # Insert only fields that exist in model
# for col in df.columns:
# if hasattr(TrenchExcavationClient, col):
# value = row[col]
# # Normalize empty values
# if pd.isna(value) or str(value).strip() in ["", "-", "—", "nan", "NaN"]:
# value = None
# record_data[col] = value
# record = TrenchExcavationClient(
# RA_Bill_No=RA_Bill_No,
# **record_data
# )
# db.session.add(record)
# saved_count += 1
# db.session.commit()
# return True, f"Clinnt Tr Ex data saved successfully. Total rows: {saved_count}"
# except Exception as e:
# db.session.rollback()
# return False, f"Clinnt Tr Ex Save Failed: {e}"
# # Mh Ex save method (ManholeExcavationClient model)
# def client_manhole_excavation(self, df, RA_Bill_No):
# # Clean column names (strip whitespace)
# df.columns = [str(c).strip() for c in df.columns]
# # If the sheet has merged cells -> forward fill Location
# if "Location" in df.columns:
# df["Location"] = df["Location"].ffill()
# # REMOVE empty rows
# df = df.dropna(how="all")
# # Identify missing location rows before insert
# missing_loc = df[df["Location"].isna() | (df["Location"].astype(str).str.strip() == "")]
# if not missing_loc.empty:
# return False, f"Error: Some rows have empty Location. Rows: {missing_loc.index.tolist()}"
# saved_count = 0
# try:
# for index, row in df.iterrows():
# record_data = {}
# # Insert only fields that exist in model
# for col in df.columns:
# if hasattr(ManholeExcavationClient, col):
# value = row[col]
# # Normalize empty values
# if pd.isna(value) or str(value).strip() in ["", "-", "—", "nan", "NaN"]:
# value = None
# record_data[col] = value
# record = ManholeExcavationClient(
# RA_Bill_No=RA_Bill_No,
# **record_data
# )
# db.session.add(record)
# saved_count += 1
# db.session.commit()
# return True, f" Client Mh Ex. data saved successfully. Total rows: {saved_count}"
# except Exception as e:
# db.session.rollback()
# return False, f"Client Mh Ex. Save Failed: {e}"
# # Mh and Dc save method (ManholeDomesticChamberClient model)
# def client_manhole_domestic_chamber(self, df, RA_Bill_No):
# Clean column names (strip whitespace) # Clean column names (strip whitespace)
# df.columns = [str(c).strip() for c in df.columns]
# # If the sheet has merged cells -> forward fill Location
# if "Location" in df.columns:
# df["Location"] = df["Location"].ffill()
# # REMOVE empty rows
# df = df.dropna(how="all")
# # Identify missing location rows before insert
# missing_loc = df[df["Location"].isna() | (df["Location"].astype(str).str.strip() == "")]
# if not missing_loc.empty:
# return False, f"Error: Some rows have empty Location. Rows: {missing_loc.index.tolist()}"
# saved_count = 0
# try:
# for index, row in df.iterrows():
# record_data = {}
# # Insert only fields that exist in model
# for col in df.columns:
# if hasattr(ManholeDomesticChamberClient, col):
# value = row[col]
# # Normalize empty values
# if pd.isna(value) or str(value).strip() in ["", "-", "—", "nan", "NaN"]:
# value = None
# record_data[col] = value
# record = ManholeDomesticChamberClient(
# RA_Bill_No=RA_Bill_No,
# **record_data
# )
# db.session.add(record)
# saved_count += 1
# db.session.commit()
# return True, f"Mh and Dc data saved successfully. Total rows: {saved_count}"
# except Exception as e:
# db.session.rollback()
# return False, f"Mh and Dc data Save Failed: {e}"
def handle_client_file_upload(self, file, RA_Bill_No):
if not RA_Bill_No:
return False, "Please Enter RA Bill No."
if not file or file.filename == "":
return False, "No file selected."
if not self.allowed_file(file.filename):
return False, "Invalid file type! Allowed: CSV, XLSX, XLS"
ensure_upload_folder()
folder = os.path.join(Config.UPLOAD_FOLDER, f"Client_Bill_{RA_Bill_No}")
os.makedirs(folder, exist_ok=True)
filename = secure_filename(file.filename)
filepath = os.path.join(folder, filename)
file.save(filepath)
try:
df_tr_ex = pd.read_excel(filepath, sheet_name="Tr.Ex.", header=12)
df_mh_ex = pd.read_excel(filepath, sheet_name="MH Ex.", header=12)
df_mh_dc = pd.read_excel(filepath, sheet_name="MH & DC", header=11)
# Single transaction
self._save_client_data(df_tr_ex, TrenchExcavationClient, "Trench Excavation", RA_Bill_No)
self._save_client_data(df_mh_ex, ManholeExcavationClient, "Manhole Excavation", RA_Bill_No)
self._save_client_data(df_mh_dc, ManholeDomesticChamberClient, "MH & DC", RA_Bill_No)
db.session.commit()
return True, "Client file uploaded and all data saved successfully."
except ClientDataSaveError as e:
db.session.rollback()
return False, str(e)
except Exception as e:
db.session.rollback()
return False, f"Unexpected system error: {e}"
def _save_client_data(self, df, model, RA_Bill_No):
# Clean column names
df.columns = [str(c).strip() for c in df.columns] df.columns = [str(c).strip() for c in df.columns]
# If the sheet has merged cells -> forward fill Location
# Forward fill Location (merged cells issue)
if "Location" in df.columns: if "Location" in df.columns:
df["Location"] = df["Location"].ffill() df["Location"] = df["Location"].ffill()
# REMOVE empty rows # Remove fully empty rows
df = df.dropna(how="all") df = df.dropna(how="all")
# Identify missing location rows before insert
missing_loc = df[df["Location"].isna() | (df["Location"].astype(str).str.strip() == "")] # Validate Location
if "Location" in df.columns:
missing_loc = df[
df["Location"].isna() |
(df["Location"].astype(str).str.strip() == "")
]
if not missing_loc.empty: if not missing_loc.empty:
return False, f"Error: Some rows have empty Location. Rows: {missing_loc.index.tolist()}" raise Exception(
f"{model.__name__}: Empty Location at rows {missing_loc.index.tolist()}"
)
saved_count = 0 # Insert rows
for _, row in df.iterrows():
try:
for index, row in df.iterrows():
record_data = {} record_data = {}
# Insert only fields that exist in model
for col in df.columns: for col in df.columns:
if hasattr(ManholeExcavationClient, col): if hasattr(model, col):
value = row[col] value = row[col]
# Normalize empty values
if pd.isna(value) or str(value).strip() in ["", "-", "", "nan", "NaN"]: if pd.isna(value) or str(value).strip() in ["", "-", "", "nan", "NaN"]:
value = None value = None
record_data[col] = value record_data[col] = value
record = ManholeExcavationClient( record = model(
subcontractor_id=subcontractor_id, RA_Bill_No=RA_Bill_No,
**record_data **record_data
) )
db.session.add(record) db.session.add(record)
saved_count += 1
db.session.commit()
return True, f" Client Mh Ex. data saved successfully. Total rows: {saved_count}"
except Exception as e:
db.session.rollback()
return False, f"Client Mh Ex. Save Failed: {e}"
# Mh and Dc save method (ManholeDomesticChamberClient model)
def client_manhole_domestic_chamber(self, df, subcontractor_id):
# Clean column names (strip whitespace)
df.columns = [str(c).strip() for c in df.columns]
# If the sheet has merged cells -> forward fill Location
if "Location" in df.columns:
df["Location"] = df["Location"].ffill()
# REMOVE empty rows
df = df.dropna(how="all")
# Identify missing location rows before insert
missing_loc = df[df["Location"].isna() | (df["Location"].astype(str).str.strip() == "")]
if not missing_loc.empty:
return False, f"Error: Some rows have empty Location. Rows: {missing_loc.index.tolist()}"
saved_count = 0
try:
for index, row in df.iterrows():
record_data = {}
# Insert only fields that exist in model
for col in df.columns:
if hasattr(ManholeDomesticChamberClient, col):
value = row[col]
# Normalize empty values
if pd.isna(value) or str(value).strip() in ["", "-", "", "nan", "NaN"]:
value = None
record_data[col] = value
record = ManholeDomesticChamberClient(
subcontractor_id=subcontractor_id,
**record_data
)
db.session.add(record)
saved_count += 1
db.session.commit()
return True, f"Mh and Dc data saved successfully. Total rows: {saved_count}"
except Exception as e:
db.session.rollback()
return False, f"Mh and Dc data Save Failed: {e}"

View File

@@ -8,17 +8,17 @@
<form method="POST" enctype="multipart/form-data"> <form method="POST" enctype="multipart/form-data">
<!-- 1. SELECT SUBCONTRACTOR --> <!-- 1. SELECT SUBCONTRACTOR -->
<label class="form-label">Select Subcontractor vs Client</label> <!-- <label class="form-label">Select Subcontractor vs Client</label>
<select name="subcontractor_id" id="subcontractor_id" class="form-select mb-3" required> <select name="subcontractor_id" id="subcontractor_id" class="form-select mb-3" 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 }}">{{ sc.subcontractor_name }}</option>
{% endfor %} {% endfor %}
</select> </select> -->
<!-- 2. FILE TYPE (MODEL NAME) --> <!-- 2. FILE TYPE (MODEL NAME) -->
<label class="form-label">Select File Type</label> <!-- <label class="form-label">Select File Type</label>
<select name="file_type" id="file_type" class="form-select mb-3" required> <select name="file_type" id="file_type" class="form-select mb-3" required>
<option value="">-- Select File Type --</option> <option value="">-- Select File Type --</option>
<option value="">Sheet</option> <option value="">Sheet</option>
@@ -26,8 +26,10 @@
<option value="mh_ex_client">Mh. Ex </option> <option value="mh_ex_client">Mh. Ex </option>
<option value="mh_dc_client">MH & DC </option> <option value="mh_dc_client">MH & DC </option>
<option value="">Laying Sheet</option> <option value="">Laying Sheet</option>
</select> </select> -->
<label class="form-label">RA Bill No</label>
<input type="text" name="RA_Bill_No" class="form-control mb-3" required>
<!-- 3. FILE UPLOAD --> <!-- 3. FILE UPLOAD -->
<label class="form-label">Choose File</label> <label class="form-label">Choose File</label>
<input type="file" name="file" class="form-control mb-3" required> <input type="file" name="file" class="form-control mb-3" required>