new hierarchy decorators

3 files changed, 8 insertions, 25 deletions

diff --git a/api/social/create_relation.py b/api/social/create_relation.py
index af81b69..5367ac5 100644
--- a/api/social/create_relation.py
+++ b/api/social/create_relation.py
@@ -1,28 +1,10 @@
 from flask import Blueprint, request
 from db import cursor, connection
-from hierarchy import auth_required
+from hierarchy import auth_required, two_person
 from socket_io import io
 import time
 
 
-# @two_person_endpoint decorator
-# defines (user_1_id, user_2_id) in endpoint handler function arguments
-def two_person_endpoint(func):
-	@auth_required("user")
-	def wrapper(user_1_id):
-		data = request.get_json()
-		user_2_id = data.get("id") or ""
-
-		if not user_1_id or \
-                                                                 not user_2_id:
-			return "", 403
-
-		return func(user_1_id, user_2_id)
-
-	wrapper.__name__ = func.__name__
-	return wrapper
-
-
 def create_relation(user_1_id, user_2_id, relation_type):
 	remove_relation(user_1_id, user_2_id)
 	remove_relation(user_2_id, user_1_id)
@@ -44,7 +26,7 @@ def remove_relation(user_1_id, user_2_id):
 
 
 def create_relation_route(relation_type):
-	@two_person_endpoint
+	@two_person
 	def route(user_1_id, user_2_id):
 		create_relation(user_1_id, user_2_id, relation_type)
 
diff --git a/api/social/destroy_relation.py b/api/social/destroy_relation.py
index ab72c48..2aa793b 100644
--- a/api/social/destroy_relation.py
+++ b/api/social/destroy_relation.py
@@ -1,15 +1,16 @@
 from flask import Blueprint, request
 from db import cursor
-from social.create_relation import remove_relation, two_person_endpoint
+from social.create_relation import remove_relation
 from user.info import get_relation_to
 from socket_io import io
+from hierarchy import two_person
 import time
 
 remove = Blueprint('remove', __name__)
 
 
 @remove.route('/remove', methods=['POST'])
-@two_person_endpoint
+@two_person
 def index(user_1_id, user_2_id):
 	relation = get_relation_to(user_1_id, user_2_id)
 	if relation == "none": return "", 403
@@ -27,7 +28,7 @@ unblock = Blueprint('unblock', __name__)
 
 
 @unblock.route('/unblock', methods=['POST'])
-@two_person_endpoint
+@two_person
 def index(user_1_id, user_2_id):
 	if get_relation_to(user_1_id, user_2_id) != "blocked": return "", 403
 
diff --git a/api/social/friend_accept.py b/api/social/friend_accept.py
index 4eb4837..b434272 100644
--- a/api/social/friend_accept.py
+++ b/api/social/friend_accept.py
@@ -1,14 +1,14 @@
 from flask import Blueprint, request
 from db import cursor, connection
-from social.create_relation import two_person_endpoint
 from socket_io import io
+from hierarchy import two_person
 import time
 
 accept = Blueprint('accept', __name__)
 
 
 @accept.route("/accept", methods=['POST'])
-@two_person_endpoint
+@two_person
 def route(user_1_id, user_2_id):
 	cursor.execute(
 		"update social set type = \"friendship\" where user_1_id = ? and user_2_id = ?",