21 lines
657 B
Python
21 lines
657 B
Python
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";"""
|