def load_movies():
"""Load movies from u.item into database."""
print "Movies"
Movie.query.delete()
with open("seed_data/u.item") as movie_file:
for row in movie_file:
row = row.rstrip().split("|")
movie_id, title, released_at, misc, imdb_url = row[0:5]
movie_id = int(movie_id)
title = title[:-7]
if released_at:
released_at = datetime.datetime.strptime(
released_at, "%d-%b-%Y")
movie = Movie(movie_id=movie_id,
title=title,
released_at=released_at,
imdb_url=imdb_url)
else:
movie = Movie(movie_id=movie_id,
title=title,
imdb_url=imdb_url)
db.session.add(movie)
db.session.commit()
def load_movies():
"""Load movies from u.item into database."""
for row in open("seed_data/u.item"):
row = row.rstrip()
split_row = row.split('|')
movie_id = split_row[0]
title = split_row[1]
released_at = split_row[2]
imdb_url = split_row[4]
if released_at:
released_at = datetime.datetime.strptime(released_at, "%d-%b-%Y")
else:
released_at = None
if title[-1] == ')': # shaving off date in title
title = title[:-7]
movie = Movie(movie_id=movie_id,
title=title,
released_at=released_at,
imdb_url=imdb_url)
db.session.add(movie)
db.session.commit()
def load_movies():
"""Load movies from u.item into database."""
print("Movies")
Movie.query.delete()
# Read u.movie and insert data
for row in open("seed_data/u.item"):
row = row.rstrip()
movie_id, title, released_str, bleh, imbd_url = row.split("|")[:5]
if released_str:
released_at = datetime.strptime(released_str, "%d-%b-%Y")
else:
released_at = None
movie = Movie(movie_id=movie_id,
title=title[:-6],
released_at=released_at,
imbd_url=imbd_url)
db.session.add(movie)
db.session.commit()
def load_movies():
"""Load movies from u.item into database."""
print "Movies"
Movie.query.delete()
for row in open("seed_data/u.item"):
row = row.rstrip()
movie_id, title, release_str, junk, imdb_url = row.split("|")[:5]
if release_str:
released_at = datetime.strptime(release_str, "%d-%b-%Y")
else:
released_at = None
title = title.split(" (")[0]
movie = Movie(movie_id=movie_id,
title=title,
released_at=released_at,
imdb_url=imdb_url)
db.session.add(movie)
db.session.commit()
def load_movies():
"""Load movies from u.item into database."""
# Prevent double-adding movies
Movie.query.delete()
# Read u.user file and insert data
f = open("seed_data/u.item")
for row in f:
row = row.rstrip()
movie_id, title, released_at, _, imdb_url = row.split("|")[:5]
title = title.split("(")[0].rstrip()
if released_at:
released_at = datetime.strptime(released_at, "%d-%b-%Y")
else:
released_at = None
movie = Movie(movie_id=movie_id,
title=title,
released_at=released_at,
imdb_url=imdb_url)
# We need to add to the session or it won't ever be stored
db.session.add(movie)
# Close file
f.close()
# Once we're done, we should commit our work
db.session.commit()
def load_movies():
"""Load movies from u.item into database."""
print "Movies"
Movie.query.delete()
for row in open("seed_data/u.item"):
row = row.rstrip()
row = row.split("|")
movie_id = row[0]
title = row[1]
released_str = row[2]
imdb_url = row[4]
if title:
title = title[:-7]
title = title.decode("latin-1") # decode from latin-1 to unicode
if released_str:
released_at = datetime.strptime(released_str, '%d-%b-%Y')
else:
released_at = None
movie = Movie(movie_id=movie_id,
title=title,
released_at=released_at,
imdb_url=imdb_url)
db.session.add(movie)
db.session.commit()
def load_movies():
"""Load movies from u.item into database."""
print("Movies")
Movie.query.delete()
for row in open("seed_data/u.item"):
row = row.rstrip()
movie_id, title, released_str, space, imdb_url = row.split("|")[0:5]
title = (title.split(" ")[0:-1])
title = ' '.join(title)
if released_str:
released_at = datetime.datetime.strptime(released_str, "%d-%b-%Y")
else:
released_at = None
# print("DEBUG +++++++++")
# print(movie_id)
# print(title)
# print("END DEBUG +++++++++")
movie = Movie(movie_id=movie_id,
title=title,
released_at=released_at,
imdb_url=imdb_url)
db.session.add(movie)
db.session.commit()
def load_movies():
"""Load movies from u.item into database."""
print("Movies")
Movie.query.delete()
for row in open("seed_data/u.item"):
row = row.rstrip()
items = row.split("|")
movie_id = items[0]
title = items[1]
released_at = items[2]
imdb_url = items[4]
#remove year from title
title = title.split(" (")
title = title[0]
#need to formate released at date
date_format = "%d-%b-%Y"
released_at = datetime.strptime(released_at, date_format)
movie = Movie(movie_id=movie_id,
title=title,
released_at=released_at,
imdb_url=imdb_url)
db.session.add(movie)
db.session.commit()
def load_movies():
"""Load movies from u.item into database."""
print("Movies")
Movie.query.delete()
for row in open("seed_data/u.item"):
row = row.rstrip().split("|")
released_str = row[2]
if released_str:
released_at = datetime.strptime(released_str, "%d-%b-%Y")
else:
released_at = None
movie = Movie(
movie_id=row[0],
title=row[1][:-7],
release_at = released_at,
imdb_url=row[4]
)
db.session.add(movie)
db.session.commit()
def load_movies():
"""Load movies from u.item into database."""
# Delete previous rows in table to avoid duplicates
Movie.query.delete()
# Read u.item file and insert data
for row in open("seed_data/u.item"):
row = row.rstrip().split("|")
movie_id = row[0]
full_title = row[1].split(" ")
full_title.pop()
title = ' '.join(full_title)
#alternative solution title[:-7]
released_at = datetime.strptime(row[2], "%d-%b-%Y")
imdb_url = row[4]
movie = Movie(movie_id=movie_id,
title=title,
released_at=released_at,
imdb_url=imdb_url)
# released_at = datetime.strptime(released_at, '%d-%b-%Y')
#Add movie to session
db.session.add(movie)
#Commit session
db.session.commit()
def load_movies():
"""Load movies from u.item into database."""
print("Movies")
# Delete all rows in table, so if we need to run this a second time,
# we won't be trying to add duplicate users
Movie.query.delete()
# Read u.user file and insert data
for row in open("seed_data/u.item"):
row = row.rstrip()
movie_id, title, date, url, *genre = row.split("|")
# https://stackoverflow.com/questions/6266727/python-cut-off-the-last-word-of-a-sentence
if title:
title = title.rsplit(' ', 1)[0];
else:
title = None
if date:
date = datetime.datetime.strptime(date,"%d-%b-%Y")
else:
date = None
movie = Movie(movie_id=movie_id,
title=title,
released_at=date,
imdb_url=url)
# We need to add to the session or it won't ever be stored
db.session.add(movie)
# Once we're done, we should commit our work
db.session.commit()
def load_movies():
"""Load movies from u.item into database."""
print("Movies")
Movie.query.delete()
for row in open("seed_data/u.item"):
row = row.rstrip()
movie_values = row.split("|")
movie_id = movie_values[0]
title = movie_values[1]
released_at = movie_values[2]
imdb_url = movie_values[4]
title = title[0:-7]
if released_at:
released_at = datetime.strptime(released_at, "%d-%b-%Y")
else:
released_at = None
movie = Movie(movie_id=movie_id,
title=title,
released_at=released_at,
imdb_url=imdb_url)
db.session.add(movie)
db.session.commit()
def load_movies():
"""Load movies from u.item into database."""
print "Movies"
# Delete all rows in table, so if we need to run this a second time,
# we won't be trying to add duplicate users
Movie.query.delete()
# Read u.user file and insert data
for row in open("seed_data/u.item"):
row = row.rstrip()
movie_id, title, released_string, video_release_date, imdb_url = row.split(
"|")[:5]
# day, month, year = released_string.split("-")
if released_string:
released_at = datetime.datetime.strptime(released_string,
"%d-%b-%Y")
else:
released_at = None
clean_title = re.sub(r"( \(\d{4}\))", "", title)
movie = Movie(movie_id=movie_id,
title=clean_title,
released_at=released_at,
imdb_url=imdb_url)
# We need to add to the session or it won't ever be stored
db.session.add(movie)
# Once we're done, we should commit our work
db.session.commit()
def add_movie():
name = request.form.get('name')
description = request.form.get('description')
generalRating = request.form.get('generalRating')
length = request.form.get('length')
src = request.form.get('src')
genres = []
try:
for genre in Genre.query.all():
cb = request.form.get('gcb' + str(genre.id))
if cb is not None:
genres.append(genre)
mov = Movie(name=name,
description=description,
generalRating=generalRating,
length=length,
isFavorite=False,
src=src)
mov.genres = genres
db.session.add(mov)
db.session.commit()
except Exception:
flash('')
return redirect(url_for('movie'))
return redirect(url_for('movie'))
def load_movies():
"""Load movies from u.item into database."""
print "Movies"
Movie.query.delete()
for row in open("seed_data/u.item"):
row = row.rstrip().split("|")
movie_id = row[0]
title = row[1]
# Removes year from movie title
if title[-1] == ")" and title[-6] == "(" and title[-5:-1].isdigit():
title = title[:-7]
if row[2]:
released_at = datetime.strptime(row[2], '%d-%b-%Y')
else:
released_at = None
imdb_url = row[4]
movie = Movie(movie_id=movie_id,
title=title,
released_at=released_at,
imdb_url=imdb_url)
db.session.add(movie)
db.session.commit()
def load_movies():
"""Load movies from u.item into database."""
print "Movies"
Movie.query.delete()
for row in open("seed_data/u.item"):
row = row.rstrip().split("|")
movie_id = row[0]
title = row[1][:-7]
release_date_str = row[2]
imdb_url = row[4]
if release_date_str:
release_date = datetime.strptime(release_date_str, '%d-%b-%Y')
else:
release_date = None
movie = Movie(movie_id=movie_id,
title=title,
release_at=release_date,
imdb_url=imdb_url)
db.session.add(movie)
db.session.commit()
def load_movies():
"""Load movies from u.item into database."""
Movie.query.delete()
for row in open("seed_data/u.item"):
row = row.rstrip().split("|")
(
movie_id,
movie_title,
released_at,
_,
imdb_url,
) = row[:5]
movie_title = movie_title[:-7]
released_at = datetime.strptime(released_at, "%d-%b-%Y")
movie = Movie(movie_id=movie_id,
movie_title=movie_title,
released_at=released_at,
imdb_url=imdb_url)
db.session.add(movie)
db.session.commit()
print("Movies")
def load_movies():
"""Load movies from u.item into database."""
print "Movies"
# Delete all rows in table, so if we need to run this a second time,
# we won't be trying to add duplicate movies
Movie.query.delete()
# Read u.item file and insert data
for row in open("seed_data/u.item"):
row = row.rstrip().split("|")
movie = Movie(
movie_id=row[0],
title=row[1][:-7], # Stripping year e.g. _(1995) from title
released_at=row[2],
imdb_url=row[4])
# If released_at is not an empty string, convert to datetime
if movie.released_at:
movie.released_at = datetime.strptime(row[2], '%d-%b-%Y')
else:
movie.released_at = None
# We need to add to the session or it won't ever be stored
db.session.add(movie)
# Once we're done, we should commit our work
db.session.commit()
def load_movies():
"""Load movies from u.item into database."""
print("Movies")
Movie.query.delete()
for row in open("seed_data/u.item"):
row = row.rstrip()
movie_id, movie_title, release_date, video_release_date, imdb_url,\
unknown, action, adventure, animation,childrens, comedy, crime,\
documentary, drama, fantasy, film_noir, horror, musical, mystery,\
romance, sci_fi, thriller, war, western = row.split("|")
if release_date:
release_date = datetime.strptime(release_date, '%d-%b-%Y')
else:
release_date = None
if video_release_date:
video_release_date = datetime.strptime(video_release_date, '%d-%b-%Y')
else:
video_release_date = None
movie_title = movie_title.rpartition(' (')[0]
movie = Movie(movie_id=movie_id,
movie_title=movie_title,
release_date=release_date,
video_release_date=video_release_date,
imdb_url=imdb_url)
db.session.add(movie)
db.session.commit()
def load_movies():
"""Load movies from u.item into database."""
print("Movies")
Movie.query.delete()
for movie in open("seed_data/u.item"):
movie = movie.rstrip()
movie_info = movie.split("|")
movie_id = movie_info[0]
title = movie_info[1]
title = title.split("(")[0].rstrip()
release_str = movie_info[2]
imdb_url = movie_info[4]
if release_str:
release_at = datetime.strptime(release_str, '%d-%b-%Y')
else:
release_at = None
movie_data = Movie(movie_id=movie_id,
title=title,
release_at=release_at,
imdb_url=imdb_url)
db.session.add(movie_data)
db.session.commit()
def load_movies():
"""Load movies from u.item into database."""
print("Movies")
Movie.query.delete()
for row in open("seed_data/u.item"):
row = row.rstrip()
row = row.split("|")
row = row[:5]
movie_id, title, released_at, video_release, imdb_url = row
if released_at:
released_at = datetime.strptime(released_at, "%d-%b-%Y")
else:
released_at = None
title = title[:-7]
movie = Movie(movie_id=movie_id,
title=title,
released_at=released_at,
imdb_url=imdb_url)
db.session.add(movie)
db.session.commit()
def load_movies():
"""Load movies from u.item into database."""
print "Movies"
Movie.query.delete()
with open("seed_data/u.item") as movie_file:
for row in movie_file:
data = row.rstrip().split("|")
movie_id = data[0]
# Slice removes date at end of movie title
title = data[1][:-7].decode("latin-1")
if data[2]:
released_at = datetime.strptime(data[2], '%d-%b-%Y')
else:
released_at = None
imdb_url = data[4]
movie = Movie(movie_id=movie_id,
title=title,
released_at=released_at,
imdb_url=imdb_url)
db.session.add(movie)
db.session.commit()
def load_movies():
"""Load movies from u.item into database."""
print "movies"
# Delete all rows in table, so if we need to run this a second time,
# we won't be trying to add duplicate movies
Movie.query.delete()
# Read u.item file and insert data
for row in open("seed_data/u.item"):
row = row.rstrip()
row = row[:-38] # removing unneeded info which was genres as 0 and 1s
movie_id, title, released_str, video_release_date, imdb_url = row.split(
"|")
#Formatting release date
if released_str:
released_at = datetime.datetime.strptime(released_str, "%d-%b-%Y")
else:
released_at = None
title = title[:-7] # removing release year
movie = Movie(movie_id=movie_id,
title=title,
released_at=released_at,
imdb_url=imdb_url)
# We need to add to the session or it won't ever be stored
db.session.add(movie)
# Once we're done, we should commit our work
db.session.commit()
def load_movies():
"""Load movies from u.item into database."""
print "Movies"
Movie.query.delete()
for row in open("seed_data/u.item"):
row = row.rstrip()
movie_id, title, released_str, video_release, imdb_url = row.split(
"|")[0:5]
# Format movie_id and released_at
title = title[:-7]
title = title.decode("latin-1")
if released_str:
released_at = datetime.datetime.strptime(released_str, "%d-%b-%Y")
else:
released_at = None
# Assign values to instance attributes
movie = Movie(movie_id=movie_id,
title=title,
released_at=released_at,
imdb_url=imdb_url)
db.session.add(movie)
db.session.commit()
def load_movies():
"""Load movies from u.item into database."""
print("Movies")
Movie.query.delete()
for row in open("seed_data/u.item"):
row = row.rstrip()
split_row = row.split("|")
movie_id = split_row[0]
full_title = split_row[1]
title = full_title[:-7]
release_date_str = split_row[2]
fmat = "%d-%b-%Y"
release_date = datetime.strptime(release_date_str, fmat)
imdb_url = split_row[4]
# match key with model.py file; released_at
movie = Movie(movie_id=movie_id,
title=title,
released_at=release_date,
imdb_url=imdb_url)
db.session.add(movie)
db.session.commit()
def load_movies():
"""Load movies from u.item into database."""
print("Movies")
Movie.query.delete()
for row in open("seed_data/u.item"):
row = row.rstrip()
# row_data = row.split("|")
movie_id, title, release_str, video_release_date, imdb_url, unknown_genre, action, adventure, animation, childrens, comedy, crime, documentary, drama, fantasy, film_noir, horror, musical, mystery, romance, sci_fi, thriller, war, western = row.split(
"|")
stuff = title.split(" ")
title = stuff[:-1]
title = " ".join(title)
# title = title.rstrip()
if release_str:
released_at = datetime.strptime(release_str, "%d-%b-%Y")
else:
released_at = None
movie = Movie(movie_id=movie_id,
title=title,
released_at=released_at,
imdb_url=imdb_url)
db.session.add(movie)
db.session.commit()
def load_movies():
"""Load movies from u.item into database."""
Movie.query.delete()
for i, row in enumerate(open("seed_data/u.item")):
row = row.strip()
movies_id, title, released_at, empty, imdb_url = row.split("|")[:5]
format_date = "%d-%b-%Y"
title = title[:-7]
movie = Movie(movies_id=movies_id,
title=title,
released_at=datetime.strptime(released_at, format_date),
imdb_url=imdb_url)
db.session.add(movie)
# Progress ticker
if i % 100 == 0:
print(i)
db.session.commit()
db.session.commit()
def load_movies():
"""Load movies from u.item into database."""
# row = row.rstrip("(")
#datetime.strptime(date_string, "%d-%b-%Y")
Movie.query.delete()
# Read u.user file and insert data
for row in open("seed_data/u.item"):
row = row.rstrip()
cols = row.split("|")
(movie_id, title, released_at, _, imdb_url) = cols[:5]
title = title[:-7]
released_at = datetime.strptime(released_at, "%d-%b-%Y")
movie = Movie(movie_id=movie_id,
title=title,
released_at=released_at,
imdb_url=imdb_url)
# We need to add to the session or it won't ever be stored
db.session.add(movie)
# Once we're done, we should commit our work
db.session.commit()
def load_movies():
"""Load movies from u.item into database."""
print("Movies")
# Delete all rows in table, so if we need to run this a second time,
# we won't be trying to add duplicate users
Movie.query.delete()
# Read u.user file and insert data
for row in open("seed_data/u.item"):
row = row.rstrip()
row = row.split("|")
title, released_at, imdb_url = row[1:4]
title = title[:-7]
if released_at:
released_at = datetime.strptime(released_at, "%d-%b-%Y")
else:
released_at = None
movie = Movie(title=title,
released_at=released_at,
imdb_url=imdb_url)
# We need to add to the session or it won't ever be stored
db.session.add(movie)
# Once we're done, we should commit our work
db.session.commit()
def load_movies(movie_filename):
"""Load movies from u.item into database."""
print("Movies")
for i, row in enumerate(open(movie_filename)):
row = row.rstrip()
# clever -- we can unpack part of the row!
movie_id, title, released_str, junk, imdb_url = row.split("|")[:5]
# The date is in the file as daynum-month_abbreviation-year;
# we need to convert it to an actual datetime object.
if released_str:
released_at = datetime.datetime.strptime(released_str, "%d-%b-%Y")
else:
released_at = None
# Remove the (YEAR) from the end of the title.
title = title[:-7] # " (YEAR)" == 7
movie = Movie(title=title, released_at=released_at, imdb_url=imdb_url)
# We need to add to the session or it won't ever be stored
db.session.add(movie)
# provide some sense of progress
if i % 100 == 0:
print(i)
# Once we're done, we should commit our work
db.session.commit()
