from sqlalchemy import (
    Column,
    String,
)

from models.base import Base, BaseMixin


class Prompt(Base, BaseMixin):
    __tablename__ = "prompt"

    name = Column(String(128), nullable=False, unique=True)
    content = Column(String(1024))

    @classmethod
    def create(cls, s, name, content):
        item = cls(name=name, content=content)
        s.add(item)
        s.flush()

    @classmethod
    def get_by_name(cls, s, name):
        return s.query(cls).filter(cls.name == name).first()