From 19657fbab2e73a8b3db58ceca6b54efafed351eb Mon Sep 17 00:00:00 2001
From: Denis Cheruiyot <Cheruiyotkp974@gmail.com>
Date: Tue, 3 Jun 2025 00:15:26 +0300
Subject: [PATCH] seed

---
 lib/models.py | 48 +++++++++++++++++++++++++++++++++++++++++++++++-
 lib/seed.py   | 29 +++++++++++++++++++++++++++++
 2 files changed, 76 insertions(+), 1 deletion(-)

diff --git a/lib/models.py b/lib/models.py
index 2681bee5a..7d971164e 100644
--- a/lib/models.py
+++ b/lib/models.py
@@ -16,14 +16,60 @@ class Company(Base):
     name = Column(String())
     founding_year = Column(Integer())
 
+    freebies = relationship("Freebie", back_populates="company")
+
     def __repr__(self):
         return f'<Company {self.name}>'
 
+    @property
+    def devs(self):
+        return list({f.dev for f in self.freebies})
+
+    def give_freebie(self, dev, item_name, value):
+        new_freebie = Freebie(item_name=item_name, value=value, dev=dev, company=self)
+        return new_freebie
+
+    @classmethod
+    def oldest_company(cls, session):
+        return session.query(cls).order_by(cls.founding_year).first()
+
+
 class Dev(Base):
     __tablename__ = 'devs'
 
     id = Column(Integer(), primary_key=True)
-    name= Column(String())
+    name = Column(String())
+
+    freebies = relationship("Freebie", back_populates="dev")
 
     def __repr__(self):
         return f'<Dev {self.name}>'
+
+    @property
+    def companies(self):
+        return list({f.company for f in self.freebies})
+
+    def received_one(self, item_name):
+        return any(f.item_name == item_name for f 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'))
+
+    # Relationships
+    dev = relationship("Dev", back_populates="freebies")
+    company = relationship("Company", back_populates="freebies")
+
+    def print_details(self):
+        return f"{self.dev.name} owns a {self.item_name} from {self.company.name}"
\ No newline at end of file
diff --git a/lib/seed.py b/lib/seed.py
index b16becbbb..685f996ba 100644
--- a/lib/seed.py
+++ b/lib/seed.py
@@ -1,3 +1,32 @@
 #!/usr/bin/env python3
 
 # Script goes here!
+
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker
+from models import Base, Company, Dev, Freebie
+
+engine = create_engine("sqlite:///freebies.db")
+Session = sessionmaker(bind=engine)
+session = Session()
+Base.metadata.create_all(engine)
+
+company1 = Company(name="ORAIMO", founding_year=2008)
+company2 = Company(name="JBL", founding_year=2012)
+company3 = Company(name="SONY", founding_year=2004)
+
+dev1 = Dev(name="Denis")
+dev2 = Dev(name="Collins")
+dev3 = Dev(name="Evans")
+
+session.add_all([company1, company2, company3, dev1, dev2, dev3])
+session.commit()
+
+f1 = Freebie(item_name="PS CONSOLE", value=95000, dev=dev1, company=company1)
+f2 = Freebie(item_name="GALAXY S25", value=120000, dev=dev2, company=company2)
+f3 = Freebie(item_name="PORTABLE JBL SYSTEM", value=75000, dev=dev1, company=company2)
+f4 = Freebie(item_name="SONY", value=60000, dev=dev3, company=company3)
+f5 = Freebie(item_name="ORAIMO BUDS", value=4500, dev=dev2, company=company1)
+
+session.add_all([f1, f2, f3, f4, f5])
+session.commit()
\ No newline at end of file