From a8b1925aaea09e6b4fab428bfb8d0a505c1893eb Mon Sep 17 00:00:00 2001
From: Egor fureunoir Gorbunov <contact@fureunoir.com>
Date: Wed, 21 May 2025 20:04:12 +0300
Subject: [PATCH] Features: 1) Add query parameters `q` and `limit` to address
 autocomplete schema for enhanced filtering; 2) Update 'create' endpoint
 schema to use `AddressCreateSerializer`;

Fixes: 1) Replace `AddressAutocompleteInputSerializer` with `AddressCreateSerializer` in imports and references;

Extra: None;
---
 core/docs/drf/viewsets.py | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/core/docs/drf/viewsets.py b/core/docs/drf/viewsets.py
index 693d4a53..fea8305e 100644
--- a/core/docs/drf/viewsets.py
+++ b/core/docs/drf/viewsets.py
@@ -5,7 +5,7 @@ from rest_framework import status
 from core.docs.drf import BASE_ERRORS
 from core.serializers import (
     AddOrderProductSerializer,
-    AddressAutocompleteInputSerializer,
+    AddressCreateSerializer,
     AddressSerializer,
     AddressSuggestionSerializer,
     AddWishlistProductSerializer,
@@ -459,7 +459,7 @@ ADDRESS_SCHEMA = {
     ),
     "create": extend_schema(
         summary=_("create a new address"),
-        request=AddressSerializer,
+        request=AddressCreateSerializer,
         responses={
             status.HTTP_201_CREATED: AddressSerializer,
             **BASE_ERRORS,
@@ -490,7 +490,18 @@ ADDRESS_SCHEMA = {
     ),
     "autocomplete": extend_schema(
         summary=_("autocomplete address suggestions"),
-        request=AddressAutocompleteInputSerializer,
+        parameters=[OpenApiParameter(
+            name="q",
+            location=OpenApiParameter.QUERY,
+            description=_("raw data query string, please append with data from geo-IP endpoint"),
+            type=str,
+        ),
+            OpenApiParameter(
+                name="limit",
+                location=OpenApiParameter.QUERY,
+                description=_("limit the results amount, 1 < limit < 10, default: 5"),
+                type=int,
+            )],
         responses={
             status.HTTP_200_OK: AddressSuggestionSerializer(many=True),
             **BASE_ERRORS,