# model/gst_release.py
from flask import request, jsonify
from model.ItemCRUD import ItemCRUD
from model.Utilities import ItemCRUDType

class GSTRelease:

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

    # ------------------- Add GST Release -------------------
    def AddGSTRelease(self, request):
        try:
            gst = ItemCRUD(itemType=ItemCRUDType.GSTRelease)

            data = {
                "PMC_No": request.form.get("PMC_No", "").strip(),
                "Invoice_No": request.form.get("Invoice_No", "").strip(),
                "Basic_Amount": float(request.form.get("Basic_Amount", 0) or 0),
                "Final_Amount": float(request.form.get("Final_Amount", 0) or 0),
                "Total_Amount": float(request.form.get("Total_Amount", 0) or 0),
                "UTR": request.form.get("UTR", "").strip(),
                "Contractor_ID": int(request.form.get("Contractor_ID", 0) or 0)
            }

            # Add GST Release
            gst.AddItem(
                request=request,
                data=data,
                storedprocfetch="CheckGSTReleaseExists",
                storedprocadd="AddGSTReleaseFromExcel"
            )
            
            # Check if addition was successful
            if gst.isSuccess:
                print(f"GST Release Added: {data}")
            else:
                print(f"Failed to add GST Release: {gst.resultMessage}")

            self.isSuccess = gst.isSuccess
            self.resultMessage = str(gst.resultMessage)

        except Exception as e:
            print("ERROR in AddGSTRelease:", e)
            self.isSuccess = False
            self.resultMessage = str(e)

        return jsonify({"success": self.isSuccess, "message": self.resultMessage})

    # ------------------- Edit GST Release -------------------
    def EditGSTRelease(self, request, gst_release_id):
        try:
            gst = ItemCRUD(itemType=ItemCRUDType.GSTRelease)

            data = {
                "PMC_No": request.form.get("PMC_No", "").strip(),
                "Invoice_No": request.form.get("Invoice_No", "").strip(),
                "Basic_Amount": float(request.form.get("Basic_Amount", 0) or 0),
                "Final_Amount": float(request.form.get("Final_Amount", 0) or 0),
                "Total_Amount": float(request.form.get("Total_Amount", 0) or 0),
                "UTR": request.form.get("UTR", "").strip()
            }

            gst.EditItem(
                request=request,
                childid=gst_release_id,
                data=data,
                storedprocupdate="UpdateGSTRelease"
            )

            self.isSuccess = gst.isSuccess
            self.resultMessage = str(gst.resultMessage)

        except Exception as e:
            print("ERROR in EditGSTRelease:", e)
            self.isSuccess = False
            self.resultMessage = str(e)

        return jsonify({"success": self.isSuccess, "message": self.resultMessage})

    # ------------------- Delete GST Release -------------------
    def DeleteGSTRelease(self, gst_release_id):
        try:
            gst = ItemCRUD(itemType=ItemCRUDType.GSTRelease)

            gst.DeleteItem(
                request=None,
                itemID=gst_release_id,
                storedprocDelete="DeleteGSTReleaseById"
            )

            self.isSuccess = gst.isSuccess
            self.resultMessage = str(gst.resultMessage)

        except Exception as e:
            print("ERROR in DeleteGSTRelease:", e)
            self.isSuccess = False
            self.resultMessage = str(e)

        return jsonify({"success": self.isSuccess, "message": self.resultMessage})

    # ------------------- Get All GST Releases -------------------
    def GetAllGSTReleases(self):
        try:
            gst = ItemCRUD(itemType=ItemCRUDType.GSTRelease)
            rows = gst.GetAllData(None, "GetAllGSTReleases")

            data = []
            for row in rows:
                data.append({
                    "gst_release_id": row[0],
                    "pmc_no": row[1],
                    "invoice_no": row[2],
                    "basic_amount": row[3],
                    "final_amount": row[4],
                    "total_amount": row[5],
                    "utr": row[6],
                    "contractor_id": row[7]
                })

            return data

        except Exception as e:
            print("ERROR in GetAllGSTReleases:", e)
            return []

    # ------------------- Get GST Release By ID -------------------
    def GetGSTReleaseByID(self, gst_release_id):
        try:
            gst = ItemCRUD(itemType=ItemCRUDType.GSTRelease)
            row = gst.GetDataByID(gst_release_id, "GetGSTReleaseById")

            if row:
                return {
                    "gst_release_id": row[0],
                    "pmc_no": row[1],
                    "invoice_no": row[2],
                    "basic_amount": row[3],
                    "final_amount": row[4],
                    "total_amount": row[5],
                    "utr": row[6],
                    "contractor_id": row[7]
                }

            return None

        except Exception as e:
            print("ERROR in GetGSTReleaseByID:", e)
            return None