From 4806894a5ddfdec84bd939b9e59219ff5c5947ee Mon Sep 17 00:00:00 2001 From: Billy6483 Date: Tue, 24 Sep 2024 01:03:01 -0400 Subject: [PATCH] freebie tracker --- lib/models.py | 18 ++++++++++++++++++ tests/test_freebies.py | 3 +++ 2 files changed, 21 insertions(+) create mode 100644 tests/test_freebies.py diff --git a/lib/models.py b/lib/models.py index 2681bee5a..1eb64197d 100644 --- a/lib/models.py +++ b/lib/models.py @@ -16,6 +16,9 @@ class Company(Base): name = Column(String()) founding_year = Column(Integer()) + freebies = relationship('Freebie', back_populates='company') + + def __repr__(self): return f'' @@ -25,5 +28,20 @@ class Dev(Base): id = Column(Integer(), primary_key=True) name= Column(String()) + freebies = relationship('Freebie', back_populates='dev') + def __repr__(self): return f'' + +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')) + + # Define relationships + dev = relationship('Dev', back_populates='freebies') + company = relationship('Company', back_populates='freebies') \ No newline at end of file diff --git a/tests/test_freebies.py b/tests/test_freebies.py new file mode 100644 index 000000000..a39cf68c8 --- /dev/null +++ b/tests/test_freebies.py @@ -0,0 +1,3 @@ +# tests/test_freebies.py +def test_example(): + assert 1 + 1 == 2