mirror of
https://github.com/mealie-recipes/mealie.git
synced 2024-12-03 19:00:17 -08:00
73 lines
2.5 KiB
Python
73 lines
2.5 KiB
Python
"""add recipe yield quantity
|
|
|
|
Revision ID: b1020f328e98
|
|
Revises: 3897397b4631
|
|
Create Date: 2024-10-23 15:50:59.888793
|
|
|
|
"""
|
|
|
|
import sqlalchemy as sa
|
|
from sqlalchemy import orm
|
|
|
|
from alembic import op
|
|
from mealie.db.models._model_utils.guid import GUID
|
|
from mealie.services.scraper.cleaner import clean_yield
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = "b1020f328e98"
|
|
down_revision: str | None = "3897397b4631"
|
|
branch_labels: str | tuple[str, ...] | None = None
|
|
depends_on: str | tuple[str, ...] | None = None
|
|
|
|
|
|
# Intermediate table definitions
|
|
class SqlAlchemyBase(orm.DeclarativeBase):
|
|
pass
|
|
|
|
|
|
class RecipeModel(SqlAlchemyBase):
|
|
__tablename__ = "recipes"
|
|
|
|
id: orm.Mapped[GUID] = orm.mapped_column(GUID, primary_key=True, default=GUID.generate)
|
|
recipe_yield: orm.Mapped[str | None] = orm.mapped_column(sa.String)
|
|
recipe_yield_quantity: orm.Mapped[float] = orm.mapped_column(sa.Float, index=True, default=0)
|
|
recipe_servings: orm.Mapped[float] = orm.mapped_column(sa.Float, index=True, default=0)
|
|
|
|
|
|
def parse_recipe_yields():
|
|
bind = op.get_bind()
|
|
session = orm.Session(bind=bind)
|
|
|
|
for recipe in session.query(RecipeModel).all():
|
|
try:
|
|
recipe.recipe_servings, recipe.recipe_yield_quantity, recipe.recipe_yield = clean_yield(recipe.recipe_yield)
|
|
except Exception:
|
|
recipe.recipe_servings = 0
|
|
recipe.recipe_yield_quantity = 0
|
|
|
|
session.commit()
|
|
|
|
|
|
def upgrade():
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
with op.batch_alter_table("recipes", schema=None) as batch_op:
|
|
batch_op.add_column(sa.Column("recipe_yield_quantity", sa.Float(), nullable=False, server_default="0"))
|
|
batch_op.create_index(batch_op.f("ix_recipes_recipe_yield_quantity"), ["recipe_yield_quantity"], unique=False)
|
|
batch_op.add_column(sa.Column("recipe_servings", sa.Float(), nullable=False, server_default="0"))
|
|
batch_op.create_index(batch_op.f("ix_recipes_recipe_servings"), ["recipe_servings"], unique=False)
|
|
|
|
# ### end Alembic commands ###
|
|
|
|
parse_recipe_yields()
|
|
|
|
|
|
def downgrade():
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
with op.batch_alter_table("recipes", schema=None) as batch_op:
|
|
batch_op.drop_index(batch_op.f("ix_recipes_recipe_servings"))
|
|
batch_op.drop_column("recipe_servings")
|
|
batch_op.drop_index(batch_op.f("ix_recipes_recipe_yield_quantity"))
|
|
batch_op.drop_column("recipe_yield_quantity")
|
|
|
|
# ### end Alembic commands ###
|