views.py 文件源码

def avatar():
    basedir = os.path.abspath(os.path.dirname(__file__))
    file_dir = os.path.join(basedir,current_app.config['UPLOAD_IMG_FOLDER'])
    username = request.values.get('username')
    form = AvatarForm()
    if not os.path.exists(file_dir):
        os.makedirs(file_dir)
    if request.method == 'POST':
        file = request.files['file']
        if file and allowed_file(file.filename):
            size = (50, 50)
            im = Image.open(file)
            im.thumbnail(size)
            if file and allowed_file(file.filename):
                filename = secure_filename(file.filename)
                newname = username+'.jpg'
                im.save(os.path.join(file_dir, newname))
                token = base64.b64encode(newname)
                user = User.query.filter_by(username=username).first()
                user.token = token
                db.session.add(user)
                db.session.commit()
                return redirect(url_for('main.index'))
    return render_template('auth/change_avatar.html', form=form)