44import uuid
55from abc import ABC , ABCMeta
66from collections import ChainMap
7- from typing import Any , Match , Optional , Tuple , Type
7+ from typing import Any , List , Match , Optional , Tuple , Type
88
99from django .http import HttpRequest
1010from ninja import ModelSchema
@@ -109,18 +109,18 @@ async def get_objs(self, request: HttpRequest, filters: str = None) -> Any: # t
109109 class DataSchema (ModelSchema ):
110110 class Config :
111111 model = model_opts .model
112- model_exclude = []
112+ model_exclude : List = []
113113 if model_opts .model_exclude :
114114 model_exclude .extend (model_opts .model_exclude )
115115 # Remove pk(id) from Create/Update Schema
116- model_exclude .extend ([model ._meta .pk .name ])
116+ model_exclude .extend ([model ._meta .pk .name ]) # type: ignore
117117 else :
118118 if model_opts .model_fields == MODEL_FIELDS_ATTR_DEFAULT :
119119 # Remove pk(id) from Create/Update Schema
120- model_exclude .extend ([model ._meta .pk .name ])
120+ model_exclude .extend ([model ._meta .pk .name ]) # type: ignore
121121 else :
122122 model_fields = (
123- model_opts .model_fields # type: ignore
123+ model_opts .model_fields
124124 if model_opts .model_fields
125125 else MODEL_FIELDS_ATTR_DEFAULT
126126 )
@@ -134,9 +134,11 @@ async def add_obj( # type: ignore
134134 """
135135 obj_id = await self .service .add_obj (** data .dict ())
136136 if obj_id :
137- return BaseApiResponse ({"id" : obj_id }, code = 201 )
137+ return BaseApiResponse ({"id" : obj_id }, code = 201 , message = "Created." )
138138 else :
139- return BaseApiResponse ("Add failed" , code = 204 ) # pragma: no cover
139+ return BaseApiResponse (
140+ code = 204 , message = "Add failed."
141+ ) # pragma: no cover
140142
141143 async def patch_obj ( # type: ignore
142144 self , request : HttpRequest , id : int , data : DataSchema
@@ -146,9 +148,9 @@ async def patch_obj( # type: ignore
146148 Update a single object
147149 """
148150 if await self .service .patch_obj (id = id , payload = data .dict ()):
149- return BaseApiResponse ("Updated." )
151+ return BaseApiResponse (message = "Updated." )
150152 else :
151- return BaseApiResponse ("Update Failed", code = 400 )
153+ return BaseApiResponse (code = 400 , message = "Updated Failed" )
152154
153155 DataSchema .__name__ = (
154156 f"{ model_opts .model .__name__ } __AutoSchema({ str (uuid .uuid4 ())[:4 ]} )"
0 commit comments