Example #1
0
def load():
    file_path = os.path.join(os.path.dirname(__file__), 'test_data.json')
    f = open(file_path, 'r')
    content = f.read().decode('utf8')
    f.close()
    data = json.loads(content)
    users = data['users']
    friends = data['friends']
    for user in users:
        pw_md5 = hashlib.md5()
        pw_md5.update(user['password'])
        u = User(name=user['name'],
                 password=pw_md5.hexdigest(),
                 head_img_url=user['head_img_url'],
                 last_login_time=datetime.now())
        u.insert()
        for detail in user['user_details']:
            ud = UserDetail(user_id=u.id,
                            field_name=detail['field_name'],
                            field_value='%s,v' % detail['field_value'],
                            updated_at=datetime.now())
            ud.insert()
            ui = UserIndex(field_name=detail['field_name'],
                           field_value=detail['field_value'],
                           user_id=u.id)
            ui.insert()
    for friend in friends:
        user_id = UserIndex.get_user_id('phone', friend['user_phone'])
        friend_id = UserIndex.get_user_id('phone', friend['friend_phone'])
        f = Friend(user_id=user_id, friend_id=friend_id)
        f.insert()
        print "u_id: %d, f_id: %d" % (user_id, friend_id)
        f = Friend(user_id=friend_id, friend_id=user_id)
        f.insert()
        print "u_id: %d, f_id: %d" % (friend_id, user_id)
Example #2
0
def create_friend(user, team):
    result = {"error_code": 0, "msg": "ok"}
    if not team:
        return {"error_code": 20501, "msg": "team not exist"}
    f_1 = Friend.select().where(Friend.user == user, Friend.team == team,
                                Friend.ftype == "f").first()
    if f_1:
        if f_1.status != "normal":
            f_1.status = "normal"
            f_1.save()
    else:
        f_1 = Friend()
        f_1.user = user
        f_1.team = team
        f_1.ftype = "f"
        f_1.save()

    f_2 = Friend.select().where(Friend.user == user, Friend.team == team,
                                Friend.ftype == "c").first()
    if f_2:
        if f_2.status != "normal":
            f_2.status = "normal"
            f_2.save()
    else:
        f_2 = Friend()
        f_2.user = user
        f_2.team = team
        f_2.ftype = "c"
        f_2.save()
    return result
Example #3
0
def build_edit_friend_form(home_username, friend_username, review):
    sql_instance = SQL()
    result = sql_instance.get_individual_friend(
        home_username=home_username, friend_username=friend_username)
    print(result)
    friend = Friend(username=result[0],
                    user_id=result[1],
                    first_name=result[2],
                    last_name=result[3])
    web_img_paths = []
    if review:
        friend_path = os.path.abspath('static/img/out/needs_review/')
        print(friend_path)
        image_paths = list(paths.list_images(friend_path))
        for image_path in image_paths:
            file_name = image_path.split('/')[-1]
            web_path = '../static/img/out/needs_review/' + file_name
            web_img_paths.append(web_path)
        print(web_img_paths)
    else:
        friend_path = os.path.abspath('static/img/data/' + str(friend.user_id))
        print(friend_path)
        image_paths = list(paths.list_images(friend_path))
        for image_path in image_paths:
            file_name = image_path.split('/')[-1]
            web_path = '../static/img/data/' + str(
                friend.user_id) + '/' + file_name
            web_img_paths.append(web_path)
        print(web_img_paths)
    return EditFriendForm(friend=friend, image_paths=web_img_paths)
def build_model_form(username):
    sql_instance = SQL()
    results = sql_instance.get_friends(username=username)
    user_id = sql_instance.get_user_id(username, username)
    friends = []
    for friend in results:
        friends.append(
            Friend(username=friend[0],
                   user_id=friend[1],
                   first_name=friend[2],
                   last_name=friend[3]))
    model_results = sql_instance.get_models_for_user(username=username)
    models = []
    for model in model_results:
        classification_results = sql_instance.get_all_classifications_for_model(
            model_id=model[0])
        classifications = []
        for c in classification_results:
            orig_path = c[6]
            print(orig_path)
            web_path = '../static/img/out/training/' + orig_path.split(
                '/')[-2] + '/' + orig_path.split('/')[-1]
            classifications.append(
                Classification(c[1], c[2], c[3], c[4], c[5], web_path))
        models.append(
            ModelData(model_id=model[0],
                      classifications=classifications,
                      file_path=model[1]))
    x_vals, y_vals = get_training_data_distribution(username, sql_instance)
    return ModelsForm(user_id=user_id,
                      friends=friends,
                      models=models,
                      x_vals=x_vals,
                      y_vals=y_vals)
Example #5
0
def build_friends_form(username):
    sql_instance = SQL()
    results = sql_instance.get_friends(username=username)
    friends = []
    for friend in results:
        friends.append(
            Friend(username=friend[0],
                   user_id=friend[1],
                   first_name=friend[2],
                   last_name=friend[3]))
    return FriendsForm(friends=friends)
Example #6
0
def sign_up():
    if request.method == "GET":
        return render_template("sign_up.html")
    else:
        form = request.form
        fullname = form["fullname"]
        username = form["username"]
        email = form["email"]
        password = form["password"]
        confirm = form["confirm"]
        bday = form["bday"]
        gender = form["gender"]
        phone = form["telephone"]
        existUsername = User.objects(username=username).first()
        existEmail = User.objects(email=email).first()
        error = None

        if fullname == "" or username == "" or email == "" or password == "" or confirm == "" or bday == "" or gender == "" or phone == "":
            error = "Bạn phải điền đầy đủ thông tin"
            return render_template("sign_up.html", error=error)
        elif existUsername is not None:
            error = "Tên đăng nhập đã tồn tại"
            return render_template("sign_up.html", error=error)
        elif len(username) < 8:
            error = "Tên người dùng không khả dụng"
            return render_template("sign_up.html", error=error)
        elif len(password) < 8:
            error = "Mật khẩu không khả dụng"
            return render_template("sign_up.html", error=error)
        elif confirm != password:
            error = "Rất tiếc. Xác nhận mật khẩu sai"
            return render_template("sign_up.html", error=error)
        else:
            m = User(
                fullname=fullname,
                username=username,
                email=email,
                password=password,
                birthday=bday,
                gender=gender,
                phone=phone,
                avt=
                "https://cdn1.iconfinder.com/data/icons/ninja-things-1/1772/ninja-simple-512.png"
            )
            m.save()
            att = Attribute(username=username)
            att.save()
            hstr_list = All_history(user=username)
            hstr_list.save()
            Friend(username=username).save()
            return redirect("/sign_in")
def respond_friend_request(json_req, user):
    status = 0
    friend_mobile = json_req['friendMobile']
    action = json_req['action']  # True: Accept, False: Reject
    friend_user = db_session.query(User).filter_by(
        Mobile=friend_mobile).first()
    if friend_user:
        user_id = user.UserID
        friend_user_id = friend_user.UserID

        if not (__friendship__(user_id, friend_user_id)
                or user_id == friend_user_id):
            friend_request_received = db_session.query(
                FriendRequest).filter_by(UserID=friend_user_id,
                                         FriendUserID=user_id,
                                         Deleted=False).first()
            if friend_request_received:
                if action:  # True: Accept
                    new_friend = Friend(UserID=friend_user_id,
                                        FriendUserID=user_id)
                    db_session.add(new_friend)

                friend_request_received.AcceptanceStatus = action
                friend_request_received.Deleted = True
                friend_request_received.RespondOn = datetime.datetime.utcnow()
                db_session.commit()

                # FCM Notification
                data = dict(type=notification_type.FRIEND_REQUEST_RESPOND,
                            acceptanceStatus=action,
                            friendUserID=user_id,
                            friendName=user.Name,
                            friendMobile=user.Mobile,
                            imageURL=images.get_serving_url(user.ImageKey,
                                                            secure_url=True)
                            if user.ImageKey else None)
                send_data(friend_user_id, data)

                status = 1
                message = 'success'
            else:
                message = 'No such friend request'
        else:
            message = 'You are already friend'
    else:
        message = 'User with mobile {} doesn\'t exist'.format(friend_mobile)
    return jsonify(status=status, message=message)
Example #8
0
def generate_friends(session):
    from faker import Faker
    import random
    from models.friend import Friend
    faker = Faker()
    profile_ids_available = list(range(1, 5001))
    profiles_used = []
    for i in range(1, 5001):
        profile_id = random.choice(profile_ids_available)
        while profile_id in profiles_used:
            profile_id = random.choice(profile_ids_available)
        friend = Friend()
        friend.friend_id = i
        friend.profile_id = profile_id
        session.add(friend)
        profiles_used.append(profile_id)
    session.commit()
Example #9
0
def new_friend():
    current_user = User.get_by_id(get_jwt_identity())

    approve_user = User.get_by_id(request.form.get("id"))

    friend = Friend(user1_id=current_user.id, user2_id=approve_user.id)

    if friend.save():
        delete = FriendRequest.delete().where(
            FriendRequest.sender_id == approve_user.id,
            FriendRequest.recipient_id == current_user.id)
        delete.execute()
        return jsonify({
            "successful": True,
            "message": "Friend request has been accepted!"
        })
    else:
        return jsonify({"error": "Something went wrong!"})