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()