completed

Author: codeitownit

lib/debug.py

@@ -1,9 +1,13 @@
 #!/usr/bin/env python3
 
 from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker
 
-from models import Company, Dev
+
+from models import Company, Dev, Freebie
 
 if __name__ == '__main__':
     engine = create_engine('sqlite:///freebies.db')
+    Session = sessionmaker(bind=engine)
+    session = Session()
     import ipdb; ipdb.set_trace()

lib/freebies.db

@@ -0,0 +1,45 @@
+"""Add Freebie Model
+
+Revision ID: cbfc11270a30
+Revises: 5f72c58bf48c
+Create Date: 2023-12-10 17:19:02.371520
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = 'cbfc11270a30'
+down_revision = '5f72c58bf48c'
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table('company_devs',
+    sa.Column('company_id', sa.Integer(), nullable=False),
+    sa.Column('dev_id', sa.Integer(), nullable=False),
+    sa.ForeignKeyConstraint(['company_id'], ['companies.id'], name=op.f('fk_company_devs_company_id_companies')),
+    sa.ForeignKeyConstraint(['dev_id'], ['devs.id'], name=op.f('fk_company_devs_dev_id_devs')),
+    sa.PrimaryKeyConstraint('company_id', 'dev_id')
+    )
+    op.create_table('freebies',
+    sa.Column('id', sa.Integer(), nullable=False),
+    sa.Column('item_name', sa.String(), nullable=True),
+    sa.Column('value', sa.Integer(), nullable=True),
+    sa.Column('company_id', sa.Integer(), nullable=True),
+    sa.Column('dev_id', sa.Integer(), nullable=True),
+    sa.ForeignKeyConstraint(['company_id'], ['companies.id'], name=op.f('fk_freebies_company_id_companies')),
+    sa.ForeignKeyConstraint(['dev_id'], ['devs.id'], name=op.f('fk_freebies_dev_id_devs')),
+    sa.PrimaryKeyConstraint('id')
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_table('freebies')
+    op.drop_table('company_devs')
+    # ### end Alembic commands ###

lib/migrations/versions/cbfc11270a30_add_freebie_model.py

@@ -1,4 +1,4 @@
-from sqlalchemy import ForeignKey, Column, Integer, String, MetaData
+from sqlalchemy import ForeignKey, Table, Column, Integer, String, MetaData
 from sqlalchemy.orm import relationship, backref
 from sqlalchemy.ext.declarative import declarative_base
 
@@ -9,21 +9,51 @@
 
 Base = declarative_base(metadata=metadata)
 
+company_dev = Table(
+    'company_devs',
+    Base.metadata,
+    Column('company_id', ForeignKey('companies.id'), primary_key=True),
+    Column('dev_id', ForeignKey('devs.id'), primary_key=True),
+    extend_existing=True,
+)
+
+
 class Company(Base):
     __tablename__ = 'companies'
 
     id = Column(Integer(), primary_key=True)
     name = Column(String())
     founding_year = Column(Integer())
 
+    freebies = relationship('Freebie', backref=backref('company'))
+    devs = relationship('Dev', secondary=company_dev, back_populates='companies')
+
+
     def __repr__(self):
         return f'<Company {self.name}>'
+    
+
+class Freebie(Base):
+    __tablename__ = 'freebies'
+
+    id = Column(Integer(), primary_key=True)
+    item_name = Column(String())
+    value = Column(Integer())
+
+    company_id = Column(Integer(), ForeignKey('companies.id'))
+    dev_id = Column(Integer(), ForeignKey('devs.id'))
+
 
 class Dev(Base):
     __tablename__ = 'devs'
 
     id = Column(Integer(), primary_key=True)
     name= Column(String())
 
+    freebies = relationship('Freebie', backref=backref('dev'))
+    companies = relationship('Company', secondary=company_dev, back_populates='devs')
+
+
     def __repr__(self):
         return f'<Dev {self.name}>'
+

lib/models.py

@@ -1,3 +1,70 @@
 #!/usr/bin/env python3
 
-# Script goes here!
+from faker import Faker
+import random
+
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker
+
+from models import Company, Dev, Freebie
+
+if __name__ == '__main__':
+    engine = create_engine('sqlite:///freebies.db')
+    Session = sessionmaker(bind=engine)
+    session = Session()
+
+    session.query(Company).delete()
+    session.query(Dev).delete()
+    session.query(Freebie).delete()
+
+    fake = Faker()
+
+    items = ['T-shirt','hoodie', 'usb drive',
+                 'charger', 'notebooks', 'caps']
+    companies = []
+    for i in range(50):
+        company = Company(
+            name=fake.company(),
+            founding_year= fake.year(),
+        )
+
+        # add and commit individually to get IDs back
+        session.add(company)
+        session.commit()
+
+        companies.append(company)
+
+
+    devs = []
+    for i in range(25):
+        dev = Dev(
+            name=fake.name(),
+        )
+
+        session.add(dev)
+        session.commit()
+
+        devs.append(dev)
+
+
+    freebies = []
+    for company in companies:
+        for i in range(random.randint(1,5)):
+            dev = random.choice(devs)
+            if company not in dev.companies:
+                dev.companies.append(company)
+                session.add(dev)
+                session.commit()
+            
+            freebie = Freebie(
+                item_name=random.choice(items),
+                value=random.randint(0, 10),
+                company_id=company.id,
+                dev_id=dev.id,
+            )
+
+            freebies.append(freebie)
+
+    session.bulk_save_objects(freebies)
+    session.commit()
+    session.close()