tjdify/api/models/intention.py

from sqlalchemy import func

from .engine import db
from .types import StringUUID


class IntentionType(db.Model):
    __tablename__ = "intention_types"
    __table_args__ = (db.PrimaryKeyConstraint("id", name="intention_type_id_pkey"),)

    id = db.Column(StringUUID, nullable=False, server_default=db.text("uuid_generate_v4()"))
    name = db.Column(db.String(255), nullable=False)
    created_by = db.Column(StringUUID, nullable=False)
    created_at = db.Column(db.DateTime, nullable=False, server_default=func.current_timestamp())
    updated_by = db.Column(StringUUID, nullable=True)
    updated_at = db.Column(db.DateTime, nullable=True, server_default=func.current_timestamp())

    @property
    def intention_count(self):
        return db.session.query(func.count(Intention.id)).filter(Intention.type_id == self.id).scalar()

    @property
    def intentions(self):
        return db.session.query(Intention).filter(Intention.type_id == self.id).all()


class Intention(db.Model):
    __tablename__ = "intentions"
    __table_args__ = (
        db.PrimaryKeyConstraint("id", name="intention_pkey"),
        db.Index("intention_type_idx", "type_id"),
    )

    id = db.Column(StringUUID, nullable=False, server_default=db.text("uuid_generate_v4()"))
    type_id = db.Column(StringUUID, nullable=False)
    name = db.Column(db.String(255), nullable=False)
    created_by = db.Column(StringUUID, nullable=False)
    created_at = db.Column(db.DateTime, nullable=False, server_default=func.current_timestamp())
    updated_by = db.Column(StringUUID, nullable=True)
    updated_at = db.Column(db.DateTime, nullable=True, server_default=func.current_timestamp())

    @property
    def type_name(self):
        return db.session.query(IntentionType.name).filter(IntentionType.id == self.type_id).first().name

    @property
    def type(self):
        return db.session.query(IntentionType).filter(IntentionType.id == self.type_id).first()

    @property
    def corpus(self):
        return db.session.query(IntentionCorpus).filter(IntentionCorpus.intention_id == self.id).all()

    @property
    def keywords(self):
        return db.session.query(IntentionKeyword).filter(IntentionKeyword.intention_id == self.id).all()

    @property
    def corpus_count(self):
        return db.session.query(func.count(IntentionCorpus.id)).filter(IntentionCorpus.intention_id == self.id).scalar()

    @property
    def keywords_count(self):
        return (
            db.session.query(func.count(IntentionKeyword.id)).filter(IntentionKeyword.intention_id == self.id).scalar()
        )


class IntentionKeyword(db.Model):
    __tablename__ = "intention_keywords"
    __table_args__ = (db.PrimaryKeyConstraint("id", name="intention_keyword_pkey"),)

    id = db.Column(StringUUID, nullable=False, server_default=db.text("uuid_generate_v4()"))
    name = db.Column(db.String(255), nullable=False)
    intention_id = db.Column(StringUUID, nullable=False)
    created_by = db.Column(StringUUID, nullable=False)
    created_at = db.Column(db.DateTime, nullable=False, server_default=func.current_timestamp())
    updated_by = db.Column(StringUUID, nullable=True)
    updated_at = db.Column(db.DateTime, nullable=True, server_default=func.current_timestamp())

    @property
    def intention(self):
        return db.session.query(Intention).filter(Intention.id == self.intention_id).first()


class IntentionCorpus(db.Model):
    __tablename__ = "intention_corpus"
    __table_args__ = (
        db.PrimaryKeyConstraint("id", name="intention_corpus_pkey"),
        db.Index("intention_corpus_idx", "intention_id"),
    )

    id = db.Column(StringUUID, nullable=False, server_default=db.text("uuid_generate_v4()"))
    question = db.Column(db.String(255), nullable=False)
    question_config = db.Column(db.JSON, nullable=False)
    intention_id = db.Column(StringUUID, nullable=False)
    created_by = db.Column(StringUUID, nullable=False)
    created_at = db.Column(db.DateTime, nullable=False, server_default=func.current_timestamp())
    updated_by = db.Column(StringUUID, nullable=True)
    updated_at = db.Column(db.DateTime, nullable=True, server_default=func.current_timestamp())

    @property
    def intention(self):
        return db.session.query(Intention).filter(Intention.id == self.intention_id).first()

    @property
    def similarity_questions(self):
        return (
            db.session.query(IntentionCorpusSimilarityQuestion)
            .filter(IntentionCorpusSimilarityQuestion.corpus_id == self.id)
            .all()
        )


class IntentionCorpusSimilarityQuestion(db.Model):
    __tablename__ = "intention_corpus_similarity_questions"
    __table_args__ = (
        db.PrimaryKeyConstraint("id", name="intention_corpus_similarity_question_pkey"),
        db.Index("intention_corpus_similarity_question_idx", "corpus_id"),
    )

    id = db.Column(StringUUID, nullable=False, server_default=db.text("uuid_generate_v4()"))
    question = db.Column(db.String(255), nullable=False)
    question_config = db.Column(db.JSON, nullable=False)
    corpus_id = db.Column(StringUUID, nullable=False)
    created_by = db.Column(StringUUID, nullable=False)
    created_at = db.Column(db.DateTime, nullable=False, server_default=func.current_timestamp())
    updated_by = db.Column(StringUUID, nullable=True)
    updated_at = db.Column(db.DateTime, nullable=True, server_default=func.current_timestamp())

    @property
    def corpus(self):
        return db.session.query(IntentionCorpus).filter(IntentionCorpus.id == self.corpus_id).first()