report formate changes all model

AppCode/ITATHandler.py

@@ -65,32 +65,123 @@ class ITATHandler:
         self.conn.commit()
 
 
-   
     def itat_report_download(self, selected_year):
-            try:
-                # Call stored procedure
-                self.cursor.callproc("GetITATByYear", [selected_year])
-                rows = []
-                for result in self.cursor.stored_results():
-                    rows = result.fetchall()
+        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}"
 
-                if not rows:
-                    return None
+            # Fetch ITAT data
+            self.cursor.callproc("GetITATByYear", [selected_year])
+            rows = []
+            for result in self.cursor.stored_results():
+                rows = result.fetchall()
 
-                df = pd.DataFrame(rows)
-
-                # Excel output
-                output = io.BytesIO()
-                with pd.ExcelWriter(output, engine="xlsxwriter") as writer:
-                    df.T.to_excel(writer, header=False, sheet_name="ITAT_Report")
-
-                output.seek(0)
-                return output
-
-            except mysql.connector.Error as e:
-                print("MySQL Error:", e)
+            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 ITAT structure
+            data = []
+            for key, label in field_mapping.items():
+                value = rows[0].get(key, 0)
+                data.append([label, value])
+
+            df = pd.DataFrame(data, columns=["Particulars", "ITAT"])
+
+            # Excel output
+            output = io.BytesIO()
+            with pd.ExcelWriter(output, engine="xlsxwriter") as writer:
+                workbook = writer.book
+                worksheet = workbook.add_worksheet("ITAT Report")
+                writer.sheets["ITAT 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", "ITAT"], 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["ITAT"], 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()