update for tags

ohcnetwork · Jan 26, 2025 · 45cdda8 · 45cdda8
1 parent a46ca16
commit 45cdda8
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/care/emr/api/viewsets/base.py b/care/emr/api/viewsets/base.py
@@ -225,6 +225,7 @@ class EMRBaseViewSet(GenericViewSet):
     pydantic_retrieve_model: EMRResource = None
     database_model: EMRBaseModel = None
     lookup_field = "external_id"
+    tags = [__name__]
 
     def get_exception_handler(self):
         return emr_exception_handler

diff --git a/care/emr/api/viewsets/condition.py b/care/emr/api/viewsets/condition.py
@@ -29,6 +29,7 @@
 )
 from care.emr.resources.questionnaire.spec import SubjectType
 from care.security.authorization import AuthorizationController
+from care.utils.decorators.schema_decorator import generate_swagger_schema_decorator
 
 
 class ValidateEncounterMixin:
@@ -58,6 +59,7 @@ class ConditionFilters(FilterSet):
     severity = CharFilter(field_name="severity", lookup_expr="iexact")
 
 
+@generate_swagger_schema_decorator
 class SymptomViewSet(
     ValidateEncounterMixin,
     EncounterBasedAuthorizationBase,
@@ -98,6 +100,7 @@ def get_queryset(self):
 InternalQuestionnaireRegistry.register(SymptomViewSet)
 
 
+@generate_swagger_schema_decorator
 class DiagnosisViewSet(
     ValidateEncounterMixin,
     EncounterBasedAuthorizationBase,
@@ -139,6 +142,7 @@ def get_queryset(self):
 InternalQuestionnaireRegistry.register(DiagnosisViewSet)
 
 
+@generate_swagger_schema_decorator
 class ChronicConditionViewSet(
     EMRQuestionnaireResponseMixin,
     EMRCreateMixin,

diff --git a/care/utils/decorators/schema_decorator.py b/care/utils/decorators/schema_decorator.py
@@ -2,8 +2,8 @@
 
 
 def generate_swagger_schema_decorator(cls):
-    if not hasattr(cls, "tags") and not cls.tags:
-        cls.tags = [cls.__name__]
+    if not cls.tags:
+        cls.tags = cls.__name__
 
     base_actions = {
         "create": {