"""RuleCondition model."""

from sqlalchemy import Column
from sqlalchemy import ForeignKey
from sqlalchemy import Integer
from sqlalchemy import String
from sqlalchemy.orm import relationship

from vector_job_rules.connectors import mysql


class RuleCondition(mysql.BaseModel):
    """RuleCondition Model."""

    __tablename__ = 'rule_condition'

    rule_condition_id = Column('rule_condition_id', Integer, primary_key=True)
    rule_id = Column('rule_id', Integer, ForeignKey('rule.rule_id'))
    rule_condition_field_id = Column(
        'rule_condition_field_id',
        Integer,
        ForeignKey('rule_condition_field.rule_condition_field_id'),
    )
    rule_condition_value = Column('rule_condition_value', String)

    rule = relationship('Rule', foreign_keys=[rule_id])
    rule_condition_field = relationship(
        'RuleConditionField', foreign_keys=[rule_condition_field_id]
    )


DEFAULT_FIELDS = {
    RuleCondition.rule_condition_id,
    RuleCondition.rule_id,
    RuleCondition.rule_condition_field_id,
    RuleCondition.rule_condition_value,
}