38 lines
987 B
Python
38 lines
987 B
Python
"""Weekly period
|
|
|
|
Revision ID: e77395969585
|
|
Revises: d18cbd50f7c6
|
|
Create Date: 2022-12-08 16:35:27.506504+00:00
|
|
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = "e77395969585"
|
|
down_revision = "d18cbd50f7c6"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
op.execute("ALTER TYPE transactions.period ADD VALUE 'weekly' AFTER 'daily'")
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.execute(
|
|
"""CREATE TYPE transactions.period_new
|
|
AS ENUM ('daily', 'monthly', 'yearly')
|
|
"""
|
|
)
|
|
op.execute("UPDATE transactions.categories_schedules SET period = DEFAULT WHERE period = 'weekly'")
|
|
op.execute(
|
|
"""ALTER TABLE transactions.categories_schedules
|
|
ALTER COLUMN period TYPE transactions.period_new
|
|
USING period::text::transactions.period_new
|
|
"""
|
|
)
|
|
op.execute("DROP TYPE transactions.period")
|
|
op.execute("ALTER TYPE transactions.period_new RENAME TO period")
|