from tortoise import BaseDBAsyncClient


async def upgrade(db: BaseDBAsyncClient) -> str:
    return """
        ALTER TABLE "banners" ADD "title" VARCHAR(255);
        ALTER TABLE "banners" ALTER COLUMN "description" DROP DEFAULT;
        ALTER TABLE "banners" ALTER COLUMN "description" DROP NOT NULL;
        ALTER TABLE "banners" ALTER COLUMN "description" TYPE VARCHAR(255) USING "description"::VARCHAR(255);"""


async def downgrade(db: BaseDBAsyncClient) -> str:
    return """
        ALTER TABLE "banners" DROP COLUMN "title";
        ALTER TABLE "banners" ALTER COLUMN "description" TYPE TEXT USING "description"::TEXT;
        ALTER TABLE "banners" ALTER COLUMN "description" SET NOT NULL;
        ALTER TABLE "banners" ALTER COLUMN "description" SET;"""