add company logo

routes/main.py

@@ -7,6 +7,7 @@ from sqlalchemy import func, case, literal_column, text
 from datetime import datetime, timedelta
 import logging
 import sys
+import time
 
 # Set up logging to show in console
 logging.basicConfig(
@@ -399,6 +400,24 @@ def init_routes(main_bp):
         
         site_settings = SiteSettings.get_settings()
         
+        # Handle logo upload
+        if 'company_logo' in request.files:
+            logo_file = request.files['company_logo']
+            if logo_file and logo_file.filename:
+                # Delete old logo if it exists
+                if site_settings.company_logo:
+                    old_logo_path = os.path.join('static', 'uploads', 'company_logos', site_settings.company_logo)
+                    if os.path.exists(old_logo_path):
+                        os.remove(old_logo_path)
+                
+                # Save new logo
+                filename = secure_filename(logo_file.filename)
+                # Add timestamp to filename to prevent caching issues
+                filename = f"{int(time.time())}_{filename}"
+                logo_path = os.path.join('static', 'uploads', 'company_logos', filename)
+                logo_file.save(logo_path)
+                site_settings.company_logo = filename
+        
         # Update all company fields
         site_settings.company_name = request.form.get('company_name')
         site_settings.company_website = request.form.get('company_website')