from sqlalchemy import Column, Integer, ForeignKey, Float
from sqlalchemy.orm import relationship

from ..database import Base

class ProdutoOrcamento(Base):
    __tablename__ = "produto_orcamento"

    orcamento_id = Column(Integer, ForeignKey("orcamento.orc_id"), primary_key=True, name="orc_id")
    produto_id = Column(Integer, ForeignKey("produto.prod_id"), primary_key=True, name="prod_id")
    quantidade = Column(Integer, name="prod_orc_quantidade")
    preco_unitario = Column(Float, name="prod_orc_preco_unitario")

    produto = relationship("Produto", back_populates="orcamentos")
    orcamento = relationship("Orcamento", back_populates="produtos")