def edit():
form = EditForm(g.user.nickname)
if request.method == 'POST': avatar_img_file = request.files[form.avatar_img.name]
else: avatar_img_file = None
if form.validate_on_submit(avatar_img_file):
g.user.nickname = form.nickname.data
g.user.about_me = form.about_me.data
db.session.add(g.user)
db.session.commit()
if form.avatar_img.data:
f = request.files[form.avatar_img.name]
pic = Photo(fname = "", timestamp = datetime.utcnow(), owner = g.user)
db.session.add(pic)
db.session.commit()
#try:
pic.fname = (str(pic.id)+"."+f.filename.split(".")[-1])
f.save(os.path.join(UPLOAD_IMG_DIR, pic.fname))
g.user.set_avatar(pic)
db.session.add(pic)
db.session.add(g.user)
db.session.commit()
flash('Your changes have been saved.', 'info')
return redirect(url_for('user', nickname=g.user.nickname))
else:
form.nickname.data = g.user.nickname
form.about_me.data = g.user.about_me
return render_template('edit.html',
form = form,
user = g.user)
def uploadImg():
imgBase64 = request.form['imgDataUrl']
includePost = request.form['includePost']
postId = request.form['postId']
print "Include Post:", includePost
print "PostId", postId
img = base64.b64decode(imgBase64.split(",")[-1])
if includePost == "true":
print "including post..."
pic = Photo(fname = "", timestamp = datetime.utcnow(), owner = g.user, post_id=(int(postId)))
else:
print "no post... only photo"
pic = Photo(fname = "", timestamp = datetime.utcnow(), owner = g.user)
db.session.add(pic)
db.session.commit()
pic.fname = (str(pic.id)+".png")
db.session.add(pic)
db.session.commit()
with open(os.path.join(UPLOAD_IMG_DIR, pic.fname), "wb") as f:
f.write(img)
return jsonify({'response':'success'})
def submitGroupEdits(group_name):
editGroupForm = EditGroupForm(group_name)
group = Group.query.filter_by(name=group_name).first()
if not group:
flash("'"+group_name+"' does not exist!", 'error')
redirect(url_for('home'))
if g.user != group.creator:
flash("You are not authorized to manage the group '"+group_name+"'!", 'error')
redirect(url_for('group', name=group_name))
avatar_img_file = request.files[editGroupForm.avatar_img.name]
if editGroupForm.validate_on_submit(avatar_img_file):
group.name = editGroupForm.name.data
group.about = editGroupForm.about.data
db.session.add(group)
db.session.commit()
if editGroupForm.avatar_img.data:
if group.avatar_photo != None: oldPic = Photo.query.get(group.avatar_photo)
else: oldPic = None
f = request.files[editGroupForm.avatar_img.name]
pic = Photo(fname = "", timestamp = datetime.utcnow(), owner = g.user)
db.session.add(pic)
db.session.commit()
#try:
pic.fname = (str(pic.id)+"."+f.filename.split(".")[-1])
f.save(os.path.join(UPLOAD_IMG_DIR, pic.fname))
group.set_avatar(pic)
db.session.add(pic)
db.session.add(group)
if oldPic: db.session.delete(oldPic)
db.session.commit()
if oldPic: oldPic.delete_files()
flash('Your changes have been saved.', 'info')
return redirect(url_for('group', group_name=group_name))
else:
return redirect(url_for('manageGroup', group_name=group_name))
