From 81ca199782558e73311255b63ac555ecbd59057c Mon Sep 17 00:00:00 2001
From: Bildad Ereggae <bildadereggae@gmail.com>
Date: Tue, 3 Jun 2025 22:40:29 +0300
Subject: [PATCH] freebie challenge done

---
 lib/models.py | 56 ++++++++++++++++++++++++++++++++++++++++++++++-----
 lib/seed.py   | 28 ++++++++++++++++++++++++++
 2 files changed, 79 insertions(+), 5 deletions(-)

diff --git a/lib/models.py b/lib/models.py
index 2681bee5a..50c2f6b05 100644
--- a/lib/models.py
+++ b/lib/models.py
@@ -9,21 +9,67 @@
 
 Base = declarative_base(metadata=metadata)
 
+
 class Company(Base):
-    __tablename__ = 'companies'
+    __tablename__ = "companies"
 
     id = Column(Integer(), primary_key=True)
     name = Column(String())
     founding_year = Column(Integer())
 
     def __repr__(self):
-        return f'<Company {self.name}>'
+        return f"<Company {self.name}>"
+
+    @property
+    def devs(self):
+        return list({freebie.dev for freebie in self.freebies})
+
+    def give_freebie(self, dev, item_name, value):
+        freebie = Freebie(item_name=item_name, value=value, dev=dev, company=self)
+        return freebie
+
+    @classmethod
+    def oldest_company(cls, session):
+        return session.query(cls).order_by(cls.founding_year).first()
+
 
 class Dev(Base):
-    __tablename__ = 'devs'
+    __tablename__ = "devs"
 
     id = Column(Integer(), primary_key=True)
-    name= Column(String())
+    name = Column(String())
 
     def __repr__(self):
-        return f'<Dev {self.name}>'
+        return f"<Dev {self.name}>"
+
+    @property
+    def companies(self):
+        return list({freebie.company for freebie in self.freebies})
+
+    def received_one(self, item_name):
+        return any(freebie.item_name == item_name for freebie in self.freebies)
+
+    def give_away(self, other_dev, freebie):
+        if freebie in self.freebies:
+            freebie.dev = other_dev
+
+
+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", backref=backref("freebies", cascade="all, delete-orphan"))
+    company = relationship(
+        "Company", backref=backref("freebies", cascade="all, delete-orphan")
+    )
+
+    def __repr__(self):
+        return f"<Freebie {self.item_name} (${self.value})>"
+
+    def print_details(self):
+        return f"{self.dev.name} owns a {self.item_name} from {self.company.name}"
diff --git a/lib/seed.py b/lib/seed.py
index b16becbbb..6331201ed 100644
--- a/lib/seed.py
+++ b/lib/seed.py
@@ -1,3 +1,31 @@
 #!/usr/bin/env python3
 
 # Script goes here!
+
+from sqlalchemy import create_engine
+from sqlalchemy.orm import Session
+from models import Base, Company, Dev, Freebie
+
+engine = create_engine("sqlite:///freebies.db")
+
+Base.metadata.create_all(engine)
+
+
+with Session(engine) as session:
+    company1 = Company(name="TechCorp", founding_year=1999)
+    company2 = Company(name="InnoSoft", founding_year=2010)
+    session.add_all([company1, company2])
+    session.commit()
+
+    dev1 = Dev(name="Bildad")
+    dev2 = Dev(name="Sakaa")
+    session.add_all([dev1, dev2])
+    session.commit()
+
+    freebie1 = Freebie(item_name="Sticker Pack", value=5, dev=dev1, company=company1)
+    freebie2 = Freebie(item_name="Coffee Mug", value=15, dev=dev1, company=company2)
+    freebie3 = Freebie(item_name="T-Shirt", value=20, dev=dev2, company=company1)
+    session.add_all([freebie1, freebie2, freebie3])
+    session.commit()
+
+    print("data created successfully.")
\ No newline at end of file