mirror of
https://github.com/mealie-recipes/mealie.git
synced 2025-12-22 02:05:13 -05:00
78 lines
2.0 KiB
Python
78 lines
2.0 KiB
Python
from sqlalchemy.orm.session import Session
|
|
|
|
from db.db_base import BaseDocument
|
|
from db.sql.meal_models import MealPlanModel
|
|
from db.sql.recipe_models import Category, RecipeModel, Tag
|
|
from db.sql.settings_models import SiteSettingsModel
|
|
from db.sql.theme_models import SiteThemeModel
|
|
|
|
"""
|
|
# TODO
|
|
- [ ] Abstract Classes to use save_new, and update from base models
|
|
- [x] Create Category and Tags Table with Many to Many relationship
|
|
"""
|
|
|
|
|
|
class _Recipes(BaseDocument):
|
|
def __init__(self) -> None:
|
|
self.primary_key = "slug"
|
|
self.sql_model = RecipeModel
|
|
|
|
def update_image(self, session: Session, slug: str, extension: str) -> str:
|
|
entry: RecipeModel = self._query_one(session, match_value=slug)
|
|
entry.image = f"{slug}.{extension}"
|
|
session.commit()
|
|
|
|
return f"{slug}.{extension}"
|
|
|
|
|
|
class _Categories(BaseDocument):
|
|
def __init__(self) -> None:
|
|
self.primary_key = "name"
|
|
self.sql_model = Category
|
|
|
|
|
|
class _Tags(BaseDocument):
|
|
def __init__(self) -> None:
|
|
self.primary_key = "name"
|
|
self.sql_model = Tag
|
|
|
|
|
|
class _Meals(BaseDocument):
|
|
def __init__(self) -> None:
|
|
self.primary_key = "uid"
|
|
self.sql_model = MealPlanModel
|
|
|
|
|
|
class _Settings(BaseDocument):
|
|
def __init__(self) -> None:
|
|
self.primary_key = "name"
|
|
self.sql_model = SiteSettingsModel
|
|
|
|
def save_new(self, session: Session, main: dict, webhooks: dict) -> str:
|
|
new_settings = self.sql_model(main.get("name"), webhooks)
|
|
|
|
session.add(new_settings)
|
|
session.commit()
|
|
|
|
return new_settings.dict()
|
|
|
|
|
|
class _Themes(BaseDocument):
|
|
def __init__(self) -> None:
|
|
self.primary_key = "name"
|
|
self.sql_model = SiteThemeModel
|
|
|
|
|
|
class Database:
|
|
def __init__(self) -> None:
|
|
self.recipes = _Recipes()
|
|
self.meals = _Meals()
|
|
self.settings = _Settings()
|
|
self.themes = _Themes()
|
|
self.categories = _Categories()
|
|
self.tags = _Tags()
|
|
|
|
|
|
db = Database()
|