def load_users():
"""Load users from u.user into database."""
print "Users"
# Delete all rows in table, so if we need to run this a second time,
# we won't be trying to add duplicate users
User.query.delete()
# Read u.user file and insert data
for row in open("seed_data/u.user"):
row = row.rstrip()
# unpacking
user_id, name, email, password, provider, dob = row.split(",")
if dob:
dob = datetime.datetime.strptime(dob, "%d-%b-%Y")
else:
dob = None
user = User(user_id=user_id,
name=name,
email=email,
password=password,
provider=provider,
dob=dob)
# We need to add to the session or it won't ever be stored
db.session.add(user)
# Once we're done, we should commit our work
db.session.commit()
def mutate(self, info, personData):
print("create mutate")
print(personData)
id = personData.id
name = personData.name
print("id:", id)
print("name:", name)
person = User(id=personData.id, name=name)
UserList.append(person)
return CreateUser1(person)
def mutate(self, info, personData):
print("update mutate")
print(personData)
global UserList
id = personData.id
for user in UserList:
if user.id == id :
print("found")
user.fn = personData.fn
user.ln = personData.ln
person = User(id = personData.id, fn = personData.fn, ln = personData.ln)
return UpdateUser1(person=person)
def mutate(self, info, personData):
print("update mutate")
print(personData)
global UserList
id = personData.id
name = personData.name
for user in UserList:
if user.id == id:
print("found")
user.name = name
person = User(id=personData.id, name=name)
return UpdateUser1(person=person)
def mutate(self, info, personData):
print("delete mutate")
print(personData)
global UserList
id = personData.id
index = -1
for idx, user in enumerate(UserList):
if user.id == id :
del UserList[idx]
break
person = User(id = personData.id)
return DeleteUser1(person=person)
def mutate(self, info, personData):
print("create mutate")
print(personData)
person = User(id = personData.id, fn = personData.fn, ln= personData.ln)
UserList.append(person)
return CreateUser1(person=person)
def example_data():
"""Create the sample data for testing"""
genre1 = Genre(name="Music", themoviedb_id=10402)
genre2 = Genre(name="War", themoviedb_id=10752)
genre3 = Genre(name="Comedy", themoviedb_id=35)
genre4 = Genre(name="Family", themoviedb_id=10751)
db.session.add_all([genre1, genre2, genre3, genre4])
db.session.commit()
fred = User(
name="Fred",
email="*****@*****.**",
password="******",
provider="Cinemania",
dob="30-Aug-2000")
ann = User(
name=None,
email="*****@*****.**",
password=
"******",
provider="Cinemania",
dob=None)
db.session.add_all([fred, ann])
db.session.commit()
movie1 = Movie(
themoviedb_id=135397,
title="Jurassic World",
poster_url=
"https://image.tmdb.org/t/p/w500/jjBgi2r5cRt36xF6iNUEhzscEcb.jpg")
movie2 = Movie(
themoviedb_id=381288,
title="Split",
poster_url=
"https://image.tmdb.org/t/p/w500/rXMWOZiCt6eMX22jWuTOSdQ98bY.jpg")
movie3 = Movie(
themoviedb_id=283366,
title="Miss Peregrine's Home for Peculiar Children",
poster_url=
"https://image.tmdb.org/t/p/w500/uSHjeRVuObwdpbECaXJnvyDoeJK.jpg")
movie4 = Movie(
themoviedb_id=205596,
title="The Imitation Game",
poster_url=
"https://image.tmdb.org/t/p/w500/noUp0XOqIcmgefRnRZa1nhtRvWO.jpg")
movie5 = Movie(
themoviedb_id=226,
title="Boys Don't Cry",
poster_url=
"https://image.tmdb.org/t/p/w500/6bqIZTEuJnUrgnxcymciszvOz8J.jpg")
movie6 = Movie(
themoviedb_id=11858,
title="Renaissance Man",
poster_url=
"https://https://image.tmdb.org/t/p/w500/uaiykEOEFs91WclowDhsVYNdGfX.jpg"
)
db.session.add_all([movie1, movie2, movie3, movie4, movie5, movie6])
db.session.commit()
fred_music = UserGenre(user_id=fred.user_id, genre_id=genre1.genre_id)
fred_war = UserGenre(user_id=fred.user_id, genre_id=genre2.genre_id)
fred_comedy = UserGenre(user_id=fred.user_id, genre_id=genre3.genre_id)
fred_family = UserGenre(user_id=fred.user_id, genre_id=genre4.genre_id)
fred_movie1 = UserMovie(user_id=fred.user_id,
movie_id=movie1.movie_id,
rating=5,
seen="f")
fred_movie2 = UserMovie(user_id=fred.user_id,
movie_id=movie2.movie_id,
rating=None,
seen="f")
fred_movie3 = UserMovie(user_id=fred.user_id,
movie_id=movie3.movie_id,
rating=8,
seen="t")
fred_movie4 = UserMovie(user_id=fred.user_id,
movie_id=movie4.movie_id,
rating=2,
seen="t")
fred_movie5 = UserMovie(user_id=fred.user_id,
movie_id=movie5.movie_id,
rating=3,
seen="f")
fred_movie6 = UserMovie(user_id=fred.user_id,
movie_id=movie6.movie_id,
rating=9,
seen="t")
db.session.add_all([
fred_music, fred_war, fred_comedy, fred_family, fred_movie1,
fred_movie2, fred_movie3, fred_movie4, fred_movie5, fred_movie6
])
db.session.commit()
