added superlike setting

Author: Seluj78

backend/PyMatcha/routes/api/like.py

@@ -2,6 +2,7 @@
 from flask import request
 from flask_jwt_extended import current_user
 from flask_jwt_extended import jwt_required
+from PyMatcha import redis
 from PyMatcha.models.like import Like
 from PyMatcha.models.match import Match
 from PyMatcha.models.user import get_user
@@ -10,21 +11,18 @@
 from PyMatcha.utils.errors import NotFoundError
 from PyMatcha.utils.success import Success
 from PyMatcha.utils.success import SuccessOutput
+from PyMatcha.utils.tasks import set_user_superlikes
 
 like_bp = Blueprint("like", __name__)
 
 
 @like_bp.route("/like/<uid>", methods=["POST"])
-@jwt_required
 @validate_params({"is_superlike": bool})
+@jwt_required
 def like_profile(uid):
 
     is_superlike = request.get_json()["is_superlike"]
 
-    if is_superlike:
-        # TODO: Check that user has superlikes left today
-        pass
-
     try:
         u = get_user(uid)
     except NotFoundError:
@@ -33,6 +31,15 @@ def like_profile(uid):
         raise BadRequestError("Cannot like yourself.")
     if current_user.already_likes(u.id):
         raise BadRequestError("You already liked this person.")
+
+    if is_superlike:
+        superlike_counter = int(redis.get(f"superlikes:{current_user.id}"))
+        if superlike_counter <= 0:
+            set_user_superlikes.apply_async(current_user.id, amount=5, eta=86400)
+            raise BadRequestError("No more superlikes today !", "Try later")
+        else:
+            redis.decr(f"superlikes:{current_user.id}")
+
     Like.create(liker_id=current_user.id, liked_id=u.id, is_superlike=is_superlike)
 
     if u.already_likes(current_user.id):

backend/PyMatcha/routes/api/messages.py

@@ -54,8 +54,8 @@ def get_opened_conversations():
 
 
 @messages_bp.route("/messages/send", methods=["POST"])
-@jwt_required
 @validate_params(REQUIRED_KEYS_NEW_MESSAGE)
+@jwt_required
 def send_message():
     current_app.logger.debug("/messages -> Call")
     data = request.get_json()

backend/PyMatcha/routes/api/profile/complete.py

@@ -22,6 +22,7 @@
 from flask import request
 from flask_jwt_extended import current_user
 from flask_jwt_extended import jwt_required
+from PyMatcha import redis
 from PyMatcha.models.tag import Tag
 from PyMatcha.utils.decorators import validate_params
 from PyMatcha.utils.errors import BadRequestError
@@ -33,8 +34,8 @@
 
 
 @profile_complete_bp.route("/profile/complete", methods=["POST"])
-@jwt_required
 @validate_params(REQUIRED_PARAMS_COMPLETE_PROFILE)
+@jwt_required
 def complete_profile():
     if current_user.is_profile_completed:
         raise BadRequestError(
@@ -76,4 +77,5 @@ def complete_profile():
     current_user.gender = gender
     current_user.birthdate = birthdate
     current_user.save()
+    redis.set(f"superlikes:{current_user.id}", 5)
     return Success("Profile completed !")

backend/PyMatcha/routes/api/profile/edit.py

@@ -53,8 +53,8 @@
 
 
 @profile_edit_bp.route("/profile/edit", methods=["PUT"])
-@jwt_required
 @validate_params(REQUIRED_PARAMS_EDIT_PROFILE)
+@jwt_required
 def edit_profile():
     if not current_user.is_profile_completed:
         raise BadRequestError("The user has not completed his profile.", "Complete your profile and try again.")
@@ -103,8 +103,8 @@ def edit_profile():
 
 
 @profile_edit_bp.route("/profile/edit/email", methods=["PUT"])
-@jwt_required
 @validate_params({"email": str})
+@jwt_required
 def edit_email():
     data = request.get_json()
     new_email = data["email"].lower()
@@ -121,8 +121,8 @@ def edit_email():
 
 
 @profile_edit_bp.route("/profile/edit/password", methods=["PUT"])
-@jwt_required
 @validate_params({"old_password": str, "new_password": str})
+@jwt_required
 def edit_password():
     data = request.get_json()
     old_password = data["old_password"]
@@ -141,8 +141,8 @@ def edit_password():
 
 
 @profile_edit_bp.route("/profile/edit/geolocation", methods=["PUT"])
-@jwt_required
 @validate_params({"ip": str}, {"lat": float, "lng": float})
+@jwt_required
 def edit_geolocation():
     data = request.get_json()
     ip = data["ip"]

backend/PyMatcha/utils/decorators.py

@@ -45,7 +45,7 @@ def wrapper(*args, **kwargs):
 
             for key, value in data.items():
                 if not value:
-                    if required[key] == int:
+                    if required[key] == int or required[key] == bool:
                         pass
                     else:
                         raise BadRequestError(f"The item {key} cannot be None or empty.")

backend/PyMatcha/utils/orm/_model.py

@@ -301,6 +301,7 @@ def get_multi(cls, **kwargs):
         if data:
             return cls(data)
         else:
+            # TODO: All db gets return None instead of valuerror
             raise ValueError("Not found.")
 
     @classmethod

backend/PyMatcha/utils/tasks.py

@@ -27,7 +27,10 @@ def update_heat_scores():
         reports_received = len(user.get_reports_received())
         views = len(user.get_views())
         matches = len(user.get_matches())
-        messages = len(Message.get_multi(to_id=user.id))
+        try:
+            messages = len(Message.get_multi(to_id=user.id))
+        except ValueError:
+            messages = 0
 
         score = 30
         if user.username == "seluj78" or user.username == "tet":
@@ -157,3 +160,8 @@ def update_user_recommendations():
         )
         redis.expire(f"user_recommendations:{str(user_to_update.id)}", 3600)
     return f"Successfully updated recommendations for {count} users."
+
+
+@celery.task
+def set_user_superlikes(user_id, amount=5):
+    redis.set(f"superlikes:{user_id}", amount)