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