from tortoise import BaseDBAsyncClient


async def upgrade(db: BaseDBAsyncClient) -> str:
    return """
        ALTER TABLE "messages" ADD "cost" INT NOT NULL  DEFAULT -1;
        ALTER TABLE "users" ALTER COLUMN "free_credit" SET DEFAULT 50;"""


async def downgrade(db: BaseDBAsyncClient) -> str:
    return """
        ALTER TABLE "users" ALTER COLUMN "free_credit" SET DEFAULT 100;
        ALTER TABLE "messages" DROP COLUMN "cost";"""