nizika_ai/migration.py

from __future__ import annotations

from config import CONFIG
from eloquent import DatabaseManager, Schema

DB = DatabaseManager (CONFIG)
SCHEMA = Schema (DB)


def main (
) -> None:
    create_queries ()
    create_answers ()
    create_users ()
    create_query_answer_histories ()
    add_constraints_to_queries ()
    add_constraints_to_answers ()
    add_constraints_to_query_answer_histories ()


def create_queries (
) -> None:
    with SCHEMA.create ('queries') as table:
        table.big_increments ('id')
        table.unsigned_big_integer ('user_id').nullable ().comment ('クエリ主')
        table.integer ('target_character').comment ('クエリ先キャラクタ')
        table.text ('content').comment ('クエリ内容')
        table.string ('image_url').nullable ().default (None).comment ('添附画像 URL')
        table.integer ('query_type').comment ('クエリ区分')
        table.integer ('model').comment ('GPT のモデル')
        table.datetime ('sent_at').comment ('送信日時')
        table.boolean ('answered').default (False).comment ('回答済')


def add_constraints_to_queries (
) -> None:
    with SCHEMA.table ('queries') as table:
        table.foreign ('user_id').references ('id').on ('users').on_update ('cascade').on_delete ('restrict')


def create_answers (
) -> None:
    with SCHEMA.create ('answers') as table:
        table.big_increments ('id')
        table.unsigned_big_integer ('query_id').nullable ().comment ('クエリ')
        table.integer ('character').comment ('キャラクタ区分')
        table.text ('content').comment ('回答内容')
        table.integer ('answer_type').comment ('回答区分')
        table.datetime ('sent_at').comment ('送信日時')
        table.boolean ('answered').default (False).comment ('回答済')


def add_constraints_to_answers (
) -> None:
    with SCHEMA.table ('answers') as table:
        table.foreign ('query_id').references ('id').on ('queries').on_update ('cascade').on_delete ('restrict')


def create_users (
) -> None:
    with SCHEMA.create ('users') as table:
        table.big_increments ('id')
        table.integer ('platform').comment ('プラットフォーム区分')
        table.string ('code').comment ('ユーザ・コード（プラットフォーム依存）')
        table.string ('name').comment ('ユーザ名（プラットフォーム内）')
        table.binary ('icon').nullable ().comment ('アイコン')
        table.unique (['platform', 'code'])


def create_query_answer_histories (
) -> None:
    with SCHEMA.create ('query_answer_histories') as table:
        table.big_increments ('id')
        table.unsigned_big_integer ('query_id')
        table.unsigned_big_integer ('answer_id')
        table.unique (['query_id', 'answer_id'])


def add_constraints_to_query_answer_histories (
) -> None:
    with SCHEMA.table ('query_answer_histories') as table:
        table.foreign ('query_id').references ('id').on ('queries').on_update ('cascade').on_delete ('cascade')
        table.foreign ('answer_id').references ('id').on ('answers').on_update ('cascade').on_delete ('cascade')


if __name__ == '__main__':
    main ()