v0.1.0

backend/app/crud/settings.py

@@ -0,0 +1,98 @@
+from sqlalchemy.orm import Session
+
+from app.core.config import settings as env_settings
+from app.core.logging import get_logger
+from app.models.settings import Settings as SettingsModel
+from app.schemas.settings import Settings as SettingsSchema
+from app.schemas.settings import SettingsCreate
+
+logger = get_logger(__name__)
+
+
+def get_settings(db: Session, with_password: bool = False) -> SettingsSchema:
+    """Retrieve application settings, prioritizing environment variables over database."""
+    logger.debug("Querying for settings")
+    db_settings = db.query(SettingsModel).first()
+
+    if not db_settings:
+        logger.info(
+            "No settings found in the database, creating new default settings from environment variables."
+        )
+
+        # Get env settings, but only for fields that exist in the DB model
+        model_fields = {c.name for c in SettingsModel.__table__.columns}
+        env_data_for_db = {
+            k: v for k, v in env_settings.model_dump().items() if k in model_fields
+        }
+
+        db_settings = SettingsModel(**env_data_for_db)
+        db.add(db_settings)
+        db.commit()
+        db.refresh(db_settings)
+        logger.info("Default settings created from environment variables.")
+
+    # Build dictionary from DB model attributes, handling possible None values
+    db_data = {
+        "id": db_settings.id,
+        "imap_server": db_settings.imap_server or "",
+        "imap_username": db_settings.imap_username or "",
+        "search_folder": db_settings.search_folder,
+        "move_to_folder": db_settings.move_to_folder,
+        "mark_as_read": db_settings.mark_as_read,
+        "email_check_interval": db_settings.email_check_interval,
+        "auto_add_new_senders": db_settings.auto_add_new_senders,
+    }
+
+    # Get all environment settings that were explicitly set.
+    env_data = env_settings.model_dump(exclude_unset=True)
+
+    # Merge them. env_data takes precedence.
+    merged_data = {**db_data, **env_data}
+
+    # The fields that came from env are the "locked" ones.
+    locked_fields = list(env_data.keys())
+    logger.debug(f"Locked fields from environment variables: {locked_fields}")
+
+    # Handle password separately for security
+    if with_password:
+        logger.debug("Including IMAP password in settings data")
+        if "imap_password" in locked_fields:
+            merged_data["imap_password"] = env_settings.imap_password
+        else:
+            merged_data["imap_password"] = db_settings.imap_password
+    elif "imap_password" in merged_data:
+        # Ensure password is not in the data if not requested
+        del merged_data["imap_password"]
+
+    # Create the final schema object
+    settings_schema = SettingsSchema.model_validate(merged_data)
+    settings_schema.locked_fields = locked_fields
+
+    return settings_schema
+
+
+def create_or_update_settings(db: Session, settings: SettingsCreate):
+    """Create or update application settings."""
+    logger.info("Creating or updating settings")
+    db_settings = db.query(SettingsModel).first()
+    if not db_settings:
+        logger.info("No existing settings found, creating new ones.")
+        db_settings = SettingsModel()
+        db.add(db_settings)
+
+    update_data = settings.model_dump()
+
+    # Do not update fields that are set by environment variables
+    locked_fields = list(env_settings.model_dump(exclude_unset=True).keys())
+    logger.debug(f"Fields locked by environment variables: {locked_fields}")
+
+    for key, value in update_data.items():
+        if key not in locked_fields:
+            setattr(db_settings, key, value)
+
+    db.commit()
+    db.refresh(db_settings)
+    logger.info("Successfully updated settings.")
+
+    # Return the updated settings including locked fields for a complete view
+    return get_settings(db)