from model.ItemCRUD import ItemCRUD
from model.Utilities import ItemCRUDType

class District:

    def __init__(self):
        self.isSuccess = False
        self.resultMessage = ""

    # Add new district
    def AddDistrict(self, request): 
        district = ItemCRUD(ItemCRUDType.District)

        district_name = request.form.get('district_Name', '').strip()
        state_id = request.form.get('state_Id', '').strip()

        if not district_name or not state_id:
            self.isSuccess = False
            self.resultMessage = "Please enter district name and select a state."
            return

        district.AddItem(
            request=request,
            parentid=state_id,
            childname=district_name,
            storedprocfetch="GetDistrictByNameAndState",
            storedprocadd="SaveDistrict"
        )
        self.isSuccess = district.isSuccess
        self.resultMessage = district.resultMessage

    # Edit existing district
    def EditDistrict(self, request, district_id):
        district = ItemCRUD(ItemCRUDType.District)

        district_name = request.form.get('district_Name', '').strip()
        state_id = request.form.get('state_Id', '').strip()

        if not district_name or not state_id:
            self.isSuccess = False
            self.resultMessage = "Please enter district name and select a state."
            return

        district.EditItem(
            request=request,
            childid=district_id,
            parentid=state_id,
            childname=district_name,
            storedprocupdate="UpdateDistrict"
        )
        self.isSuccess = district.isSuccess
        self.resultMessage = district.resultMessage

    # Get all districts
    def GetAllDistricts(self, request):
        district = ItemCRUD(ItemCRUDType.District)
        districtsdata = district.GetAllData(request=request, storedproc="GetAllDistricts")
        self.isSuccess = district.isSuccess
        self.resultMessage = district.resultMessage
        return districtsdata

    # Check district exists (used for AJAX, optional)
    def CheckDistrict(self, request):
        district = ItemCRUD(ItemCRUDType.District)

        if request.is_json:
            district_name = request.json.get('district_Name', '').strip()
            state_id = request.json.get('state_Id', '').strip()
        else:
            district_name = request.form.get('district_Name', '').strip()
            state_id = request.form.get('state_Id', '').strip()

        result = district.CheckItem(
            request=request,
            parentid=state_id,
            childname=district_name,
            storedprocfetch="GetDistrictByNameAndState"
        )
        self.isSuccess = district.isSuccess
        self.resultMessage = district.resultMessage
        return result

    # Get district by ID
    def GetDistrictByID(self, request, district_id):
        district = ItemCRUD(ItemCRUDType.District)
        districtdata = district.GetDataByID(
            id=district_id,
            storedproc="GetDistrictDataByID"
        )
        if districtdata:
            self.isSuccess = True
        else:
            self.isSuccess = False
            self.resultMessage = "District not found"
        return districtdata

    # Delete district
    def DeleteDistrict(self, request, district_id):
        district = ItemCRUD(ItemCRUDType.District)
        district.DeleteItem(
            request=request,
            itemID=district_id,
            storedprocDelete="DeleteDistrict"
        )
        self.isSuccess = district.isSuccess
        self.resultMessage = str(district.resultMessage)