From 35a89aeb848fdfc707174cb501cce9d56f722ebf Mon Sep 17 00:00:00 2001
From: Marvin <okongomarvin971@gmail.com>
Date: Mon, 1 Apr 2024 00:26:43 +0300
Subject: [PATCH] Completed Assignment

---
 lib/debug.py                              | 12 ++++---
 lib/migrations/versions/freebies_table.py | 16 +++++++++
 lib/models.py                             | 40 ++++++++++++++---------
 lib/seed.py                               | 23 +++++++++++--
 4 files changed, 70 insertions(+), 21 deletions(-)
 create mode 100644 lib/migrations/versions/freebies_table.py

diff --git a/lib/debug.py b/lib/debug.py
index 4f922eb69..10aeb71d5 100644
--- a/lib/debug.py
+++ b/lib/debug.py
@@ -1,9 +1,13 @@
 #!/usr/bin/env python3
 
 from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker
+from models import Base
 
-from models import Company, Dev
+engine = create_engine('sqlite:///freebies.db')
+Base.metadata.bind = engine
+DBSession = sessionmaker(bind=engine)
+session = DBSession()
 
-if __name__ == '__main__':
-    engine = create_engine('sqlite:///freebies.db')
-    import ipdb; ipdb.set_trace()
+# Start debugging session
+import ipdb; ipdb.set_trace()
diff --git a/lib/migrations/versions/freebies_table.py b/lib/migrations/versions/freebies_table.py
new file mode 100644
index 000000000..66c688918
--- /dev/null
+++ b/lib/migrations/versions/freebies_table.py
@@ -0,0 +1,16 @@
+from alembic import op
+import sqlalchemy as sa
+
+def upgrade():
+    op.create_table(
+        'freebies',
+        sa.Column('id', sa.Integer(), nullable=False),
+        sa.Column('dev_id', sa.Integer(), sa.ForeignKey('devs.id'), nullable=False),
+        sa.Column('company_id', sa.Integer(), sa.ForeignKey('companies.id'), nullable=False),
+        sa.Column('item_name', sa.String(), nullable=False),
+        sa.Column('value', sa.Integer(), nullable=False),
+        sa.PrimaryKeyConstraint('id')
+    )
+
+def downgrade():
+    op.drop_table('freebies')
diff --git a/lib/models.py b/lib/models.py
index 2681bee5a..eeba147ee 100644
--- a/lib/models.py
+++ b/lib/models.py
@@ -1,29 +1,39 @@
-from sqlalchemy import ForeignKey, Column, Integer, String, MetaData
-from sqlalchemy.orm import relationship, backref
+from sqlalchemy import ForeignKey, Column, Integer, String
+from sqlalchemy.orm import relationship
 from sqlalchemy.ext.declarative import declarative_base
 
-convention = {
-    "fk": "fk_%(table_name)s_%(column_0_name)s_%(referred_table_name)s",
-}
-metadata = MetaData(naming_convention=convention)
-
-Base = declarative_base(metadata=metadata)
+Base = declarative_base()
 
 class Company(Base):
     __tablename__ = 'companies'
 
-    id = Column(Integer(), primary_key=True)
-    name = Column(String())
-    founding_year = Column(Integer())
+    id = Column(Integer, primary_key=True)
+    name = Column(String)
+    founding_year = Column(Integer)
 
     def __repr__(self):
-        return f'<Company {self.name}>'
+        return f'<Company(name={self.name}, founding_year={self.founding_year})>'
 
 class Dev(Base):
     __tablename__ = 'devs'
 
-    id = Column(Integer(), primary_key=True)
-    name= Column(String())
+    id = Column(Integer, primary_key=True)
+    name = Column(String)
+
+    def __repr__(self):
+        return f'<Dev(name={self.name})>'
+
+class Freebie(Base):
+    __tablename__ = 'freebies'
+
+    id = Column(Integer, primary_key=True)
+    dev_id = Column(Integer, ForeignKey('devs.id'))
+    company_id = Column(Integer, ForeignKey('companies.id'))
+    item_name = Column(String)
+    value = Column(Integer)
+
+    dev = relationship("Dev", backref="freebies")
+    company = relationship("Company", backref="freebies")
 
     def __repr__(self):
-        return f'<Dev {self.name}>'
+        return f'<Freebie(dev={self.dev}, company={self.company}, item_name={self.item_name}, value={self.value})>'
diff --git a/lib/seed.py b/lib/seed.py
index b16becbbb..856cfd313 100644
--- a/lib/seed.py
+++ b/lib/seed.py
@@ -1,3 +1,22 @@
-#!/usr/bin/env python3
+from models import Base, Company, Dev, Freebie
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker
 
-# Script goes here!
+engine = create_engine('sqlite:///freebies.db')
+Base.metadata.create_all(engine)
+Session = sessionmaker(bind=engine)
+session = Session()
+
+# Add sample data
+company1 = Company(name="Company A", founding_year=2000)
+company2 = Company(name="Company B", founding_year=2010)
+dev1 = Dev(name="Developer 1")
+dev2 = Dev(name="Developer 2")
+
+freebie1 = Freebie(dev=dev1, company=company1, item_name="T-shirt", value=10)
+freebie2 = Freebie(dev=dev2, company=company2, item_name="Stickers", value=5)
+
+session.add_all([company1, company2, dev1, dev2, freebie1, freebie2])
+session.commit()
+
+print("Sample data added successfully.")