from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from database_setup import User, Category, Base, Item

engine = create_engine('postgresql://vagrant:word@localhost/itemcatalog')
Base.metadata.bind = engine

DBSession = sessionmaker(bind=engine)
session = DBSession()


# Create dummy user
User1 = User(name="Sam Smith", email="sam.smith@udacity.com")
session.add(User1)
session.commit()


# Items for 'Household Items'
category1 = Category(user_id=1, name="Household Items")
session.add(category1)
session.commit()

item1 = Item(user_id=1, name="Pencil",
             description="A writing instrument (erasable)",
             price="$2.99", category=category1)
session.add(item1)
session.commit()

item2 = Item(user_id=1, name="Pen",
             description="A writing instrument (non-erassable)",
             price="$3.50", category=category1)
session.add(item2)
session.commit()

item3 = Item(user_id=1, name="Tissue",
             description="Magical artifact with many uses",
             price="$300", category=category1)
session.add(item3)
session.commit()

item4 = Item(user_id=1, name="Blanket",
             description="Large flat sheet for warmth",
             price="$10.50", category=category1)
session.add(item4)
session.commit()


# Items for 'Sports Equipment'
category2 = Category(user_id=1, name="Sports Equipment")
session.add(category2)
session.commit()

item5 = Item(user_id=1, name="Tennis racquet",
             description="Thing with a head and handle to hit balls with",
             price="$85", category=category2)
session.add(item5)
session.commit()

item6 = Item(user_id=1, name="Basketball",
             description="A ball to throw into a basket",
             price="$33", category=category2)
session.add(item6)
session.commit()

item7 = Item(user_id=1, name="Towel",
             description="Wipe off perspiration",
             price="$3", category=category2)
session.add(item7)
session.commit()


print "All categories and items successfully added."
