"""Der Soll-Betrag kann sich aus den Vertraegen rechnen.

Bisher war er eine Zahl, die man hinschreibt. Wer fuer Streaming 80 plant,
weil 58 fest gebunden sind und 22 Luft sein sollen, musste beim naechsten
Preisaufschlag nachrechnen – und dabei aufpassen, seine 22 nicht zu verlieren.

Mit "berechnet" rechnet sich der Soll-Betrag selbst: Summe der Vertraege
dieser Kategorie plus einem Zuschlag, den man behaelt.

Revision ID: 0032
Revises: 0031
"""

import sqlalchemy as sa
from alembic import op

revision = "0032"
down_revision = "0031"
branch_labels = None
depends_on = None

TABELLEN = ("budget_template_lines", "budget_period_lines")


def upgrade() -> None:
    for tabelle in TABELLEN:
        with op.batch_alter_table(tabelle) as batch:
            batch.add_column(sa.Column("modus", sa.String(length=16), nullable=False,
                                       server_default="fest"))
            batch.add_column(sa.Column("zuschlag_cents", sa.Integer(), nullable=False,
                                       server_default="0"))


def downgrade() -> None:
    for tabelle in TABELLEN:
        with op.batch_alter_table(tabelle) as batch:
            batch.drop_column("zuschlag_cents")
            batch.drop_column("modus")