report code changes by pankaj

model/State.py

@@ -0,0 +1,168 @@
+from flask import request, redirect, url_for
+from flask_login import current_user
+
+from model.Utilities import RegEx, ResponseHandler, HtmlHelper, ItemCRUDType
+from model.Log import LogHelper
+from model.ItemCRUD import ItemCRUD  
+
+import config
+import re
+import mysql.connector
+
+
+class State:
+
+    def __init__(self):
+        self.isSuccess = False
+        self.resultMessage = ""
+
+    # ----------------------------------------------------------
+    # ADD STATE (USING ITEM CRUD)
+    # ----------------------------------------------------------
+    def AddState(self, request):
+
+        state_name = request.form['state_Name'].strip()
+
+        crud = ItemCRUD(ItemCRUDType.State)
+
+        crud.AddItem(
+            request=request,
+            childname=state_name,
+            storedprocfetch="CheckStateExists",
+            storedprocadd="SaveState"
+        )
+
+        self.isSuccess = crud.isSuccess
+        self.resultMessage = crud.resultMessage
+
+        return self.resultMessage
+
+
+    # ----------------------------------------------------------
+    # GET ALL STATES (NO CHANGE - THIS IS CORRECT)
+    # ----------------------------------------------------------
+    def GetAllStates(self, request):
+
+        connection = config.get_db_connection()
+        data = []
+
+        if not connection:
+            return []
+
+        cursor = connection.cursor()
+
+        try:
+            cursor.callproc("GetAllStates")
+            for res in cursor.stored_results():
+                data = res.fetchall()
+
+            self.isSuccess = True
+
+        except mysql.connector.Error as e:
+            print(f"Error fetching states: {e}")
+            self.isSuccess = False
+            self.resultMessage = HtmlHelper.json_response(
+                ResponseHandler.fetch_failure("state"), 500
+            )
+            return []
+
+        finally:
+            cursor.close()
+            connection.close()
+
+        return data
+
+
+    # ----------------------------------------------------------
+    # CHECK STATE (USING ITEM CRUD)
+    # ----------------------------------------------------------
+    def CheckState(self, request):
+
+        state_name = request.json.get('state_Name', '').strip()
+
+        crud = ItemCRUD(ItemCRUDType.State)
+
+        return crud.CheckItem(
+            request=request,
+            parentid=None,
+            childname=state_name,
+            storedprocfetch="CheckStateExists"
+        )
+
+
+    # ----------------------------------------------------------
+    # DELETE STATE (USING ITEM CRUD)
+    # ----------------------------------------------------------
+    def DeleteState(self, request, id):
+
+        crud = ItemCRUD(ItemCRUDType.State)
+
+        crud.DeleteItem(
+            request=request,
+            itemID=id,
+            storedprocDelete="DeleteState"
+        )
+
+        self.isSuccess = crud.isSuccess
+        self.resultMessage = crud.resultMessage
+
+        return self.resultMessage
+
+
+    # ----------------------------------------------------------
+    # EDIT STATE (USING ITEM CRUD)
+    # ----------------------------------------------------------
+    def EditState(self, request, id):
+
+        state_name = request.form['state_Name'].strip()
+
+        crud = ItemCRUD(ItemCRUDType.State)
+
+        crud.EditItem(
+            request=request,
+            childid=id,
+            parentid=None,
+            childname=state_name,
+            storedprocupdate="UpdateStateById"
+        )
+
+        self.isSuccess = crud.isSuccess
+        self.resultMessage = crud.resultMessage
+
+        return redirect(url_for('state.add_state'))
+
+
+    # ----------------------------------------------------------
+    # GET STATE BY ID (KEEP SAME)
+    # ----------------------------------------------------------
+    def GetStateByID(self, request, id):
+
+        connection = config.get_db_connection()
+        data = None
+
+        if not connection:
+            return None
+
+        cursor = connection.cursor()
+
+        try:
+            cursor.callproc("GetStateByID", (id,))
+            for res in cursor.stored_results():
+                data = res.fetchone()
+
+            if data:
+                self.isSuccess = True
+                self.resultMessage = "Success"
+            else:
+                self.isSuccess = False
+                self.resultMessage = "Not Found"
+
+        except mysql.connector.Error as e:
+            print(f"Error fetching state: {e}")
+            self.isSuccess = False
+
+        finally:
+            cursor.close()
+            connection.close()
+
+        return data