brewman/alembic/versions/eed0b382c287_lowercase_fastapi.py

"""lowercase and fastapi

Revision ID: eed0b382c287
Revises: 0bf3d70ee7de
Create Date: 2020-05-10 19:52:58.163810

"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy import table, column, select
from sqlalchemy.dialects import postgresql

# revision identifiers, used by Alembic.
revision = "eed0b382c287"
down_revision = "0bf3d70ee7de"
branch_labels = None
depends_on = None


def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_table("msg_subscribers")
    op.drop_table("msg_posts")
    op.drop_table("msg_threadtags")
    op.drop_table("msg_threads")
    op.drop_table("msg_tags")

    op.rename_table("auth_roles", "auth_permissions")
    with op.batch_alter_table("auth_permissions") as batch_op:
        batch_op.alter_column("RoleID", new_column_name="id")
        batch_op.alter_column("Name", new_column_name="name")

    op.rename_table("auth_groups", "auth_roles")
    with op.batch_alter_table("auth_roles") as batch_op:
        batch_op.alter_column("GroupID", new_column_name="id")
        batch_op.alter_column("Name", new_column_name="name")

    op.rename_table("auth_rolegroups", "role_permissions")
    with op.batch_alter_table("role_permissions") as batch_op:
        batch_op.alter_column("RoleGroupID", new_column_name="id")
        batch_op.alter_column("RoleID", new_column_name="permission_id")
        batch_op.alter_column("GroupID", new_column_name="role_id")

    op.rename_table("auth_usergroups", "user_roles")
    with op.batch_alter_table("user_roles") as batch_op:
        batch_op.alter_column("UserGroupID", new_column_name="id")
        batch_op.alter_column("UserID", new_column_name="user_id")
        batch_op.alter_column("GroupID", new_column_name="role_id")

    with op.batch_alter_table("attendances") as batch_op:
        batch_op.alter_column("Amount", new_column_name="amount")
        batch_op.alter_column("UserID", new_column_name="user_id")
        batch_op.alter_column("AttendanceType", new_column_name="attendance_type")
        batch_op.alter_column("CreationDate", new_column_name="creation_date")
        batch_op.alter_column("Date", new_column_name="date", type_=sa.Date(), nullable=False)
        batch_op.alter_column("EmployeeID", new_column_name="employee_id")
        batch_op.alter_column("AttendanceID", new_column_name="id")
        batch_op.alter_column("IsValid", new_column_name="is_valid")

    with op.batch_alter_table("auth_users") as batch_op:
        batch_op.alter_column("UserID", new_column_name="id")
        batch_op.alter_column("Name", new_column_name="username")
        batch_op.alter_column("Password", new_column_name="password")
        batch_op.alter_column("LockedOut", new_column_name="disabled")

    with op.batch_alter_table("batches") as batch_op:
        batch_op.alter_column("BatchID", new_column_name="id")
        batch_op.alter_column("Discount", new_column_name="discount")
        batch_op.alter_column("Name", new_column_name="name", type_=sa.Date(), nullable=False)
        batch_op.alter_column("ProductID", new_column_name="product_id")
        batch_op.alter_column("QuantityRemaining", new_column_name="quantity_remaining")
        batch_op.alter_column("Rate", new_column_name="rate")
        batch_op.alter_column("Tax", new_column_name="tax")

    with op.batch_alter_table("cost_centres") as batch_op:
        batch_op.alter_column("CostCentreID", new_column_name="id")
        batch_op.alter_column("Name", new_column_name="name")
        batch_op.alter_column("IsFixture", new_column_name="is_fixture")

    with op.batch_alter_table("product_groups") as batch_op:
        batch_op.alter_column("ProductGroupID", new_column_name="id")
        batch_op.alter_column("Name", new_column_name="name")
        batch_op.alter_column("IsFixture", new_column_name="is_fixture")

    with op.batch_alter_table("inventories") as batch_op:
        batch_op.alter_column("InventoryID", new_column_name="id")
        batch_op.alter_column("BatchID", new_column_name="batch_id")
        batch_op.alter_column("Discount", new_column_name="discount")
        batch_op.alter_column("ProductID", new_column_name="product_id")
        batch_op.alter_column("Quantity", new_column_name="quantity")
        batch_op.alter_column("Rate", new_column_name="rate")
        batch_op.alter_column("Tax", new_column_name="tax")
        batch_op.alter_column("VoucherID", new_column_name="voucher_id")

    with op.batch_alter_table("journals") as batch_op:
        batch_op.alter_column("Amount", new_column_name="amount")
        batch_op.alter_column("CostCentreID", new_column_name="cost_centre_id")
        batch_op.alter_column("Debit", new_column_name="debit")
        batch_op.alter_column("JournalID", new_column_name="id")
        batch_op.alter_column("VoucherID", new_column_name="voucher_id")

    with op.batch_alter_table("fingerprints") as batch_op:
        batch_op.alter_column("FingerprintID", new_column_name="id")
        batch_op.alter_column("Date", new_column_name="date")
        batch_op.alter_column("EmployeeID", new_column_name="employee_id")

    with op.batch_alter_table("products") as batch_op:
        batch_op.alter_column("ProductID", new_column_name="id")
        batch_op.alter_column("Code", new_column_name="code")
        batch_op.alter_column("Fraction", new_column_name="fraction")
        batch_op.alter_column("FractionUnits", new_column_name="fraction_units")
        batch_op.alter_column("IsActive", new_column_name="is_active")
        batch_op.alter_column("IsFixture", new_column_name="is_fixture")
        batch_op.alter_column("Name", new_column_name="name")
        batch_op.alter_column("ProductGroupID", new_column_name="product_group_id")
        batch_op.alter_column("ProductYield", new_column_name="product_yield")
        batch_op.alter_column("Units", new_column_name="units")

    with op.batch_alter_table("recipes") as batch_op:
        batch_op.alter_column("recipe_id", new_column_name="id")

    op.rename_table("salary_deductions", "employee_benefit")
    with op.batch_alter_table("employee_benefit") as batch_op:
        batch_op.alter_column("SalaryDeductionID", new_column_name="id")
        batch_op.alter_column("DaysWorked", new_column_name="days_worked")
        batch_op.alter_column("EsiEmployee", new_column_name="esi_employee")
        batch_op.alter_column("EsiEmployer", new_column_name="esi_employer")
        batch_op.alter_column("GrossSalary", new_column_name="gross_salary")
        batch_op.alter_column("JournalID", new_column_name="journal_id")
        batch_op.alter_column("PfEmployee", new_column_name="pf_employee")
        batch_op.alter_column("PfEmployer", new_column_name="pf_employer")
        batch_op.alter_column("VoucherID", new_column_name="voucher_id")

    with op.batch_alter_table("vouchers") as batch_op:
        batch_op.alter_column("VoucherID", new_column_name="id")
        batch_op.alter_column("date", type_=sa.Date(), nullable=False)
        batch_op.alter_column("reconcile_date", type_=sa.Date(), nullable=False)

    op.rename_table("service_charges", "incentives")

    with op.batch_alter_table("employees") as batch_op:
        batch_op.alter_column("Designation", new_column_name="designation", nullable=False)
        batch_op.alter_column("Salary", new_column_name="salary", nullable=False)
        batch_op.alter_column("ServicePoints", new_column_name="points", nullable=False)
        batch_op.alter_column(
            "JoiningDate", new_column_name="joining_date", type_=sa.Date(), nullable=False,
        )
        batch_op.alter_column(
            "LeavingDate", new_column_name="leaving_date", type_=sa.Date(), nullable=True,
        )

    with op.batch_alter_table("settings") as batch_op:
        batch_op.alter_column("SettingID", new_column_name="id")
        batch_op.alter_column("Name", new_column_name="name")
        batch_op.alter_column("Data", new_column_name="data")

    op.create_unique_constraint(op.f("uq_accounts_name"), "accounts", ["name"])
    op.drop_constraint("accounts_name_key", "accounts", type_="unique")
    op.create_unique_constraint(op.f("uq_auth_clients_code"), "auth_clients", ["code"])
    op.create_unique_constraint(op.f("uq_auth_clients_name"), "auth_clients", ["name"])
    op.drop_constraint("auth_clients_Code_key", "auth_clients", type_="unique")
    op.drop_constraint("auth_clients_Name_key", "auth_clients", type_="unique")
    op.create_unique_constraint(
        op.f("uq_auth_login_history_user_id"), "auth_login_history", ["user_id", "client_id", "date"],
    )
    op.drop_constraint(
        "auth_login_history_user_id_client_id_date_key", "auth_login_history", type_="unique",
    )
    op.create_unique_constraint(op.f("uq_auth_permissions_name"), "auth_permissions", ["name"])
    op.drop_constraint("auth_roles_Name_key", "auth_permissions", type_="unique")
    op.create_unique_constraint(op.f("uq_auth_roles_name"), "auth_roles", ["name"])
    op.drop_constraint("auth_groups_Name_key", "auth_roles", type_="unique")
    op.create_unique_constraint(op.f("uq_auth_users_username"), "auth_users", ["username"])
    op.drop_constraint("auth_users_Name_key", "auth_users", type_="unique")
    op.create_unique_constraint(op.f("uq_cost_centres_name"), "cost_centres", ["name"])
    op.drop_constraint("entities_costcenters_Name_key", "cost_centres", type_="unique")
    op.drop_constraint("uq_employee_id_date", "fingerprints", type_="unique")
    op.create_index(op.f("ix_inventories_voucher_id"), "inventories", ["voucher_id"], unique=False)
    op.create_unique_constraint(op.f("uq_inventories_voucher_id"), "inventories", ["voucher_id", "batch_id"])
    op.drop_index("ix_inventories_VoucherID", table_name="inventories")
    op.create_index(op.f("ix_journals_voucher_id"), "journals", ["voucher_id"], unique=False)
    op.drop_index("ix_journals_VoucherID", table_name="journals")
    op.create_unique_constraint(op.f("uq_product_groups_name"), "product_groups", ["name"])
    op.drop_constraint("entities_productgroups_Name_key", "product_groups", type_="unique")
    op.create_unique_constraint(op.f("uq_products_code"), "products", ["code"])
    op.create_unique_constraint(op.f("uq_products_name"), "products", ["name", "units"])
    op.drop_constraint("products_Code_key", "products", type_="unique")
    op.drop_constraint("products_Name_Units_key", "products", type_="unique")
    op.create_unique_constraint(op.f("uq_recipe_items_recipe_id"), "recipe_items", ["recipe_id", "product_id"])
    op.drop_constraint("recipe_items_recipe_id_product_id_key", "recipe_items", type_="unique")
    op.create_unique_constraint(op.f("uq_settings_name"), "settings", ["name"])
    op.drop_constraint("settings_Name_key", "settings", type_="unique")
    op.create_index(op.f("ix_vouchers_date"), "vouchers", ["date"], unique=False)
    op.create_foreign_key(
        "fk_incentives_journal_id_journals", "incentives", "journals", ["journal_id"], ["id"],
    )
    op.create_foreign_key(
        "fk_incentives_voucher_id_vouchers", "incentives", "vouchers", ["voucher_id"], ["id"],
    )
    op.drop_constraint("service_charges_journal_id_fkey", "incentives", type_="foreignkey")
    op.drop_constraint("service_charges_voucher_id_fkey", "incentives", type_="foreignkey")
    op.create_foreign_key(
        "fk_employee_benefit_journal_id_journals", "employee_benefit", "journals", ["journal_id"], ["id"],
    )
    op.create_foreign_key(
        "fk_employee_benefit_voucher_id_vouchers", "employee_benefit", "vouchers", ["voucher_id"], ["id"],
    )
    op.drop_constraint(
        "entities_salarydeductions_JournalID_fkey", "employee_benefit", type_="foreignkey",
    )
    op.drop_constraint("salary_deductions_VoucherID_fkey", "employee_benefit", type_="foreignkey")

    permission = table("auth_permissions", column("id", postgresql.UUID()), column("name", sa.String))
    op.execute(
        permission.update()
        .where(permission.c.name == op.inline_literal("Service Charge"))
        .values({"name": op.inline_literal("Incentive")})
    )
    op.execute(
        permission.update()
        .where(permission.c.name == op.inline_literal("Salary Deduction"))
        .values({"name": op.inline_literal("Employee Benefit")})
    )
    account = table("accounts", column("name", sa.String))
    op.execute(
        account.update()
        .where(account.c.name == op.inline_literal("Service Charges"))
        .values({"name": op.inline_literal("Incentives")})
    )
    # Remove unneeded Payment and Receipt permissions
    permission = table("auth_permissions", column("id", postgresql.UUID()), column("name", sa.String))
    role_permission = table(
        "role_permissions", column("permission_id", postgresql.UUID()), column("role_id", postgresql.UUID()),
    )
    op.execute(
        role_permission.delete().where(
            role_permission.c.permission_id.in_(
                select([permission.c.id], permission.c.name.in_(["Payment", "Receipt"]))
            )
        )
    )
    op.execute(permission.delete().where(permission.c.name.in_(["Payment", "Receipt"])))

    ### end Alembic commands ###


def downgrade():
    raise Exception("Irreversible migration")