mealie/mealie/db/db_settings.py

from app_config import USE_MONGO, USE_SQL

from db.db_base import BaseDocument
from db.db_setup import USE_MONGO, USE_SQL
from db.mongo.settings_models import SiteSettingsDocument, WebhooksDocument
from db.sql.db_session import create_session
from db.sql.settings_models import SiteSettingsModel


class _Settings(BaseDocument):
    def __init__(self) -> None:

        self.primary_key = "name"

        if USE_SQL:
            self.sql_model = SiteSettingsModel
            self.create_session = create_session

        self.document = SiteSettingsDocument

    def save_new(self, main: dict, webhooks: dict) -> str:

        if USE_MONGO:
            main["webhooks"] = WebhooksDocument(**webhooks)
            new_doc = self.document(**main)
            return new_doc.save()

        elif USE_SQL:
            session = create_session()
            new_settings = self.sql_model(main.get("name"), webhooks)

            session.add(new_settings)
            session.commit()

            return new_settings.dict()

    def update_mongo(self, name: str, new_data: dict) -> dict:
        if USE_MONGO:
            document = self.document.objects.get(name=name)
            if document:
                document.update(set__webhooks=WebhooksDocument(**new_data["webhooks"]))
                document.save()
        elif USE_SQL:
            return