Base code for the search route

Author: Seluj78

backend/PyMatcha/__init__.py

@@ -227,6 +227,7 @@ def check_if_token_is_revoked(decrypted_token):
 from PyMatcha.routes.api.messages import messages_bp
 from PyMatcha.routes.api.recommendations import recommendations_bp
 from PyMatcha.routes.api.profile.images import images_bp
+from PyMatcha.routes.api.search import search_bp
 
 logging.debug("Registering Flask blueprints")
 application.register_blueprint(user_bp)
@@ -243,6 +244,7 @@ def check_if_token_is_revoked(decrypted_token):
 application.register_blueprint(messages_bp)
 application.register_blueprint(recommendations_bp)
 application.register_blueprint(images_bp)
+application.register_blueprint(search_bp)
 
 if application.debug:
     logging.debug("Registering debug route")

backend/PyMatcha/routes/api/search.py

@@ -0,0 +1,72 @@
+import datetime
+
+from flask import Blueprint
+from flask import request
+from flask_jwt_extended import current_user
+from flask_jwt_extended import jwt_required
+from PyMatcha.utils.decorators import validate_params
+from PyMatcha.utils.match_score import _get_common_tags
+from PyMatcha.utils.match_score import _get_distance
+from PyMatcha.utils.match_score import _get_gender_query
+from PyMatcha.utils.success import SuccessOutput
+
+search_bp = Blueprint("search", __name__)
+
+REQUIRED_PARAMS_SEARCH = {
+    "min_age": int,
+    "max_age": int,
+    "min_score": int,
+    "max_score": int,
+    "tags": list,
+    "max_distance": int,
+}
+
+
+@search_bp.route("/search", methods=["POST"])
+@validate_params(REQUIRED_PARAMS_SEARCH)
+@jwt_required
+def search():
+    data = request.get_json()
+    min_age = int(data["min_age"])
+    max_age = int(data["max_age"])
+    min_score = int(data["min_score"])
+    max_score = int(data["max_score"])
+    max_distance = int(data["max_distance"])
+    tags = data["tags"]
+
+    today = datetime.datetime.utcnow()
+
+    query = _get_gender_query(current_user.orientation, current_user.gender)
+    returned_list = []
+    for user in query:
+        user_age = (
+            today.year - user.birthdate.year - ((today.month, today.day) < (user.birthdate.month, user.birthdate.day))
+        )
+
+        if max_age != -1:
+            if user_age > max_age:
+                continue
+        if min_age != -1:
+            if user_age < min_age:
+                continue
+
+        if max_score != -1:
+            if user.heat_score > max_score:
+                continue
+        if min_score != -1:
+            if user.heat_score < min_score:
+                continue
+
+        if max_distance != -1:
+            distance = _get_distance(current_user.geohash, user.geohash)
+            if distance > max_distance:
+                continue
+
+        if tags:
+            user_tags = [t.name for t in user.get_tags()]
+            common_tags = _get_common_tags(tags, user_tags)
+            if not common_tags:
+                continue
+
+        returned_list.append(user.to_dict())
+    return SuccessOutput("search_results", returned_list)