This commit is contained in:
Trullemans Gregory 2021-06-20 14:34:56 +02:00
parent 787ba706aa
commit 088895ec43
1 changed files with 41 additions and 65 deletions

View File

@ -20,36 +20,12 @@ def get_number_of_unread_message(request):
return Message.objects.filter(recipient=request.user).filter(read_at__isnull=True).count()
@login_required
@require_http_methods(["GET"])
def get_messages(request, message_type="received"):
"""Récupère des messages associés l'utilisateur actuellement connecté.
Args:
request (django.http.HttpRequest): (voir mes notes ci-dessous)
message_type (str): { received | sent }
Returns:
Retourne les messages reçus ou envoyés par l'utilisateur connecté.
Si le paramètre `message_type` est vide, la liste renvoyée est vide également.
"""
if message_type == "received":
message_list = request.user.received_messages.all()
elif message_type == "sent":
message_list = request.user.sent_messages.all()
else:
message_list = None
context = {"message_list": message_list, "message_type": message_type}
return render(request, "message_list.html", context)
@login_required
@require_http_methods(["GET"])
def get_received_messages(request):
"""Récupère des messages recus pour l'utilisateur connecté.
"""
return get_messages(request, "received")
return request.user.received_messages.all()
@login_required
@ -57,7 +33,7 @@ def get_received_messages(request):
def get_sent_messages(request):
"""Récupère des messages envoyés par l'utilisateur connecté.
"""
return get_messages(request, "sent")
return request.user.sent_messages.all()
@login_required