models.py

# lib/models.py

from sqlalchemy import Column, Integer, String, ForeignKey
from sqlalchemy.orm import relationship
from sqlalchemy.ext.declarative import declarative_base
from lib.db import session

Base = declarative_base()

class Company(Base):
    __tablename__ = "companies"
    id = Column(Integer, primary_key=True)
    name = Column(String)
    founding_year = Column(Integer)

    freebies = relationship("Freebie", back_populates="company")
    devs = relationship("Dev", secondary="freebies", back_populates="companies", viewonly=True)

    def give_freebie(self, dev, item_name, value):
        """Create a freebie for a dev"""
        freebie = Freebie(item_name=item_name, value=value, dev=dev, company=self)
        session.add(freebie)
        session.commit()
        return freebie

    @classmethod
    def oldest_company(cls):
        """Return the company with the earliest founding year"""
        return session.query(cls).order_by(cls.founding_year).first()


class Dev(Base):
    __tablename__ = "devs"
    id = Column(Integer, primary_key=True)
    name = Column(String)

    freebies = relationship("Freebie", back_populates="dev")
    companies = relationship("Company", secondary="freebies", back_populates="devs", viewonly=True)

    def received_one(self, item_name):
        """Check if dev has received a freebie with a specific name"""
        return any(f.item_name == item_name for f in self.freebies)

    def give_away(self, dev, freebie):
        """Transfer a freebie to another dev if self owns it"""
        if freebie in self.freebies:
            freebie.dev = dev
            session.commit()
            return freebie
        return None


class Freebie(Base):
    __tablename__ = "freebies"
    id = Column(Integer, primary_key=True)
    item_name = Column(String)
    value = Column(Integer)
    dev_id = Column(Integer, ForeignKey("devs.id"))
    company_id = Column(Integer, ForeignKey("companies.id"))

    dev = relationship("Dev", back_populates="freebies")
    company = relationship("Company", back_populates="freebies")

    def print_details(self):
        """Return a string describing the freebie ownership"""
        return f"{self.dev.name} owns a {self.item_name} from {self.company.name}"