Implement Excel file generation with variable injection from config.json

2025-09-12 10:51:02 +03:00
parent 1b9cb50961
commit 9c0435ddd8
7 changed files with 143 additions and 511 deletions
--- a/update_excel.py
+++ b/update_excel.py
@@ -0,0 +1,123 @@
+#!/usr/bin/env python3
+import json
+import os
+import openpyxl
+from openpyxl.utils import get_column_letter
+
+def update_excel_variables(excel_path):
+    """
+    Update the Variables sheet in the Excel file with values from config.json
+    
+    Args:
+        excel_path (str): Path to the Excel file to update
+    
+    Returns:
+        bool: True if successful, False otherwise
+    """
+    # Define paths
+    script_dir = os.path.dirname(os.path.abspath(__file__))
+    config_path = os.path.join(script_dir, 'config.json')
+    
+    try:
+        # Load config.json
+        with open(config_path, 'r') as f:
+            config = json.load(f)
+            user_data = config.get('user_data', {})
+        
+        # Load Excel workbook
+        print(f"Opening Excel file: {excel_path}")
+        wb = openpyxl.load_workbook(excel_path)
+        
+        # Try to access the Variables sheet
+        try:
+            # First try by name
+            sheet = wb['Variables']
+        except KeyError:
+            # If not found by name, try to access the last sheet
+            sheet_names = wb.sheetnames
+            if sheet_names:
+                print(f"Variables sheet not found by name. Using last sheet: {sheet_names[-1]}")
+                sheet = wb[sheet_names[-1]]
+            else:
+                print("No sheets found in the workbook")
+                return False
+        
+        # Map config variables to Excel cells based on the provided mapping
+        cell_mappings = {
+            'B2': user_data.get('store_name', ''),
+            'B31': user_data.get('starting_date', ''),
+            'B32': user_data.get('duration', 36),
+            'B37': user_data.get('open_days_per_month', 0),
+            
+            # Convenience store type
+            'H37': user_data.get('convenience_store_type', {}).get('stores_number', 0),
+            'C37': user_data.get('convenience_store_type', {}).get('monthly_transactions', 0),
+            # Convert boolean to 1/0 for has_digital_screens
+            'I37': 1 if user_data.get('convenience_store_type', {}).get('has_digital_screens', False) else 0,
+            'J37': user_data.get('convenience_store_type', {}).get('screen_count', 0),
+            'K37': user_data.get('convenience_store_type', {}).get('screen_percentage', 0),
+            # Convert boolean to 1/0 for has_in_store_radio
+            'M37': 1 if user_data.get('convenience_store_type', {}).get('has_in_store_radio', False) else 0,
+            'N37': user_data.get('convenience_store_type', {}).get('radio_percentage', 0),
+            
+            # Supermarket store type
+            'H38': user_data.get('supermarket_store_type', {}).get('stores_number', 0),
+            'C38': user_data.get('supermarket_store_type', {}).get('monthly_transactions', 0),
+            # Convert boolean to 1/0 for has_digital_screens
+            'I38': 1 if user_data.get('supermarket_store_type', {}).get('has_digital_screens', False) else 0,
+            'J38': user_data.get('supermarket_store_type', {}).get('screen_count', 0),
+            'K38': user_data.get('supermarket_store_type', {}).get('screen_percentage', 0),
+            # Convert boolean to 1/0 for has_in_store_radio
+            'M38': 1 if user_data.get('supermarket_store_type', {}).get('has_in_store_radio', False) else 0,
+            'N38': user_data.get('supermarket_store_type', {}).get('radio_percentage', 0),
+            
+            # Hypermarket store type
+            'H39': user_data.get('hypermarket_store_type', {}).get('stores_number', 0),
+            'C39': user_data.get('hypermarket_store_type', {}).get('monthly_transactions', 0),
+            # Convert boolean to 1/0 for has_digital_screens
+            'I39': 1 if user_data.get('hypermarket_store_type', {}).get('has_digital_screens', False) else 0,
+            'J39': user_data.get('hypermarket_store_type', {}).get('screen_count', 0),
+            'K39': user_data.get('hypermarket_store_type', {}).get('screen_percentage', 0),
+            # Convert boolean to 1/0 for has_in_store_radio
+            'M39': 1 if user_data.get('hypermarket_store_type', {}).get('has_in_store_radio', False) else 0,
+            'N39': user_data.get('hypermarket_store_type', {}).get('radio_percentage', 0),
+            
+            # On-site channels
+            'B43': user_data.get('website_visitors', 0),
+            'B44': user_data.get('app_users', 0),
+            'B45': user_data.get('loyalty_users', 0),
+            
+            # Off-site channels
+            'B49': user_data.get('facebook_followers', 0),
+            'B50': user_data.get('instagram_followers', 0),
+            'B51': user_data.get('google_views', 0),
+            'B52': user_data.get('email_subscribers', 0),
+            'B53': user_data.get('sms_users', 0),
+            'B54': user_data.get('whatsapp_contacts', 0)
+        }
+        
+        # Update the cells
+        for cell_ref, value in cell_mappings.items():
+            try:
+                sheet[cell_ref] = value
+                print(f"Updated {cell_ref} with value: {value}")
+            except Exception as e:
+                print(f"Error updating cell {cell_ref}: {e}")
+        
+        # Save the workbook
+        wb.save(excel_path)
+        print(f"Excel file updated successfully: {excel_path}")
+        return True
+        
+    except Exception as e:
+        print(f"Error updating Excel file: {e}")
+        return False
+
+if __name__ == "__main__":
+    # For testing purposes
+    import sys
+    if len(sys.argv) > 1:
+        excel_path = sys.argv[1]
+        update_excel_variables(excel_path)
+    else:
+        print("Please provide the path to the Excel file as an argument")