-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathdatabase_init.py
More file actions
47 lines (35 loc) · 1.24 KB
/
database_init.py
File metadata and controls
47 lines (35 loc) · 1.24 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from database_setup import *
engine = create_engine('sqlite:///catalog.db')
Base.metadata.bind = engine
DBsession = sessionmaker(bind=engine)
session = DBsession()
## Delete all previous entries
session.query(User).delete()
session.query(Category).delete()
session.query(Item).delete()
# Fake Users
User1 = User(id=1, name="Frank", email="frank@gmail.com",
picture="http://www.pieglobal.com/wp-content/uploads/2015/10/placeholder-user.png")
session.add(User1)
session.commit()
User2 = User(id=2, name="James", email="james@gmail.com",
picture="http://www.pieglobal.com/wp-content/uploads/2015/10/placeholder-user.png")
session.add(User2)
session.commit()
# Fake Categories
Category1 = Category(id=1, name="Tops")
session.add(Category1)
session.commit()
Category2 = Category(id=2, name="Bottoms")
session.add(Category2)
session.commit()
# Fake Items
Item1 = Item(id = 1, name="T-shirt", category_id = 1, user_id=1, description="Plain white T-shirt")
session.add(Item1)
session.commit()
Item2 = Item(id=2, name="Jeans", category_id=2, user_id=2, description="Long black Jeans")
session.add(Item2)
session.commit()
print "Database successfully populated"