from tortoise import BaseDBAsyncClient


async def upgrade(db: BaseDBAsyncClient) -> str:
    return """
        CREATE TABLE IF NOT EXISTS "plans" (
    "created_at" TIMESTAMPTZ NOT NULL  DEFAULT CURRENT_TIMESTAMP,
    "updated_at" TIMESTAMPTZ NOT NULL  DEFAULT CURRENT_TIMESTAMP,
    "id" VARCHAR(20) NOT NULL  PRIMARY KEY,
    "price" INT NOT NULL,
    "coins" INT NOT NULL,
    "free_coins" INT NOT NULL,
    "title" VARCHAR(255) NOT NULL,
    "active" BOOL NOT NULL  DEFAULT False
);"""


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