from tortoise import BaseDBAsyncClient


async def upgrade(db: BaseDBAsyncClient) -> str:
    return """
        CREATE TABLE IF NOT EXISTS "code" (
    "created_at" TIMESTAMPTZ NOT NULL  DEFAULT CURRENT_TIMESTAMP,
    "updated_at" TIMESTAMPTZ NOT NULL  DEFAULT CURRENT_TIMESTAMP,
    "id" SERIAL NOT NULL PRIMARY KEY,
    "usage_count" INT NOT NULL  DEFAULT 0,
    "code" VARCHAR(6) NOT NULL UNIQUE,
    "message" VARCHAR(255)
);
        ALTER TABLE "plans" ADD "description" VARCHAR(255);"""


async def downgrade(db: BaseDBAsyncClient) -> str:
    return """
        ALTER TABLE "plans" DROP COLUMN "description";
        DROP TABLE IF EXISTS "code";"""