-
Notifications
You must be signed in to change notification settings - Fork 0
/
seed.py
39 lines (30 loc) · 1.06 KB
/
seed.py
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
from sqlalchemy.orm import sessionmaker
from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from models import Actor, Role, ActorRole, engine
Base = declarative_base()
engine = create_engine('sqlite:///actor_roles.db')
Base.metadata.create_all(engine)
session = sessionmaker()
session.configure(bind=engine)
Base.metadata.bind = engine
session = session()
bale = Actor(name="Christian Bale")
hathaway = Actor(name="Anne Hathaway")
pfeiffer = Actor(name="Michelle Pfeiffer")
keaton = Actor(name="Michael Keaton")
arnett = Actor(name="Will Arnett")
batman = Role(character="Batman")
catwoman = Role(character="Catwoman")
burry = Role(character="Dr. Michael Burry")
american_psycho = Role(character="Patrick Bateman")
batman.actors.append(bale)
batman.actors.append(keaton)
batman.actors.append(arnett)
catwoman.actors.append(pfeiffer)
catwoman.actors.append(hathaway)
bale.roles.append(burry)
bale.roles.append(american_psycho)
session.add_all([bale, hathaway, pfeiffer, keaton, arnett])
session.add_all([batman, catwoman])
session.commit()