24 lines
814 B
Python
24 lines
814 B
Python
from tortoise import BaseDBAsyncClient
|
|
|
|
|
|
async def upgrade(db: BaseDBAsyncClient) -> str:
|
|
return """
|
|
CREATE TABLE IF NOT EXISTS "suno_tasks" (
|
|
"created_at" TIMESTAMPTZ NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
|
"updated_at" TIMESTAMPTZ NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
|
"id" UUID NOT NULL PRIMARY KEY,
|
|
"task_id" VARCHAR(255) NOT NULL UNIQUE,
|
|
"status" VARCHAR(50) NOT NULL DEFAULT 'PENDING',
|
|
"audio_url" VARCHAR(500),
|
|
"image_url" VARCHAR(500),
|
|
"audio_url2" VARCHAR(500),
|
|
"image_url2" VARCHAR(500),
|
|
"chatbot_id" INT NOT NULL REFERENCES "chatbots" ("id") ON DELETE CASCADE,
|
|
"user_id" UUID NOT NULL REFERENCES "users" ("id") ON DELETE CASCADE
|
|
);"""
|
|
|
|
|
|
async def downgrade(db: BaseDBAsyncClient) -> str:
|
|
return """
|
|
DROP TABLE IF EXISTS "suno_tasks";"""
|