import csv

import datetime
from pathlib import Path
from PyQt5.QtWidgets import QMessageBox
from Logger import setup_logger
from Word import WordHandler
from constants import WIRLogFilePath
from load_combobox_items_1 import ComboBoxLoader

class SearchEditing:
def __init__(self, main_window, ui):
self.main_window = main_window
self.ui = ui
self.logger = setup_logger()
self.file_path = Path(WIRLogFilePath)
self.combo_loader = ComboBoxLoader()
self.combo_loader.load_activities(self.ui.Activity_QCB2)
self.combo_loader.load_data(self.ui.Villa_QCB2, 0)
self.combo_loader.load_dates(self.ui.Inspection_Date_QCB2, 27)

def Search_WIRLog(self):
wir_no_to_find = self.ui.WIRNo_QLE2.text().strip()
rev_to_find = self.ui.Rev_QCB2.currentText().strip()
self.ui.WIRNo_QLE2.setEnabled(False)
self.ui.Rev_QCB2.setEnabled(False)

if not wir_no_to_find:
self._show_dialog("Error", "Please enter WIR No to search.",
QMessageBox.Warning)
self.ui.WIRNo_QLE2.setEnabled(True)
self.ui.Rev_QCB2.setEnabled(True)
return

try:
wir_no_int = int(wir_no_to_find)
except ValueError:
self._show_dialog("Error", "Invalid WIR No. Please enter a valid
number.", QMessageBox.Warning)
self.ui.WIRNo_QLE2.setEnabled(True)
self.ui.Rev_QCB2.setEnabled(True)
return

# Initialize variables
latest_revision = None
found_record = None
revisions = [] # Store all revisions for the WIR No

try:
with open(self.file_path, 'r', newline='') as f:
reader = csv.DictReader(f)

for row in reader:

current_wir_no = row['WIR No'].strip()
current_rev = row['Revision'].strip()

if current_wir_no == wir_no_to_find:
revisions.append(int(current_rev)) # Store revision
numbers as integers
 # Determine the latest revision
if revisions:
latest_revision = str(max(revisions)) # Convert back to string for
UI

# If no revision was selected, notify the user and set the latest
revision
if not rev_to_find:
if latest_revision:
self._show_dialog(
"Information",
f"No revision selected. The latest revision for WIR No
{wir_no_to_find} is {latest_revision}.",
QMessageBox.Information
)
self.ui.Rev_QCB2.setCurrentText(latest_revision) # Auto-set
latest revision
self.ui.WIRNo_QLE2.setEnabled(True)
self.ui.Rev_QCB2.setEnabled(True)
return # **STOP HERE TO AVOID FALSE "NOT FOUND" MESSAGE**

# Perform the search again with the selected or updated revision

with open(self.file_path, 'r', newline='') as f:
reader = csv.DictReader(f)
for row in reader:
if row['WIR No'].strip() == wir_no_to_find and
row['Revision'].strip() == rev_to_find:
found_record = row
break # Stop once found

# Notify user if an older revision is selected

if latest_revision and rev_to_find < latest_revision:
self._show_dialog(
"Information",
f"You have selected an older revision. The latest revision for
WIR No {wir_no_to_find} is {latest_revision}.",
QMessageBox.Information
)

# Process found record

if found_record:
self.ui.groupBox_2.setEnabled(True)
self.ui.WIRNo_QLE2.setEnabled(False)
self.ui.Rev_QCB2.setEnabled(False)
self._populate_ui_fields(found_record)
else:
self._show_dialog(
"Information",
f"WIR No {wir_no_to_find} Rev {rev_to_find} not found.",
QMessageBox.Information
)
self.ui.groupBox_2.setEnabled(False)
self.ui.WIRNo_QLE2.setEnabled(True)
self.ui.Rev_QCB2.setEnabled(True)

except FileNotFoundError:
self._show_dialog("Error", "WIR Log file not found!",
QMessageBox.Critical)
 self.ui.WIRNo_QLE2.setEnabled(True)
self.ui.Rev_QCB2.setEnabled(True)

except Exception as e:
self._show_dialog("Error", f"An unexpected error occurred during
search: {str(e)}", QMessageBox.Critical)
self.ui.WIRNo_QLE2.setEnabled(True)
self.ui.Rev_QCB2.setEnabled(True)

def _populate_ui_fields(self, record):

"""Populates UI fields with the data from the found record."""
def set_combo_value(combo_box, value):
value = f" {value.strip()}" if value else ""
if value and combo_box.findText(value) == -1:
combo_box.addItem(value) # Add missing value
combo_box.setCurrentText(value) # Set value

# Set values for ComboBoxes

Villa_value = record.get('Villa No', '')
if Villa_value.startswith(("Plot #", "Villa #")):
Villa_value = Villa_value.split('#', 1)[-1]

set_combo_value(self.ui.Activity_QCB2, record.get('Activity', ''))

set_combo_value(self.ui.Sub_Activity_QCB2, record.get('Sub Activity', ''))
set_combo_value(self.ui.Villa_QCB2, Villa_value)
set_combo_value(self.ui.Level_QCB2, record.get('Level', ''))
set_combo_value(self.ui.Part_QCB2, record.get('Part', ''))

# Set text in QTextEdit

self.ui.Description_QTE2.setPlainText(record.get('Description', ''))

# Handle Date - Ensure correct format

inspection_date_str_csv = record.get('Inspection Date', '')
print(inspection_date_str_csv)
if inspection_date_str_csv:
try:
# Corrected format string to parse 2-digit year '%y'
inspection_date_obj =
datetime.datetime.strptime(inspection_date_str_csv.strip(), '%d %b %Y')
formatted_date = inspection_date_obj.strftime('%d %b %Y') # Keep
UI format as '%Y' (4-digit)
set_combo_value(self.ui.Inspection_Date_QCB2, formatted_date)
except ValueError as e:
print(f"Error parsing date: {e}") # Debugging
self.ui.Inspection_Date_QCB2.setCurrentText('')

def Save_WIRLog(self):
wir_no_to_update = self.ui.WIRNo_QLE2.text().strip()
rev_to_update = self.ui.Rev_QCB2.currentText().strip()

# Validate required fields

if not wir_no_to_update:
self._show_dialog("Error", "WIR No. is missing. Search for a WIR to
edit first.", QMessageBox.Warning)
return

activity_index = self.ui.Activity_QCB2.currentIndex()
villa_text = self.ui.Villa_QCB2.currentText().strip()
 # Collect data from controls
updated_record = {
'WIR No': wir_no_to_update,
'Revision': rev_to_update,
'Activity': self.ui.Activity_QCB2.currentText().strip(),
'Sub Activity': self.ui.Sub_Activity_QCB2.currentText().strip(),
'Description': self.ui.Description_QTE2.toPlainText().strip(),
'Villa': f"Plot # {villa_text}" if activity_index < 5 else f"Villa #
{villa_text}",
'Level': self.ui.Level_QCB2.currentText().strip(),
'Part': self.ui.Part_QCB2.currentText().strip(),
'Created Date':datetime.date.today().strftime('%d %b %Y'),
'Inspection Date': self.ui.Inspection_Date_QCB2.currentText().strip(),
'Status': "Under Review",
}
print(f" villa # {'Villa'}")
# Remove empty fields
updated_record = {k: v for k, v in updated_record.items() if v}

# Read the existing file to get field names

try:
with open(self.file_path, 'r', newline='') as readFile:
reader = csv.DictReader(readFile)
fieldnames = reader.fieldnames # Original fieldnames from file

if not fieldnames:
self._show_dialog("Error", "CSV file is missing headers!",
QMessageBox.Critical)
return

# Ensure updated_record only contains valid field names

updated_record = {k: v for k, v in updated_record.items() if k in
fieldnames}

if len(updated_record) < 3: # Ensure at least WIR No, Rev, and

some data exist
self._show_dialog("Error", "No valid data to update the WIR
Log.", QMessageBox.Warning)
return

all_records = []
record_updated = False

for row in reader:

current_wir_no = row['WIR No'].strip()
current_rev = row['Revision'].strip()
if current_wir_no == wir_no_to_update and current_rev ==
rev_to_update:
all_records.append(updated_record) # Replace with updated
record
record_updated = True
else:
all_records.append(row) # Keep existing record

if record_updated:
with open(self.file_path, 'w', newline='') as writeFile:
writer = csv.DictWriter(writeFile, fieldnames=fieldnames)
writer.writeheader()
 writer.writerows(all_records)
self._show_dialog("Success", f"WIR No {wir_no_to_update} Rev
{rev_to_update} updated successfully!",
QMessageBox.Information)

word_handler = WordHandler(wir_no_to_update, rev_to_update)

word_handler.update_bookmarks(updated_record)

self.ui.groupBox_2.setEnabled(False)
self.ui.WIRNo_QLE2.setEnabled(True)
self.ui.Rev_QCB2.setEnabled(True)
self._clear_ui_fields()
else:
self._show_dialog("Error", f"WIR No {wir_no_to_update} Rev
{rev_to_update} not found for update!",
QMessageBox.Warning)

except FileNotFoundError:
self._show_dialog("Error", "WIR Log file not found!",
QMessageBox.Critical)
except Exception as e:
self._show_dialog("Error", f"An unexpected error occurred during save:
{str(e)}", QMessageBox.Critical)

def Cancel_SearchWIR(self):
"""Resets the Search WIR tab to its initial state."""
self.ui.groupBox_2.setEnabled(False) # Disable the group box
self.ui.WIRNo_QLE2.setEnabled(True) # Re-enable WIR No input
self.ui.Rev_QCB2.setEnabled(True) # Re-enable Revision input

self._clear_ui_fields() # Clear populated UI fields

def _clear_ui_fields(self):
"""Clears the UI fields in Tab 3."""
self.ui.WIRNo_QLE2.clear() # Clear WIR No input field (optional)
self.ui.Rev_QCB2.setCurrentIndex(0) # Reset Revision Combobox to default
(index 0, which is "") (optional)
self.ui.Activity_QCB2.setCurrentIndex(-1) # or setCurrentText("") to clear
without triggering currentIndexChanged signal if needed
self.ui.Sub_Activity_QCB2.setCurrentIndex(-1)
self.ui.Description_QTE2.clear()
self.ui.Villa_QCB2.setCurrentIndex(-1)
self.ui.Level_QCB2.setCurrentIndex(-1)
self.ui.Part_QCB2.setCurrentIndex(-1)
self.ui.Inspection_Date_QCB2.setCurrentIndex(-1)

def _show_dialog(self, title, message, icon=QMessageBox.Warning):

msg = QMessageBox(self.main_window)
msg.setIcon(icon)
msg.setText(message)
msg.setWindowTitle(title)
msg.exec_()