From d8c18cee78d359d33ea0cd9f9d6ee03ef2ab90f2 Mon Sep 17 00:00:00 2001
From: ITQ <itq.dev@ya.ru>
Date: Sat, 25 Jan 2025 22:10:40 +0300
Subject: [PATCH] fix: removed null values output from some comments endpoints

---
 solution/api/v1/user/views.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/solution/api/v1/user/views.py b/solution/api/v1/user/views.py
index cd29fdb..6506de4 100644
--- a/solution/api/v1/user/views.py
+++ b/solution/api/v1/user/views.py
@@ -282,6 +282,7 @@ def delete_like(
         status.CREATED: schemas.CommentOut,
         status.BAD_REQUEST: global_schemas.BadRequestError,
     },
+    exclude_none=True,
 )
 def add_comment(
     request: HttpRequest, promocode_id: str, comment: schemas.CommentIn
@@ -314,6 +315,7 @@ def list_comments(
     request: HttpRequest,
     filters: Query[schemas.PromocodeCommentsFilters],
     promocode_id: str,
+    response: HttpResponse
 ) -> tuple[int, schemas.CommentOut]:
     promocodes = Promocode.objects.filter(id=promocode_id)
 
@@ -324,6 +326,8 @@ def list_comments(
 
     comments = promocodes.first().comments.all()
 
+    response["X-Total-Count"] = len(comments)
+
     comments = comments[filters.offset : filters.offset + filters.limit]
 
     return status.OK, [
@@ -362,6 +366,7 @@ def get_comment(
         status.OK: schemas.CommentOut,
         status.BAD_REQUEST: global_schemas.BadRequestError,
     },
+    exclude_none=True,
 )
 def update_comment(
     request: HttpRequest,