mirror of
https://github.com/mealie-recipes/mealie.git
synced 2025-05-24 01:12:54 -04:00
3c2744a3da
* add basic pre-commit file * add flake8 * add isort * add pep585-upgrade (typing upgrades) * use namespace for import * add mypy * update ci for backend * flake8 scope * fix version format * update makefile * disable strict option (temporary) * fix mypy issues * upgrade type hints (pre-commit) * add vscode typing check * add types to dev deps * remote container draft * update setup script * update compose version * run setup on create * dev containers update * remove unused pages * update setup tips * expose ports * Update pre-commit to include flask8-print (#1053) * Add in flake8-print to pre-commit * pin version of flake8-print * formatting * update getting strated docs * add mypy to pre-commit * purge .mypy_cache on clean * drop mypy Co-authored-by: zackbcom <zackbcom@users.noreply.github.com>
66 lines
2.1 KiB
Python
66 lines
2.1 KiB
Python
from collections.abc import Iterable
|
|
from typing import TypeVar
|
|
|
|
from pydantic import UUID4, BaseModel
|
|
from slugify import slugify
|
|
from sqlalchemy.orm import Session
|
|
|
|
from mealie.repos.all_repositories import AllRepositories
|
|
from mealie.repos.repository_factory import RepositoryGeneric
|
|
from mealie.schema.recipe import RecipeCategory
|
|
from mealie.schema.recipe.recipe import RecipeTag
|
|
from mealie.schema.recipe.recipe_category import CategoryOut, CategorySave, TagOut, TagSave
|
|
|
|
T = TypeVar("T", bound=BaseModel)
|
|
|
|
|
|
class DatabaseMigrationHelpers:
|
|
def __init__(self, db: AllRepositories, session: Session, group_id: UUID4, user_id: UUID4) -> None:
|
|
self.group_id = group_id
|
|
self.user_id = user_id
|
|
self.session = session
|
|
self.db = db
|
|
|
|
def _get_or_set_generic(
|
|
self, accessor: RepositoryGeneric, items: Iterable[str], create_model: type[T], out_model: type[T]
|
|
) -> list[T]:
|
|
"""
|
|
Utility model for getting or setting categories or tags. This will only work for those two cases.
|
|
|
|
This is probably a bad implementation of this pattern.
|
|
"""
|
|
items_out = []
|
|
|
|
for item_name in items:
|
|
slug_lookup = slugify(item_name)
|
|
|
|
item_model = accessor.get_one(value=slug_lookup, key="slug", override_schema=out_model)
|
|
|
|
if not item_model:
|
|
item_model = accessor.create(
|
|
create_model(
|
|
group_id=self.group_id,
|
|
name=item_name,
|
|
slug=slug_lookup,
|
|
)
|
|
)
|
|
|
|
items_out.append(item_model.dict())
|
|
return items_out
|
|
|
|
def get_or_set_category(self, categories: Iterable[str]) -> list[RecipeCategory]:
|
|
return self._get_or_set_generic(
|
|
self.db.categories.by_group(self.group_id),
|
|
categories,
|
|
CategorySave,
|
|
CategoryOut,
|
|
)
|
|
|
|
def get_or_set_tags(self, tags: Iterable[str]) -> list[RecipeTag]:
|
|
return self._get_or_set_generic(
|
|
self.db.tags.by_group(self.group_id),
|
|
tags,
|
|
TagSave,
|
|
TagOut,
|
|
)
|