Example #1
0
    def post(self, userID):

        """
            上传头像
            :return:
            """
        from werkzeug.utils import secure_filename
        from faker import Faker

        f = Faker()
        file = request.files.get('file')
        if not file:
            return jsonify(myResponse(ResponseCode.ERROR, None, ResponseError.NO_FIlE))
        if not userID:
            return jsonify(myResponse(ResponseCode.ERROR, None, cantEmpty("userID")))
        user = User.get(userID)
        fileName = secure_filename(file.filename)
        filePath = getAvatarPath(f.pystr() + '_' + fileName)
        user.avatar = filePath

        try:
            file.save(filePath)
            user.save()
            return jsonify(myResponse(ResponseCode.SUCCESS, user.id, ResponseError.OK))
        except Exception as e:
            log.error(e)
            return jsonify(myResponse(ResponseCode.ERROR, None, ResponseCode.ERROR_TRY_AGAIN))
Example #2
0
 def get(self, userID):
     user = User.get(userID)
     avatar = user.avatar
     with open(avatar, "rb") as f:
         avatar = f.read()
     return Response(avatar, mimetype="image/jpeg")