def cart():
model = ShoppingCart(current_user.get_id())
videos = model.get_items()
if videos:
db.execute("SELECT title,dvd_price FROM video WHERE video_id IN (%s)" %
(",".join(str(i) for i in videos), ))
videos_info = db.fetchall()
else:
videos_info = []
items = []
for video_id, video_info in zip(videos, videos_info):
items.append({
"video_id": video_id,
"title": video_info[0],
"dvd_price": video_info[1]
})
app.logger.debug("Query returned\n%s", repr(items))
if flask.request.method == "POST":
action = flask.request.form.get("action")
app.logger.info("Cart action %s triggered.", action)
if action == "checkout":
db.execute("SELECT balance FROM users WHERE user_id=%s",
(current_user.get_id(), ))
balance = int(db.fetchone()[0])
price_sum = sum(i["dvd_price"] for i in items)
if price_sum > balance:
flask.flash("You don't have enough money in your account.",
category="error")
else:
db.execute(
'UPDATE users SET balance = balance - %s WHERE user_id = %s',
(price_sum, current_user.get_id()))
model.complete_order()
flask.flash("Your order has been placed!", category="message")
return flask.redirect(flask.url_for("cart"))
elif action == "clear":
model.clear_cart()
flask.flash("Cart has been cleared.", category="message")
return flask.redirect(flask.url_for("cart"))
return flask.render_template("cart.html", items=items)
def cart():
model = ShoppingCart(current_user.get_id())
videos = model.get_items()
if videos:
db.execute(
"SELECT title,dvd_price FROM video WHERE video_id IN (%s)" %
(",".join(str(i) for i in videos), )
)
videos_info = db.fetchall()
else:
videos_info = []
items = []
for video_id, video_info in zip(videos, videos_info):
items.append({
"video_id": video_id,
"title": video_info[0],
"dvd_price": video_info[1]
})
app.logger.debug("Query returned\n%s", repr(items))
if flask.request.method == "POST":
action = flask.request.form.get("action")
app.logger.info("Cart action %s triggered.", action)
if action == "checkout":
db.execute("SELECT balance FROM users WHERE user_id=%s", (current_user.get_id(), ))
balance = int(db.fetchone()[0])
price_sum = sum(i["dvd_price"] for i in items)
if price_sum > balance:
flask.flash("You don't have enough money in your account.", category = "error")
else:
db.execute('UPDATE users SET balance = balance - %s WHERE user_id = %s', (price_sum, current_user.get_id()))
model.complete_order()
flask.flash("Your order has been placed!", category = "message")
return flask.redirect(flask.url_for("cart"))
elif action == "clear":
model.clear_cart()
flask.flash("Cart has been cleared.", category = "message")
return flask.redirect(flask.url_for("cart"))
return flask.render_template("cart.html", items = items)
def movie_info(movie_id):
if flask.request.method == 'POST' and current_user:
if flask.request.form['hiddinput'] == 'online':
# get current balance
db.execute('SELECT balance FROM users WHERE user_id = %s', (current_user.get_id()))
balance = db.fetchall()[0][0]
# get online price
db.execute('SELECT online_price FROM video WHERE video_id = %s', (movie_id))
price = db.fetchall()[0][0]
if balance < price:
flask.flash('You do not have enough money to watch this movie online', 'error')
return flask.redirect(flask.url_for('movie_info', movie_id = movie_id))
db.execute('INSERT INTO watched (user_id, video_id) VALUES (%s, %s)', (current_user.get_id(), movie_id))
db.execute('UPDATE users, video SET users.balance = (users.balance - video.online_price) WHERE users.user_id = %s AND video.video_id = %s', (current_user.get_id(), movie_id))
elif flask.request.form['hiddinput'] == 'dvd':
cart = ShoppingCart(current_user.get_id())
try:
cart.add_item(movie_id)
except:
flask.flash("Movie already in cart.", category = "error")
elif flask.request.form['hiddinput'] == 'fave_up':
db.execute('INSERT INTO likes (user_id, video_id) VALUES (%s, %s)', (current_user.get_id(), movie_id))
elif flask.request.form['hiddinput'] == 'fave_down':
db.execute('DELETE FROM likes WHERE user_id = %s AND video_id = %s', (current_user.get_id(), movie_id))
elif flask.request.form['hiddinput'] == 'comment':
db.execute('INSERT INTO comment (user_id, video_id, content) VALUES (%s, %s, %s)', (current_user.get_id(), movie_id, flask.request.form['hidden_comment']))
else:
# user entered new rating
new_rating = flask.request.form['hiddinput']
if new_rating != 'None':
db.execute('INSERT INTO rate (user_id, video_id, rating) VALUES (%s, %s, %s) ON DUPLICATE KEY UPDATE rating = %s', (current_user.get_id(), movie_id, new_rating, new_rating))
# get relevant info from db
db.execute('SELECT * FROM video WHERE video_id=%s', (movie_id))
results = db.fetchall()
# make sure movie exists
if len(results) != 1:
flask.flash('Movie does not exist', category = 'error')
return flask.redirect(flask.url_for('browse'))
video = results[0]
info = {}
info['title'] = video[1]
info['year'] = video[2]
info['online_price'] = video[3]
info['dvd_price'] = video[4]
info['votes'] = video[5]
info['rating'] = video[6]
# get genre information from database
db.execute('SELECT genre_name FROM genre')
genres = [genre[0] for genre in db.fetchall()]
db.execute('SELECT genre_id FROM categorize WHERE video_id = %s', (movie_id))
genre_string = ''
for genre in db.fetchall():
genre_string += genres[genre[0] - 1] + ', '
info['genre'] = genre_string[0:-2]
# get watch count from db
db.execute('SELECT COUNT(*) FROM watched WHERE video_id = %s', (movie_id))
info['watch_count'] = db.fetchall()[0][0]
# get stars from db
db.execute('SELECT first_name, last_name FROM star')
stars = [star[0] + ' ' + star[1] for star in db.fetchall()]
db.execute('SELECT star_id FROM played WHERE video_id = %s', (movie_id))
video_stars = []
for star in db.fetchall():
video_stars.append(stars[star[0] - 1])
info['stars'] = video_stars
# get writers from db
db.execute('SELECT first_name, last_name FROM author')
authors = [author[0] + ' ' + author[1] for author in db.fetchall()]
db.execute('SELECT author_id FROM written WHERE video_id = %s', (movie_id))
author_string = ''
for author in db.fetchall():
author_string += authors[author[0] - 1] + ', '
info['authors'] = author_string[0:-2]
# check if user likes this video already
db.execute('SELECT COUNT(*) FROM likes WHERE user_id = %s AND video_id = %s', (current_user.get_id(), movie_id))
info['likes'] = db.fetchall()[0][0] > 0
# get rating information from db
db.execute('SELECT AVG(rate.rating) FROM rate, users WHERE rate.video_id = %s AND rate.user_id = users.user_id AND users.rank_perm = 2', (movie_id))
info['overall_rating'] = db.fetchall()[0][0]
db.execute('SELECT AVG(rate.rating) FROM rate, follow, users WHERE follow.user_id_from = %s AND rate.user_id = follow.user_id_to AND rate.video_id = %s AND rate.user_id = users.user_id AND users.rank_perm >= 1', (current_user.get_id(), movie_id))
info['follower_rating'] = db.fetchall()[0][0]
db.execute('SELECT rating FROM rate WHERE user_id = %s AND video_id = %s', (current_user.get_id(), movie_id))
results = db.fetchall()
user_rating = results[0][0] if len(results) > 0 else None
if user_rating:
info['user_rating'] = int(user_rating)
else:
info['user_rating'] = 0
# check for TV series info
if video[7]:
info['episode'] = video[7]
db.execute('''SELECT season.season_number, series.title
FROM season, series
WHERE season.season_id = %s AND series.series_id = season.series_id''', (video[8]))
results = db.fetchall()[0]
info['season'] = results[0]
info['series'] = results[1]
# get comments from db
db.execute('SELECT users.first_name, users.last_name, comment.comment_time, comment.content FROM comment, users WHERE video_id = %s AND comment.user_id = users.user_id', (movie_id))
results = db.fetchall()
comments = []
for comment in results:
comments.append({ 'name' : comment[0] + ' ' + comment[1],
'time' : comment[2],
'content' : comment[3] })
return flask.render_template('movie_info.html', movie_id = movie_id, info = info, comments = comments)
def movie_info(movie_id):
if flask.request.method == 'POST' and current_user:
if flask.request.form['hiddinput'] == 'online':
# get current balance
db.execute('SELECT balance FROM users WHERE user_id = %s',
(current_user.get_id()))
balance = db.fetchall()[0][0]
# get online price
db.execute('SELECT online_price FROM video WHERE video_id = %s',
(movie_id))
price = db.fetchall()[0][0]
if balance < price:
flask.flash(
'You do not have enough money to watch this movie online',
'error')
return flask.redirect(
flask.url_for('movie_info', movie_id=movie_id))
db.execute(
'INSERT INTO watched (user_id, video_id) VALUES (%s, %s)',
(current_user.get_id(), movie_id))
db.execute(
'UPDATE users, video SET users.balance = (users.balance - video.online_price) WHERE users.user_id = %s AND video.video_id = %s',
(current_user.get_id(), movie_id))
elif flask.request.form['hiddinput'] == 'dvd':
cart = ShoppingCart(current_user.get_id())
try:
cart.add_item(movie_id)
except:
flask.flash("Movie already in cart.", category="error")
elif flask.request.form['hiddinput'] == 'fave_up':
db.execute('INSERT INTO likes (user_id, video_id) VALUES (%s, %s)',
(current_user.get_id(), movie_id))
elif flask.request.form['hiddinput'] == 'fave_down':
db.execute(
'DELETE FROM likes WHERE user_id = %s AND video_id = %s',
(current_user.get_id(), movie_id))
elif flask.request.form['hiddinput'] == 'comment':
db.execute(
'INSERT INTO comment (user_id, video_id, content) VALUES (%s, %s, %s)',
(current_user.get_id(), movie_id,
flask.request.form['hidden_comment']))
else:
# user entered new rating
new_rating = flask.request.form['hiddinput']
if new_rating != 'None':
db.execute(
'INSERT INTO rate (user_id, video_id, rating) VALUES (%s, %s, %s) ON DUPLICATE KEY UPDATE rating = %s',
(current_user.get_id(), movie_id, new_rating, new_rating))
# get relevant info from db
db.execute('SELECT * FROM video WHERE video_id=%s', (movie_id))
results = db.fetchall()
# make sure movie exists
if len(results) != 1:
flask.flash('Movie does not exist', category='error')
return flask.redirect(flask.url_for('browse'))
video = results[0]
info = {}
info['title'] = video[1]
info['year'] = video[2]
info['online_price'] = video[3]
info['dvd_price'] = video[4]
info['votes'] = video[5]
info['rating'] = video[6]
# get genre information from database
db.execute('SELECT genre_name FROM genre')
genres = [genre[0] for genre in db.fetchall()]
db.execute('SELECT genre_id FROM categorize WHERE video_id = %s',
(movie_id))
genre_string = ''
for genre in db.fetchall():
genre_string += genres[genre[0] - 1] + ', '
info['genre'] = genre_string[0:-2]
# get watch count from db
db.execute('SELECT COUNT(*) FROM watched WHERE video_id = %s', (movie_id))
info['watch_count'] = db.fetchall()[0][0]
# get stars from db
db.execute('SELECT first_name, last_name FROM star')
stars = [star[0] + ' ' + star[1] for star in db.fetchall()]
db.execute('SELECT star_id FROM played WHERE video_id = %s', (movie_id))
video_stars = []
for star in db.fetchall():
video_stars.append(stars[star[0] - 1])
info['stars'] = video_stars
# get writers from db
db.execute('SELECT first_name, last_name FROM author')
authors = [author[0] + ' ' + author[1] for author in db.fetchall()]
db.execute('SELECT author_id FROM written WHERE video_id = %s', (movie_id))
author_string = ''
for author in db.fetchall():
author_string += authors[author[0] - 1] + ', '
info['authors'] = author_string[0:-2]
# check if user likes this video already
db.execute(
'SELECT COUNT(*) FROM likes WHERE user_id = %s AND video_id = %s',
(current_user.get_id(), movie_id))
info['likes'] = db.fetchall()[0][0] > 0
# get rating information from db
db.execute(
'SELECT AVG(rate.rating) FROM rate, users WHERE rate.video_id = %s AND rate.user_id = users.user_id AND users.rank_perm = 2',
(movie_id))
info['overall_rating'] = db.fetchall()[0][0]
db.execute(
'SELECT AVG(rate.rating) FROM rate, follow, users WHERE follow.user_id_from = %s AND rate.user_id = follow.user_id_to AND rate.video_id = %s AND rate.user_id = users.user_id AND users.rank_perm >= 1',
(current_user.get_id(), movie_id))
info['follower_rating'] = db.fetchall()[0][0]
db.execute('SELECT rating FROM rate WHERE user_id = %s AND video_id = %s',
(current_user.get_id(), movie_id))
results = db.fetchall()
user_rating = results[0][0] if len(results) > 0 else None
if user_rating:
info['user_rating'] = int(user_rating)
else:
info['user_rating'] = 0
# check for TV series info
if video[7]:
info['episode'] = video[7]
db.execute(
'''SELECT season.season_number, series.title
FROM season, series
WHERE season.season_id = %s AND series.series_id = season.series_id''',
(video[8]))
results = db.fetchall()[0]
info['season'] = results[0]
info['series'] = results[1]
# get comments from db
db.execute(
'SELECT users.first_name, users.last_name, comment.comment_time, comment.content FROM comment, users WHERE video_id = %s AND comment.user_id = users.user_id',
(movie_id))
results = db.fetchall()
comments = []
for comment in results:
comments.append({
'name': comment[0] + ' ' + comment[1],
'time': comment[2],
'content': comment[3]
})
return flask.render_template('movie_info.html',
movie_id=movie_id,
info=info,
comments=comments)