from sqlalchemy import Column, Integer, String, ForeignKey
from sqlalchemy.orm import relationship
from src.utils.db import Base


class Recommendation(Base):
    __tablename__ = "recommendations"

    recommendation_id = Column(Integer, primary_key=True, autoincrement=True)
    recommendation = Column(String, nullable=False)
    topic = Column(String, nullable=False)
    severity = Column(String, nullable=False)
    rank = Column(Integer, nullable=False)
    branch_id = Column(Integer, ForeignKey("branches.branch_id"), nullable=False)
    
    #Relationships
    branch = relationship("Branch", back_populates="recommendations")