@auth_required in /social/accept endpoint

1 files changed, 3 insertions, 14 deletions

diff --git a/api/social/friend_accept.py b/api/social/friend_accept.py
index 8340274..cd11159 100644
--- a/api/social/friend_accept.py
+++ b/api/social/friend_accept.py
@@ -1,25 +1,14 @@
 from flask import Blueprint, request
 from db import cursor, connection
-from auth.login_token import token_login
+from social.create_relation import two_person_endpoint
 from socket_io import io
 import time
 
 accept = Blueprint('accept', __name__)
 
 @accept.route("/accept", methods = ['POST'])
-def route():
-    data = request.get_json()
-
-    user_1_id = data.get("id") or ""
-    token = request.cookies.get("token") or ""
-
-    if not token: return "", 401
-    user_2_id = token_login(token) or ""
-
-    if not user_1_id or \
-       not user_2_id:
-           return "", 403
-
+@two_person_endpoint
+def route(user_1_id, user_2_id):
     cursor.execute("update social set type = \"friendship\" where user_1_id = ? and user_2_id = ?",
             [user_1_id, user_2_id])
     connection.commit()