from sqlalchemy import (
    Column, Integer, String
)

from models.base import Base, BaseMixin


class UserFeedback(Base, BaseMixin):
    __tablename__ = 'user_feedback'

    created_by = Column(String(255), nullable=False)
    session_context_id = Column(Integer, nullable=False)
    question_id = Column(String(255), nullable=False)
    feedback = Column(String(255), nullable=False)
    comment = Column(String(255), nullable=False)
    raw_question = Column(String(512))
    sql = Column(String(512))

    @classmethod
    def create(cls, s, created_by, session_context_id, question_id, feedback, comment, raw_question=None, sql=None):
        item = cls(
            created_by=created_by, session_context_id=session_context_id, question_id=question_id, feedback=feedback,
            comment=comment, raw_question=raw_question, sql=sql,
        )

        s.add(item)
        s.flush()

        return item