def song2(track_id):
try:
# Establish connection
c, conn = connection()
#song = {}
result = c.execute("SELECT * FROM tracks WHERE track_id = (%s)" %
track_id)
track = c.fetchone()
#track = c.fetchone()
id = track['track_id']
track['track_id'] = int(id)
result = c.execute(
"SELECT t.track_id, t.track_no, t.title, a.album_id, a.title, a.genre, a.year_released, a.artwork, ar.artist_id, ar.name FROM tracks T JOIN albums a ON t.album_id=a.album_id JOIN artists ar ON a.artist_id=ar.artist_id WHERE t.album_id=(%s) ORDER BY t.track_no",
[track['album_id']])
album = c.fetchall()
# Close Connection
conn.commit()
c.close()
conn.close()
#gc.collect()
track['album'] = album
return jsonify({'track': track})
except Exception as e:
return (str(e))
def addSong(artist_id, album_id):
data = request.get_json(force=True)
if userAuth(data['username'], data['password']) == False:
print "sup"
return jsonify({'error': True, 'message': 'You are not logged in!'} )
try:
song = {}
song['album_id'] = str(album_id)
#song['album_id'] = data[1]
song['track_number'] = str(data['track_number'])
song['title'] = data['title'].encode('utf-8')
song['lyrics'] = data['lyrics'].encode('utf-8')
song['media_link'] = data['media_link'].encode('utf-8')
c, conn = connection()
c.execute("INSERT INTO tracks (album_id, track_no, title, lyrics, media_link) VALUES (%s, %s, %s, %s, %s);", [sanitize(song['album_id']), sanitize(song['track_number']), sanitize(song['title']), song['lyrics'], sanitize(song['media_link'])])
c.execute("SELECT LAST_INSERT_ID();")
data = c.fetchone()
# Close Connection
conn.commit()
c.close()
conn.close()
print data
return jsonify({"error": False, "message": "Song added!", "track_id": data['LAST_INSERT_ID()']})
except Exception as e:
print "err: " + str(e)
return jsonify({'error': True, 'message': 'Something went wrong... Content may not have saved.'} )
def editSong(artist_id, album_id, track_id):
data = request.get_json(force=True)
if userAuth(data['username'], data['password']) == False:
print "sup"
return jsonify({'error': True, 'message': 'You are not logged in!'} )
try:
song = {}
song['id'] = str(track_id)
#song['album_id'] = data[1]
song['track_number'] = str(data['track_number'])
song['title'] = data['title'].encode('utf-8')
song['lyrics'] = data['lyrics'].encode('utf-8')
song['media_link'] = data['media_link'].encode('utf-8')
c, conn = connection()
c.execute("UPDATE tracks SET title = (%s), track_no = (%s), lyrics = (%s), media_link = (%s) WHERE track_id = (%s);", [sanitize(song['title']), sanitize(song['track_number']), song['lyrics'], sanitize(song['media_link']),sanitize(song['id'])])
# Close Connection
conn.commit()
c.close()
conn.close()
return jsonify({'error': False, 'message': 'Save successful!'} )
except Exception as e:
print "err: " + str(e)
return jsonify({'error': True, 'message': 'Something went wrong... Content may not have saved.'} )
def register():
data = request.get_json(force=True)
email = data['email']
username = data['username']
password = data['password']
password2 = data['confirmPassword']
try:
password = sha256_crypt.encrypt(str(password))
# Check if username is available
c, conn = connection()
db_username = c.execute("SELECT * FROM users WHERE username = (%s)", [sanitize(username)])
if int(db_username) > 0:
return jsonify({ "userdata": False, "error": "Username Taken." })
# Check if email is already used.
db_email = c.execute("SELECT * FROM users WHERE email = (%s)", [sanitize(email)])
if int(db_email) > 0:
return jsonify({ "userdata": False, "error": "Email already used by another account." })
# Credentials are good! Add new user and close the connection.
c.execute("INSERT INTO users (username, password, email) VALUES (%s, %s, %s)",
[sanitize(username), sanitize(password), sanitize(email)])
conn.commit()
c.close()
conn.close()
return jsonify({ "userdata": True, "message": "You are now registered!" })
except Exception as e:
return(str(e))
def userAuth(attempted_username, attempted_password):
try:
#check username and password with db
c, conn = connection()
res = c.execute("SELECT * FROM users WHERE username = (%s)",
[attempted_username])
user = c.fetchone()
conn.commit()
c.close()
conn.close()
if int(res) > 0:
db_password = user['password']
if sha256_crypt.verify(attempted_password, db_password):
id = user['user_id']
user['user_id'] = int(id)
user['password'] = attempted_password
user['logged_in'] = True
return user
return False
else:
return False
except Exception as e:
return False
def add_Artist():
data = request.get_json(force=True)
if userAuth(data['username'], data['password']) == False:
return jsonify({'error': True, 'message': 'You are not logged in!'} )
print 1
try:
artist = {}
artist['name'] = data['name']
artist['bio'] = data['bio'].encode('utf-8')
artist['genre'] = data['genre']
artist['country'] = data['country']
artist['year'] = str(data['year'])
print 2
c, conn = connection()
c.execute("INSERT INTO artists(name, biography, genre, country, year_formed) VALUES ((%s), (%s), (%s), (%s), (%s))", [sanitize(artist['name']), artist['bio'], sanitize(artist['genre']), sanitize(artist['country']),sanitize(artist['year'])])
c.execute("SELECT LAST_INSERT_ID();")
data = c.fetchone()
# Close Connection
conn.commit()
c.close()
conn.close()
print 3
return jsonify({'error': False, 'message': 'Updated artist!', "artist_id": data['LAST_INSERT_ID()']})
except Exception as e:
return(str(e))
def addAlbum(artist_id):
data = request.get_json(force=True)
if userAuth(data['username'], data['password']) == False:
return jsonify({'error': True, 'message': 'You are not logged in!'} )
print 1
try:
album = {}
print 2
album['artist_id'] = str(artist_id)
album['title'] = data['title']
album['genre'] = data['genre']
album['year'] = str(data['year'])
print 3
c, conn = connection()
c.execute("INSERT INTO albums ( artist_id, title, genre, year_released) VALUES(%s, %s, %s, %s)", [sanitize(album['artist_id']), sanitize(album['title']), sanitize(album['genre']), sanitize(album['year'])])
# Close Connection
conn.commit()
c.close()
conn.close()
print 4
return jsonify({'error': False, "message": "Album added!"})
except Exception as e:
return(str(e))
def album(album_id):
try:
# Establish connection
c, conn = connection()
result = c.execute(
"SELECT a.album_id, a.title, a.genre, a.year_released, a.artwork, ar.artist_id, ar.name FROM albums a JOIN artists ar ON a.artist_id=ar.artist_id WHERE a.album_id=(%s)",
[album_id])
album = c.fetchone()
id = album['album_id']
album['album_id'] = int(id)
result = c.execute(
"SELECT track_id, title, track_no FROM tracks WHERE album_id = (%s) ORDER BY track_no;",
[album['album_id']])
tracklist = c.fetchall()
album['tracklist'] = tracklist
# Close Connection
conn.commit()
c.close()
conn.close()
return jsonify({'album': album})
except Exception as e:
return (str(e))
def song2(track_id):
try:
# Establish connection
c, conn = connection()
#song = {}
result = c.execute("SELECT * FROM tracks WHERE track_id = (%s)" % track_id)
track = c.fetchone()
#track = c.fetchone()
id = track['track_id']
track['track_id'] = int(id)
result = c.execute("SELECT t.track_id, t.track_no, t.title, a.album_id, a.title, a.genre, a.year_released, a.artwork, ar.artist_id, ar.name FROM tracks T JOIN albums a ON t.album_id=a.album_id JOIN artists ar ON a.artist_id=ar.artist_id WHERE t.album_id=(%s) ORDER BY t.track_no", [track['album_id']])
album = c.fetchall()
# Close Connection
conn.commit()
c.close()
conn.close()
#gc.collect()
track['album'] = album
return jsonify({ 'track': track })
except Exception as e:
return(str(e))
def search():
search = request.args.get('search')
try:
title = "Search"
query = "%" + search + "%"
# Establish connection
c, conn = connection()
result = c.execute("SELECT t.track_id, t.title, a.album_id, a.title, a.artwork, ar.artist_id, ar.name FROM tracks t, albums a, artists ar WHERE (t.title LIKE (%s) OR t.lyrics LIKE (%s)) AND t.album_id = a.album_id AND a.artist_id = ar.artist_id ORDER BY t.title;", [sanitize(query), sanitize(query)])
songResults = c.fetchall()
result = c.execute("SELECT a.album_id, a.title, a.genre, a.year_released, a.artwork, ar.artist_id, ar.name FROM albums a, artists ar WHERE (a.title LIKE (%s) OR a.genre LIKE (%s) OR a.year_released LIKE (%s)) AND a.artist_id = ar.artist_id;", [sanitize(query), sanitize(query), sanitize(query)])
albumResults = c.fetchall()
result = c.execute("SELECT ar.artist_id, ar.name, ar.genre, ar.country, a.artwork FROM artists ar LEFT JOIN albums a ON a.artist_id = ar.artist_id WHERE (name LIKE (%s) OR ar.genre LIKE (%s) OR country LIKE (%s)) GROUP BY ar.artist_id;", [sanitize(query), sanitize(query), sanitize(query)])
artistResults = c.fetchall()
conn.commit()
c.close()
conn.close()
return jsonify({ "tracks": songResults, "albums": albumResults, "artists": artistResults })
except Exception as e:
return str(e)
def addAlbum(artist_id):
data = request.get_json(force=True)
if userAuth(data['username'], data['password']) == False:
return jsonify({'error': True, 'message': 'You are not logged in!'})
print 1
try:
album = {}
print 2
album['artist_id'] = str(artist_id)
album['title'] = data['title']
album['genre'] = data['genre']
album['year'] = str(data['year'])
print 3
c, conn = connection()
c.execute(
"INSERT INTO albums ( artist_id, title, genre, year_released) VALUES(%s, %s, %s, %s)",
[
sanitize(album['artist_id']),
sanitize(album['title']),
sanitize(album['genre']),
sanitize(album['year'])
])
# Close Connection
conn.commit()
c.close()
conn.close()
print 4
return jsonify({'error': False, "message": "Album added!"})
except Exception as e:
return (str(e))
def editArtist(artist_id):
data = request.get_json(force=True)
if userAuth(data['username'], data['password']) == False:
return jsonify({'error': True, 'message': 'You are not logged in!'} )
try:
artist = {}
artist['id'] = str(artist_id)
artist['name'] = data['name']
artist['bio'] = data['bio'].encode('utf-8')
artist['genre'] = data['genre']
artist['country'] = data['country']
artist['year'] = str(data['year'])
c, conn = connection()
c.execute("UPDATE artists SET name = (%s), biography = (%s), genre = (%s), country = (%s), year_formed = (%s) WHERE artist_id = (%s)", [sanitize(artist['name']), artist['bio'], sanitize(artist['genre']), sanitize(artist['country']),sanitize(artist['year']), sanitize(artist['id'])])
# Close Connection
conn.commit()
c.close()
conn.close()
return jsonify({'error': False, 'message': 'Updated artist!'})
except Exception as e:
return(str(e))
def artist(artist_id):
try:
# Establish connection
c, conn = connection()
result = c.execute("SELECT * FROM artists WHERE artist_id = (%s);" % artist_id)
artist = c.fetchone()
id = artist['artist_id']
artist['artist_id'] = int(id)
result = c.execute("SELECT name, role FROM members WHERE artist_id = (%s)", [artist['artist_id']])
members = c.fetchall()
result = c.execute("SELECT * FROM albums WHERE artist_id = (%s) ORDER BY year_released DESC;", [artist_id])
albums = c.fetchall()
for album in albums:
result = c.execute("SELECT track_id, title, track_no FROM tracks WHERE album_id = (%s) ORDER BY track_no;", [album['album_id']])
tracklist = c.fetchall()
album['tracklist'] = tracklist
artist['members'] = members
artist['albums'] = albums
c.close()
conn.close()
return jsonify({'artist': artist})
except:
abort(404)
def album(album_id):
try:
# Establish connection
c, conn = connection()
result = c.execute("SELECT a.album_id, a.title, a.genre, a.year_released, a.artwork, ar.artist_id, ar.name FROM albums a JOIN artists ar ON a.artist_id=ar.artist_id WHERE a.album_id=(%s)", [album_id])
album = c.fetchone()
id = album['album_id']
album['album_id'] = int(id)
result = c.execute("SELECT track_id, title, track_no FROM tracks WHERE album_id = (%s) ORDER BY track_no;", [album['album_id']])
tracklist = c.fetchall()
album['tracklist'] = tracklist
# Close Connection
conn.commit()
c.close()
conn.close()
return jsonify({ 'album': album })
except Exception as e:
return(str(e))
def addAlbumToArtist(artist_id):
try:
title = "Add Album"
album = {}
if request.method == "POST":
# TODO modify to recieve JSON data.
album['title'] = request.form['title']
#album['artwork'] = request.form['artwork']
album['genre'] = request.form['genre']
album['year'] = request.form['year']
file = request.files['artwork']
title = album['title']
c, conn = connection()
if file and allowed_file(file.filename):
image = secure_filename("album_img" + str(album['id']) + "." + file.filename.rsplit('.', 1)[1])
file.save(os.path.join(app.config['UPLOAD_FOLDER'], image))
c.execute("INSERT INTO albums ( artist_id, title, genre, year_released, artwork) VALUES(%s, %s, %s, %s, %s)", [sanitize(album['artist_id']), sanitize(album['title']), sanitize(album['genre']), sanitize(album['year']), sanitize(image)])
else:
c.execute("INSERT INTO albums ( artist_id, title, genre, year_released) VALUES(%s, %s, %s, %s)", [sanitize(album['artist_id']), sanitize(album['title']), sanitize(album['genre']), sanitize(album['year'])])
# Close Connection
conn.commit()
c.close()
conn.close()
#gc.collect()
return redirect('/artist/?artist=' + str(album['artist_id']) + "#album" + str(album['id']))
elif request.method == "GET":
title = "Add Album"
album = {}
artist = {}
# ------------------------------
return render_template('pages/edit-album.html', title = title, album = album, artist = artist)
else:
return render_template('err/404.html', title = title)
except Exception as e:
return(str(e))
def addSong(artist_id, album_id):
data = request.get_json(force=True)
if userAuth(data['username'], data['password']) == False:
print "sup"
return jsonify({'error': True, 'message': 'You are not logged in!'})
try:
song = {}
song['album_id'] = str(album_id)
#song['album_id'] = data[1]
song['track_number'] = str(data['track_number'])
song['title'] = data['title'].encode('utf-8')
song['lyrics'] = data['lyrics'].encode('utf-8')
song['media_link'] = data['media_link'].encode('utf-8')
c, conn = connection()
c.execute(
"INSERT INTO tracks (album_id, track_no, title, lyrics, media_link) VALUES (%s, %s, %s, %s, %s);",
[
sanitize(song['album_id']),
sanitize(song['track_number']),
sanitize(song['title']), song['lyrics'],
sanitize(song['media_link'])
])
c.execute("SELECT LAST_INSERT_ID();")
data = c.fetchone()
# Close Connection
conn.commit()
c.close()
conn.close()
print data
return jsonify({
"error": False,
"message": "Song added!",
"track_id": data['LAST_INSERT_ID()']
})
except Exception as e:
print "err: " + str(e)
return jsonify({
'error':
True,
'message':
'Something went wrong... Content may not have saved.'
})
def editAlbum(artist_id, album_id):
data = request.get_json(force=True)
if userAuth(data['username'], data['password']) == False:
return jsonify({'error': True, 'message': 'You are not logged in!'})
print 1
try:
album = {}
print 2
album['id'] = str(album_id)
album['artist'] = str(artist_id)
album['title'] = data['title']
#album['artwork'] = request.form['artwork']
album['genre'] = data['genre']
album['year'] = str(data['year'])
print 3
c, conn = connection()
'''
file = request.files['artwork']
if file and allowed_file(file.filename):
image = secure_filename("album_img" + str(album['id']) + "." + file.filename.rsplit('.', 1)[1])
file.save(os.path.join(app.config['UPLOAD_FOLDER'], image))
c.execute("UPDATE albums SET artwork = (%s) WHERE album_id = (%s)", [sanitize(image), sanitize(album['id'])])
'''
print 4
c.execute(
"UPDATE albums SET title = (%s), genre = (%s), year_released = (%s) WHERE album_id = (%s)",
[
sanitize(album['title']),
sanitize(album['genre']),
sanitize(album['year']),
sanitize(album['id'])
])
# Close Connection
conn.commit()
c.close()
conn.close()
print 5
return jsonify({"error": False, "message": "Album Updated!"})
except Exception as e:
return (str(e))
def register():
data = request.get_json(force=True)
email = data['email']
username = data['username']
password = data['password']
password2 = data['confirmPassword']
try:
password = sha256_crypt.encrypt(str(password))
# Check if username is available
c, conn = connection()
db_username = c.execute("SELECT * FROM users WHERE username = (%s)",
[sanitize(username)])
if int(db_username) > 0:
return jsonify({"userdata": False, "error": "Username Taken."})
# Check if email is already used.
db_email = c.execute("SELECT * FROM users WHERE email = (%s)",
[sanitize(email)])
if int(db_email) > 0:
return jsonify({
"userdata": False,
"error": "Email already used by another account."
})
# Credentials are good! Add new user and close the connection.
c.execute(
"INSERT INTO users (username, password, email) VALUES (%s, %s, %s)",
[sanitize(username),
sanitize(password),
sanitize(email)])
conn.commit()
c.close()
conn.close()
return jsonify({
"userdata": True,
"message": "You are now registered!"
})
except Exception as e:
return (str(e))
def all_artists():
try:
# Establish connection
c, conn = connection()
result = c.execute("SELECT * FROM artists;")
data = c.fetchall()
for row in data:
id = row['artist_id']
row['artist_id'] = int(id)
c.close()
conn.close()
return jsonify({'artists': data})
except:
return (str(e))
def search():
search = request.args.get('search')
try:
title = "Search"
query = "%" + search + "%"
# Establish connection
c, conn = connection()
result = c.execute(
"SELECT t.track_id, t.title, a.album_id, a.title, a.artwork, ar.artist_id, ar.name FROM tracks t, albums a, artists ar WHERE (t.title LIKE (%s) OR t.lyrics LIKE (%s)) AND t.album_id = a.album_id AND a.artist_id = ar.artist_id ORDER BY t.title;",
[sanitize(query), sanitize(query)])
songResults = c.fetchall()
result = c.execute(
"SELECT a.album_id, a.title, a.genre, a.year_released, a.artwork, ar.artist_id, ar.name FROM albums a, artists ar WHERE (a.title LIKE (%s) OR a.genre LIKE (%s) OR a.year_released LIKE (%s)) AND a.artist_id = ar.artist_id;",
[sanitize(query),
sanitize(query),
sanitize(query)])
albumResults = c.fetchall()
result = c.execute(
"SELECT ar.artist_id, ar.name, ar.genre, ar.country, a.artwork FROM artists ar LEFT JOIN albums a ON a.artist_id = ar.artist_id WHERE (name LIKE (%s) OR ar.genre LIKE (%s) OR country LIKE (%s)) GROUP BY ar.artist_id;",
[sanitize(query),
sanitize(query),
sanitize(query)])
artistResults = c.fetchall()
conn.commit()
c.close()
conn.close()
return jsonify({
"tracks": songResults,
"albums": albumResults,
"artists": artistResults
})
except Exception as e:
return str(e)
def editSong(artist_id, album_id, track_id):
data = request.get_json(force=True)
if userAuth(data['username'], data['password']) == False:
print "sup"
return jsonify({'error': True, 'message': 'You are not logged in!'})
try:
song = {}
song['id'] = str(track_id)
#song['album_id'] = data[1]
song['track_number'] = str(data['track_number'])
song['title'] = data['title'].encode('utf-8')
song['lyrics'] = data['lyrics'].encode('utf-8')
song['media_link'] = data['media_link'].encode('utf-8')
c, conn = connection()
c.execute(
"UPDATE tracks SET title = (%s), track_no = (%s), lyrics = (%s), media_link = (%s) WHERE track_id = (%s);",
[
sanitize(song['title']),
sanitize(song['track_number']), song['lyrics'],
sanitize(song['media_link']),
sanitize(song['id'])
])
# Close Connection
conn.commit()
c.close()
conn.close()
return jsonify({'error': False, 'message': 'Save successful!'})
except Exception as e:
print "err: " + str(e)
return jsonify({
'error':
True,
'message':
'Something went wrong... Content may not have saved.'
})
def all_artists():
try:
# Establish connection
c, conn = connection()
result = c.execute("SELECT * FROM artists;")
data = c.fetchall()
for row in data:
id = row['artist_id']
row['artist_id'] = int(id)
c.close()
conn.close()
return jsonify({'artists': data})
except:
return(str(e))
def add_Artist():
data = request.get_json(force=True)
if userAuth(data['username'], data['password']) == False:
return jsonify({'error': True, 'message': 'You are not logged in!'})
print 1
try:
artist = {}
artist['name'] = data['name']
artist['bio'] = data['bio'].encode('utf-8')
artist['genre'] = data['genre']
artist['country'] = data['country']
artist['year'] = str(data['year'])
print 2
c, conn = connection()
c.execute(
"INSERT INTO artists(name, biography, genre, country, year_formed) VALUES ((%s), (%s), (%s), (%s), (%s))",
[
sanitize(artist['name']), artist['bio'],
sanitize(artist['genre']),
sanitize(artist['country']),
sanitize(artist['year'])
])
c.execute("SELECT LAST_INSERT_ID();")
data = c.fetchone()
# Close Connection
conn.commit()
c.close()
conn.close()
print 3
return jsonify({
'error': False,
'message': 'Updated artist!',
"artist_id": data['LAST_INSERT_ID()']
})
except Exception as e:
return (str(e))
def artist(artist_id):
try:
# Establish connection
c, conn = connection()
result = c.execute("SELECT * FROM artists WHERE artist_id = (%s);" %
artist_id)
artist = c.fetchone()
id = artist['artist_id']
artist['artist_id'] = int(id)
result = c.execute(
"SELECT name, role FROM members WHERE artist_id = (%s)",
[artist['artist_id']])
members = c.fetchall()
result = c.execute(
"SELECT * FROM albums WHERE artist_id = (%s) ORDER BY year_released DESC;",
[artist_id])
albums = c.fetchall()
for album in albums:
result = c.execute(
"SELECT track_id, title, track_no FROM tracks WHERE album_id = (%s) ORDER BY track_no;",
[album['album_id']])
tracklist = c.fetchall()
album['tracklist'] = tracklist
artist['members'] = members
artist['albums'] = albums
c.close()
conn.close()
return jsonify({'artist': artist})
except:
abort(404)
def editAlbum(artist_id, album_id):
data = request.get_json(force=True)
if userAuth(data['username'], data['password']) == False:
return jsonify({'error': True, 'message': 'You are not logged in!'} )
print 1
try:
album = {}
print 2
album['id'] = str(album_id)
album['artist'] = str(artist_id)
album['title'] = data['title']
#album['artwork'] = request.form['artwork']
album['genre'] = data['genre']
album['year'] = str(data['year'])
print 3
c, conn = connection()
'''
file = request.files['artwork']
if file and allowed_file(file.filename):
image = secure_filename("album_img" + str(album['id']) + "." + file.filename.rsplit('.', 1)[1])
file.save(os.path.join(app.config['UPLOAD_FOLDER'], image))
c.execute("UPDATE albums SET artwork = (%s) WHERE album_id = (%s)", [sanitize(image), sanitize(album['id'])])
'''
print 4
c.execute("UPDATE albums SET title = (%s), genre = (%s), year_released = (%s) WHERE album_id = (%s)", [sanitize(album['title']), sanitize(album['genre']), sanitize(album['year']), sanitize(album['id'])])
# Close Connection
conn.commit()
c.close()
conn.close()
print 5
return jsonify({"error": False, "message": "Album Updated!"})
except Exception as e:
return(str(e))
def userAuth(attempted_username, attempted_password):
try:
#check username and password with db
c, conn = connection()
res = c.execute("SELECT * FROM users WHERE username = (%s)", [attempted_username] )
user = c.fetchone()
conn.commit()
c.close()
conn.close()
if int(res) > 0:
db_password = user['password']
if sha256_crypt.verify(attempted_password, db_password):
id = user['user_id']
user['user_id'] = int(id)
user['password'] = attempted_password
user['logged_in'] = True
return user
return False
else:
return False
except Exception as e:
return False
def editArtist(artist_id):
data = request.get_json(force=True)
if userAuth(data['username'], data['password']) == False:
return jsonify({'error': True, 'message': 'You are not logged in!'})
try:
artist = {}
artist['id'] = str(artist_id)
artist['name'] = data['name']
artist['bio'] = data['bio'].encode('utf-8')
artist['genre'] = data['genre']
artist['country'] = data['country']
artist['year'] = str(data['year'])
c, conn = connection()
c.execute(
"UPDATE artists SET name = (%s), biography = (%s), genre = (%s), country = (%s), year_formed = (%s) WHERE artist_id = (%s)",
[
sanitize(artist['name']), artist['bio'],
sanitize(artist['genre']),
sanitize(artist['country']),
sanitize(artist['year']),
sanitize(artist['id'])
])
# Close Connection
conn.commit()
c.close()
conn.close()
return jsonify({'error': False, 'message': 'Updated artist!'})
except Exception as e:
return (str(e))
def addAlbumToArtist(artist_id):
try:
title = "Add Album"
album = {}
if request.method == "POST":
# TODO modify to recieve JSON data.
album['title'] = request.form['title']
#album['artwork'] = request.form['artwork']
album['genre'] = request.form['genre']
album['year'] = request.form['year']
file = request.files['artwork']
title = album['title']
c, conn = connection()
if file and allowed_file(file.filename):
image = secure_filename("album_img" + str(album['id']) + "." +
file.filename.rsplit('.', 1)[1])
file.save(os.path.join(app.config['UPLOAD_FOLDER'], image))
c.execute(
"INSERT INTO albums ( artist_id, title, genre, year_released, artwork) VALUES(%s, %s, %s, %s, %s)",
[
sanitize(album['artist_id']),
sanitize(album['title']),
sanitize(album['genre']),
sanitize(album['year']),
sanitize(image)
])
else:
c.execute(
"INSERT INTO albums ( artist_id, title, genre, year_released) VALUES(%s, %s, %s, %s)",
[
sanitize(album['artist_id']),
sanitize(album['title']),
sanitize(album['genre']),
sanitize(album['year'])
])
# Close Connection
conn.commit()
c.close()
conn.close()
#gc.collect()
return redirect('/artist/?artist=' + str(album['artist_id']) +
"#album" + str(album['id']))
elif request.method == "GET":
title = "Add Album"
album = {}
artist = {}
# ------------------------------
return render_template('pages/edit-album.html',
title=title,
album=album,
artist=artist)
else:
return render_template('err/404.html', title=title)
except Exception as e:
return (str(e))
