Update app.py

app.py

@@ -332,42 +332,90 @@ def drop_missing_rows(df, columns):
         
 #     workbook.save('Log.xlsx')
 
+# def write_log(sheet_data_dict):
+#     workbook = openpyxl.Workbook()
+    
+#     for sheet_name, data_dict in sheet_data_dict.items():
+#         sheet = workbook.create_sheet(title=sheet_name)
+        
+#         for title, data in data_dict.items():
+#             # Write the title
+#             sheet.append([title])
+#             title_cell = sheet.cell(row=sheet.max_row, column=1)
+#             title_cell.font = Font(size=14, bold=True)  # Set the font size and bold
+            
+#             # Extract the list values and color
+#             lst = data[0]  # Extract the list of items to write
+#             color = data[1] if len(data) > 1 else None  # Extract the color if it exists
+            
+#             # Write each item of the list in a new row and apply the color if provided
+#             for item in lst:
+#                 sheet.append([item])  # Write item in a new row
+#                 if color:  # If color is provided, apply it to the row
+#                     fill = PatternFill(start_color=color, end_color=color, fill_type="solid")
+#                     cell = sheet.cell(row=sheet.max_row, column=1)
+#                     cell.fill = fill
+            
+#         # Adjust the column width
+#         for column in sheet.columns:
+#             max_length = 0
+#             column = [cell for cell in column]
+#             for cell in column:
+#                 try:  # Necessary to avoid error on empty cells
+#                     if len(str(cell.value)) > max_length:
+#                         max_length = len(cell.value)
+#                 except:
+#                     pass
+#             adjusted_width = (max_length + 2)
+#             sheet.column_dimensions[column[0].column_letter].width = adjusted_width
+    
+#     # Remove the default sheet created and save the workbook
+#     if "Sheet" in workbook.sheetnames:
+#         workbook.remove(workbook["Sheet"])
+        
+#     workbook.save('Log.xlsx')
+
 def write_log(sheet_data_dict):
     workbook = openpyxl.Workbook()
     
     for sheet_name, data_dict in sheet_data_dict.items():
         sheet = workbook.create_sheet(title=sheet_name)
         
+        col_index = 1  # Initialize column index to 1 to start writing from the first column
+        row_index = 1  # Initialize row index
+        
         for title, data in data_dict.items():
+            lst, color, adjacent = data[0], (data[1] if len(data) > 1 else None), data[2]
+            
+            if adjacent:  # Check if the key should be written adjacent to the previous key
+                row_index = max(row_index, sheet.max_row + 1)  # Get the maximum row index used so far in the sheet + 1
+            else:
+                col_index += 1  # Move to the next column for the next title
+                row_index = 1  # Reset row index
+            
             # Write the title
-            sheet.append([title])
-            title_cell = sheet.cell(row=sheet.max_row, column=1)
+            title_cell = sheet.cell(row=row_index, column=col_index)
+            title_cell.value = title
             title_cell.font = Font(size=14, bold=True)  # Set the font size and bold
             
-            # Extract the list values and color
-            lst = data[0]  # Extract the list of items to write
-            color = data[1] if len(data) > 1 else None  # Extract the color if it exists
-            
             # Write each item of the list in a new row and apply the color if provided
-            for item in lst:
-                sheet.append([item])  # Write item in a new row
-                if color:  # If color is provided, apply it to the row
+            for item_index, item in enumerate(lst, start=row_index + 1):
+                cell = sheet.cell(row=item_index, column=col_index)
+                cell.value = item  # Write item in the cell
+                if color:  # If color is provided, apply it to the cell
                     fill = PatternFill(start_color=color, end_color=color, fill_type="solid")
-                    cell = sheet.cell(row=sheet.max_row, column=1)
                     cell.fill = fill
             
-        # Adjust the column width
-        for column in sheet.columns:
+            # Adjust the column width
             max_length = 0
-            column = [cell for cell in column]
-            for cell in column:
+            for cell in sheet[column_letter(col_index)]:
                 try:  # Necessary to avoid error on empty cells
                     if len(str(cell.value)) > max_length:
                         max_length = len(cell.value)
                 except:
                     pass
             adjusted_width = (max_length + 2)
-            sheet.column_dimensions[column[0].column_letter].width = adjusted_width
+            sheet.column_dimensions[column_letter(col_index)].width = adjusted_width
     
     # Remove the default sheet created and save the workbook
     if "Sheet" in workbook.sheetnames:
@@ -375,6 +423,14 @@ def write_log(sheet_data_dict):
         
     workbook.save('Log.xlsx')
 
+def column_letter(index):
+    """Convert a column index into a column letter"""
+    letters = ""
+    while index > 0:
+        index, remainder = divmod(index - 1, 26)
+        letters = chr(65 + remainder) + letters
+    return letters
+
 warnings = []
 def is_found(c,text):
   global warnings
@@ -615,9 +671,9 @@ def process(files,button):
             sheet_data[name] = {
                 "Before columns": [old_olds],
                 "After columns": [columns, '00FF00'],
-                'Filled quarters':[filled_warn],
+                'Filled quarters':[filled_warn,None,True],
                 "Warnings": sheetwarnings
-            }
+            }            
             
             c_name = name.split('.')[0]+'_cleaned.csv'
             df.to_csv(c_name,index=False)