Blank Column in Excel Addition

scripts/core/handler/event_handler.py

@@ -201,11 +201,18 @@ class CustomReportHandler:
 
         # Write the column headers with the defined format.
         for col_index, value in enumerate(result_df.columns.values):
-            worksheet.write(
-                start_row, col_index, value,
-                workbook.add_format(
-                    SterliteRefineryTemplate.COLUMN_HEADER_FORMAT)
-            )
+            if value:
+                worksheet.write(
+                    start_row, col_index, value,
+                    workbook.add_format(
+                        SterliteRefineryTemplate.COLUMN_HEADER_FORMAT)
+                )
+            else:
+                worksheet.write(
+                    start_row, col_index, value,
+                    workbook.add_format(
+                        SterliteRefineryTemplate.BLANK_COLUMN_HEADER_FORMAT)
+                )
         logger.info(f"Shape of current data frame is {result_df.shape}")
 
         return result_df.shape, worksheet
@@ -251,9 +258,6 @@ class CustomReportHandler:
                         for each_blocks in report_template:
 
                             logger.info("Fetching each KPI data from queries")
-                            print("--------------------")
-                            print(each_blocks)
-                            print("-- \n")
                             each_blocks = self.get_queries_from_db(
                                 input_json=each_blocks, date_filter=each_date_range
                             )

scripts/core/utilities/postgresql_db_utils.py

@@ -22,7 +22,6 @@ class PostgresDBUtility:
 
     def create_connection(self, db=None):
         _db = db or self.db_name
-        print("Database ====================================> ", _db)
         conn = psycopg2.connect(
             database=_db,
             user=self.username,

scripts/template/sterlite_report_template.py

@@ -20,6 +20,11 @@ class SterliteRefineryTemplate:
         "font_color": "#021b5e",
         "border": 1,
     }
+    BLANK_COLUMN_HEADER_FORMAT = {
+        "bold": True,
+        "align": "center",
+        "valign": "center",
+    }
     REPORT_TEMPLATE = [
         {
             "CATHODE PRODUCTION REVISED": {