debug.py

#!/usr/bin/env python3

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

from models import Company, Dev, Freebie, Base

if __name__ == '__main__':
    engine = create_engine('sqlite:///freebies.db')
    Session = sessionmaker(bind=engine)
    session = Session()

    # Test relationships
    rick = session.query(Dev).filter_by(name="Rick").first()
    print("Rick's freebies:", rick.freebies)
    print("Rick's companies:", rick.companies)

    google = session.query(Company).filter_by(name="Google").first()
    print("Google's freebies:", google.freebies)
    print("Google's devs:", google.devs)

    freebie = session.query(Freebie).first()
    print("Freebie details:", freebie.print_details())

    print("Oldest company:", Company.oldest_company())

    print("Rick received MacBook:", rick.received_one("MacBook"))

    # Test give_freebie
    new_freebie = google.give_freebie(rick, "Laptop", 2000)
    session.add(new_freebie)
    session.commit()

    print("Rick's freebies after give_freebie:", rick.freebies)

    # Test give_away
    rick.give_away(morty, new_freebie)
    session.add(new_freebie)
    session.commit()

    print("Morty received Laptop:", morty.received_one("Laptop"))

    session.close()