def test_retrieval(self): uid1 = User.create("jack", "password") user = User.get(uid1) user2 = User.get_by_name("jack") assert user.id == user2.id assert None is User.get(999) assert None is User.get_by_name("not-existed")
def connect(): username = request.values.get("username") if "user_id" not in session: return jsonify({"error": 1003, "error_msg": "login first"}) target_user = User.get_by_name(username) if target_user is None: return jsonify({"error": 1004, "error_msg": "user not found"}) ContactRelation.connect(session["user_id"], target_user.id) return jsonify({"error": 0})