from AppCode.Config import DBConfig
import mysql.connector
import pandas as pd
import io


class CITHandler:

    def __init__(self):
        db = DBConfig()
        self.conn = db.get_db_connection()
        self.cursor = self.conn.cursor(dictionary=True)

    # GET ALL CIT RECORDS
    def get_all_cit(self):
        self.cursor.callproc("GetAllCIT")
        records = []

        for result in self.cursor.stored_results():
            records = result.fetchall()

        return records

    # GET CIT BY ID
    def get_cit_by_id(self, id):
        self.cursor.callproc("GetCITById", [id])
        records = []

        for result in self.cursor.stored_results():
            records = result.fetchall()

        if records:
            return records[0]

        return None

    # INSERT CIT RECORD
    def add_cit(self, data):
        columns = [
            'year', 'gross_total_income', 'disallowance_14a', 'disallowance_37',
            'deduction_80ia_business', 'deduction_80ia_misc', 'deduction_80ia_other',
            'deduction_sec37_disallowance', 'deduction_80g', 'net_taxable_income',
            'tax_30_percent', 'tax_book_profit_18_5', 'tax_payable', 'surcharge_12',
            'edu_cess_3', 'total_tax_payable', 'mat_credit', 'interest_234c',
            'total_tax', 'advance_tax', 'tds', 'tcs','sat', 'tax_on_assessment', 'refund', 'Remarks'
        ]

        values = [data.get(col, 0) for col in columns]

        self.cursor.callproc("InsertCIT", values)
        self.conn.commit()

    # UPDATE CIT RECORD
    def update_cit(self, id, data):
        columns = [
            'year', 'gross_total_income', 'disallowance_14a', 'disallowance_37',
            'deduction_80ia_business', 'deduction_80ia_misc', 'deduction_80ia_other',
            'deduction_sec37_disallowance', 'deduction_80g', 'net_taxable_income',
            'tax_30_percent', 'tax_book_profit_18_5', 'tax_payable', 'surcharge_12',
            'edu_cess_3', 'total_tax_payable', 'mat_credit', 'interest_234c',
            'total_tax', 'advance_tax', 'tds', 'tcs','sat', 'tax_on_assessment', 'refund', 'Remarks'
        ]

        values = [id] + [data.get(col, 0) for col in columns]

        self.cursor.callproc("UpdateCITById", values)
        self.conn.commit()

    # DELETE CIT RECORD
    def delete_cit(self, id):
        self.cursor.callproc("DeleteCITById", [id])
        self.conn.commit()
    # 
    def cit_report_download(self, selected_year):
        try:
            # AY calculation (2020 -> AY 2020-2021)
            ay_start = int(selected_year)
            ay_end = ay_start + 1
            assessment_year = f"AY {ay_start}-{ay_end}"

            # Fetch CIT records
            self.cursor.callproc("GetCITByYear", [selected_year])

            rows = []
            for result in self.cursor.stored_results():
                rows = result.fetchall()

            if not rows:
                return None

            # Remove id column if exists
            for row in rows:
                row.pop("id", None)

            # Mapping DB fields → readable Excel labels
            field_mapping = {
                "gross_total_income": "Gross Total Income",
                "disallowance_14a": "Add: Disallowance u/s 14A",
                "disallowance_37": "Add: Disallowance u/s 37",
                "deduction_80ia_business": "Less: Deduction u/s 80IA - On Business Income",
                "deduction_80ia_misc": "On Misc Receipts",
                "deduction_80ia_other": "On Other",
                "deduction_sec37_disallowance": "On Sec 37 Disallowance",
                "deduction_80g": "Less: Deduction u/s 80G",
                "net_taxable_income": "Net Taxable Income",
                "tax_30_percent": "Tax @ 30%",
                "tax_book_profit_18_5": "Tax @ 18.5% on Book Profit",
                "tax_payable": "Tax Payable",
                "surcharge_12": "Surcharge @ 12%",
                "edu_cess_3": "Education Cess @ 3%",
                "total_tax_payable": "Total Tax Payable",
                "mat_credit": "Less: MAT Credit Utilized",
                "interest_234c": "Add: Interest u/s 234C",
                "total_tax": "Total Tax",
                "advance_tax": "Advance Tax",
                "tds": "TDS",
                "tcs": "TCS",
                "sat": "SAT",
                "tax_on_assessment": "Tax on Regular Assessment",
                "refund": "Refund",
                "Remarks": "Remarks"
            }

            # Vertical CIT structure (single record per year)
            data = []
            for key, label in field_mapping.items():
                value = rows[0].get(key, 0)
                data.append([label, value])

            df = pd.DataFrame(data, columns=["Particulars", "CIT"])

            # Excel output
            output = io.BytesIO()
            with pd.ExcelWriter(output, engine="xlsxwriter") as writer:
                workbook = writer.book
                worksheet = workbook.add_worksheet("CIT Report")
                writer.sheets["CIT Report"] = worksheet

                # Formats
                title_fmt = workbook.add_format({
                    "bold": True,
                    "align": "center",
                    "font_size": 14
                })
                header_fmt = workbook.add_format({
                    "bold": True,
                    "border": 1,
                    "align": "center"
                })
                text_fmt = workbook.add_format({"border": 1})
                num_fmt = workbook.add_format({
                    "border": 1,
                    "num_format": "#,##0.00"
                })

                # Company Name
                worksheet.merge_range(
                    "A1:B1",
                    "Laxmi Civil Engineering Services Pvt Ltd",
                    title_fmt
                )

                # Assessment Year
                worksheet.merge_range(
                    "A2:B2",
                    f"Assessment Year : {assessment_year}",
                    workbook.add_format({"bold": True, "align": "center"})
                )

                # Header
                worksheet.write_row("A4", ["Particulars", "CIT"], header_fmt)

                # Data rows
                row_no = 4
                for _, row in df.iterrows():
                    worksheet.write(row_no, 0, row["Particulars"], text_fmt)
                    worksheet.write(row_no, 1, row["CIT"], num_fmt)
                    row_no += 1

                # Column widths
                worksheet.set_column("A:A", 45)
                worksheet.set_column("B:B", 20)

            output.seek(0)
            return output

        except mysql.connector.Error as e:
            print("MySQL Error →", e)
            return None


 # CLOSE CONNECTION
    def close(self):
        self.cursor.close()
        self.conn.close()