Example #1
0
 def add_test_users():
     """
     Method adds users to the database for testing
     """
     _pword = "test"
     user = User(email='*****@*****.**', password=_pword)
     user2 = User(email='*****@*****.**', password=_pword)
     db.session.add(user)
     db.session.add(user2)
     db.session.commit()
Example #2
0
def user_logout(user_id):
    app.logger.info("request:[%s],[%s],[%s]" % (request.headers, request.args, request.json))
    app.logger.info("current_user :%s" % current_user.user_id)
    app.logger.info("logout: %s" % user_id)

    user = User(user_id)
    ret = user.logout()
    ret_json = jsonify(ret)
    app.logger.info("logout %s:[%s]" % (user_id, ret_json))
    return ret_json
Example #3
0
def send_score(user_id):
    app.logger.info("request:[%s],[%s],[%s]" % (request.headers, request.args, request.json))
    app.logger.info("current_user :%s" % current_user.user_id)

    user = User(current_user.user_id)
    ret = user.show_user()

    ret_json = jsonify(ret)
    app.logger.info("send_score %s:[%s]" % (user_id, ret_json))
    return ret_json
Example #4
0
 def test_add_user(self):
     """
     Method checks that add user method actually adds a user
     to the database
     """
     _pword = "test"
     user = User(email='*****@*****.**', password=_pword)
     check = user.add_user()
     self.assertTrue(check, "User should be added")
     self.assertTrue(
         user.id, "User doesnot contain id so he is not added to the db")
Example #5
0
 def test_no_repeated_users_added(self):
     """
     Method checks that add user method actually adds a user
     to the database
     """
     _pword = "test"
     user = User(email='*****@*****.**', password=_pword)
     check = user.add_user()
     self.assertFalse(check, "User should already exist")
     self.assertFalse(
         user.id, "User doesnot contain id so he is not added to the db")
Example #6
0
def add_friend_confirm():
    app.logger.info("request:[%s],[%s],[%s]" % (request.headers, request.args, request.json))
    app.logger.info("current_user :%s" % current_user.user_id)
    if request.json is None or 'friend' not in request.json:
        app.logger.error("missing something:friend key is lost")
        abort(400)

    friend = request.json.get('friend')

    user = User(current_user.user_id)
    ret = user.add_friend_confirm(friend)

    ret_json = jsonify(ret)
    app.logger.info("add_friend_confirm %s:[%s]" % (friend, ret_json))
    return ret_json
Example #7
0
def get_blacks(user_id):
    app.logger.info("request:[%s],[%s],[%s]" % (request.headers, request.args, request.json))
    app.logger.info("current_user :%s" % current_user.user_id)

    if current_user.user_id != user_id:
        if current_user.user_id != 'admin':
            app.logger.error("access deny for user not the true user")
            abort(400)

    user = User(user_id)
    ret = user.get_blocks(user_id)

    ret_json = jsonify(ret)
    app.logger.info("black_sb %s:[%s]" % (user_id, ret_json))
    return ret_json
Example #8
0
def user_upload_head_img(user_id):
    app.logger.info("request:[%s],[%s],[%s]" % (request.headers, request.args, request.json))
    app.logger.info("current_user :%s" % current_user.user_id)
    app.logger.info("start upload file")

    if current_user.user_id != user_id:
        app.logger.warn("user not assist:%s,%s\n", current_user.user_id, user_id)
        abort(401)

    # upload new head portrait to store and update info's url
    post_file = request.files.get('file1')
    app.logger.info("files :%s" % request.files)
    if post_file:
        # get file and save it to local tmp
        fname = secure_filename(post_file.filename)
        ext_name = fname.split('.')[-1]
        obj_id = str(ObjectId())
        pic_name = '%s.%s' % (obj_id, ext_name)

        localfile = os.path.join(app.config['UPLOAD_FOLDER'], pic_name)
        app.logger.info("start upload file to local store:[%s],[%s]" % (fname, localfile))
        post_file.save(localfile)

        # upload file to oss
        pic_url = 'zuohaoshi/%s' % current_user.user_id
        # pic_name = '%s' % fname  # current_user.user_id + ext_name
        file_url = upload_file_to_store(pic_url, pic_name, localfile)
        if file_url is None:
            app.logger.error("file upload failed")
            abort(400)

        app.logger.info("end upload file to store:%s\n" % file_url)

        # update user's portrait info
        img_info = dict()
        img_info['head_img'] = '%s' % file_url

        # delete local tmp file
        os.remove(localfile)

        user = User(current_user.user_id)
        ret = user.modify_user(img_info)
        ret_json = jsonify(dict(ret, **img_info))
        app.logger.info("modify user head image %s:[%s,%s]\n" % (current_user.user_id, ret, img_info))
        return ret_json
    else:
        app.logger.error("missing something:file key is lost")
        abort(400)
Example #9
0
def un_black_sb():
    app.logger.info("request:[%s],[%s],[%s]" % (request.headers, request.args, request.json))
    app.logger.info("current_user :%s" % current_user.user_id)

    if request.json is None or 'black' not in request.json:
        app.logger.error("missing something:follow key is lost")
        abort(400)

    black = request.json.get('black')

    user = User(current_user.user_id)
    ret = user.unblock(current_user.user_id, black)

    ret_json = jsonify(ret)
    app.logger.info("black_sb %s:[%s]" % (black, ret_json))
    return ret_json
Example #10
0
def un_follow_sb(user_id):
    app.logger.info("request:[%s],[%s],[%s]" % (request.headers, request.args, request.json))
    app.logger.info("current_user :%s" % current_user.user_id)

    if request.json is None or 'follow' not in request.json:
        app.logger.error("missing something:follow key is lost")
        abort(400)

    follow = request.json.get('follow')

    user = User(current_user.user_id)
    ret = user.un_follow_sb(follow)

    ret_json = jsonify(ret)
    app.logger.info("un_follow_sb %s:[%s]" % (user_id, ret_json))
    return ret_json
Example #11
0
def modify_user():
    app.logger.info("modify_user req:[%s],[%s],[%s]" % (request.headers, request.args, request.json))
    app.logger.info("current_user :%s" % current_user.user_id)

    user = User(current_user.user_id)
    info = request.json

    if info is None:
        app.logger.error("missing something:info key is lost")
        abort(400)

    ret = user.modify_user(info)

    ret_json = jsonify(ret)
    app.logger.info("modify_user %s:[%s]" % (current_user.user_id, ret))
    return ret_json
Example #12
0
def del_user():
    app.logger.info("request:[%s],[%s],[%s]" % (request.headers, request.args, request.json))
    app.logger.info("current_user :%s" % current_user.user_id)

    user_id = request.json.get("uid")
    if user_id is None:
        app.logger.error("missing something:uid key is lost")
        abort(400)

    # may delete by self or admin
    user = User(current_user.user_id)
    ret = user.del_user(user_id)

    ret_json = jsonify(ret)
    app.logger.info("logout %s:[%s]" % (user_id, ret_json))
    return ret_json
Example #13
0
def verify_token(token=None):
    """
    Verifies the token before a restricted application process occurs
    """
    token = request.headers.get('x-access-token') or token
    user_id = User.verify_authentication_token(token)
    if user_id:
        g.current_user = User.query.filter_by(id=user_id).first()
        return True
    return False
Example #14
0
    def post(self):
        """
        Handles post requests for registration of a new user
        """
        post_data = request.get_json()
        fname = strip_white_space(post_data.get('fname')) or None
        lname = strip_white_space(post_data.get('lname')) or None
        email = strip_white_space(post_data.get('email'),
                                  skip_check_symbols=True) or None
        password = strip_white_space(post_data.get('password')) or None

        if not validate_email(email):
            return abort(400, 'Invalid Email Address')
        if not fname or not lname or not fname.isalpha() or not lname.isalpha(
        ):
            return abort(400, "First and last name must be provided")
        if not password:
            return abort(400, "Password must be provided")

        user = User(email=email,
                    fname=fname.capitalize(),
                    lname=lname.capitalize(),
                    password=password)

        try:
            check = user.save_user()
            if check:
                auth_token = user.generate_authentication_token()
                user_data = dict(data=marshal(user, USER))
                response = {
                    'status': 'success',
                    'message': 'Successfully Registered',
                    'auth_token': auth_token.decode('utf-8')
                }
                response.update(user_data)
                return response, 201
            else:
                response = {'status': 'fail', 'message': 'User Exists'}
                return response, 409
        except Exception as e:
            return abort(500, 'Error creating your account:{}'.format(e))
Example #15
0
def user_login():
    app.logger.info("request:[%s],[%s],[%s]" % (request.headers, request.args, request.json))

    account = request.json.get('account')
    password = request.json.get('passwd')
    if account is None or password is None:
        app.logger.error("missing something")
        abort(400)
    ret = User.login(account, password)

    ret_json = jsonify(ret)
    app.logger.info("user_login %s:[%s]" % (account, ret_json))
    return ret_json
Example #16
0
def insert():
    user1 = User(name='张三',
                 password='******',
                 email='*****@*****.**',
                 gender=True)
    user2 = User(name='李四',
                 password='******',
                 email='*****@*****.**',
                 gender=True)
    user3 = User(name='王五',
                 password='******',
                 email='*****@*****.**',
                 gender=True)
    user4 = User(name='莫愁',
                 password='******',
                 email='*****@*****.**',
                 gender=False)
    user5 = User(name='凤姐',
                 password='******',
                 email='*****@*****.**',
                 gender=False)
    db.session.add_all([user1, user2, user3, user4, user5])
    db.session.commit()
    return '插入成功'
Example #17
0
def register_user():
    app.logger.info("request:[%s],[%s],[%s]" % (request.headers, request.args, request.json))
    # json:post,args:get,header have format

    # import pydevd
    # pydevd.settrace('192.168.3.1', port=12345, stdoutToServer=True, stderrToServer=True)

    # get identify code
    if request.method == 'GET':
        account = request.args.get("account")
        if account is None:
            app.logger.error("missing something")
            abort(400)  # missing phone_num
        ret = User.register_user(account, None, None)
        return jsonify(ret)
    elif request.method == 'POST':
        account = request.json.get('account')
        identify_code = request.json.get('identify_code')
        password = request.json.get('passwd')
        if account is None or identify_code is None or password is None:
            app.logger.error("missing something")
            abort(400)
        ret = User.register_user(account, identify_code, password)
        return jsonify(ret)
Example #18
0
def new_user():
    user = User(**request.get_json())
    db.session.add(user)
    db.session.commit()
    return '', 201, {'Location': url_for('api.get_user', id=user.id)}