1010from common .aspect .interface_auth import UserInterfaceAuthDependency
1111from common .aspect .pre_auth import CurrentUserDependency , PreAuthDependency
1212from common .enums import BusinessType
13+ from common .vo import DataResponseModel , PageResponseModel , ResponseBaseModel
1314from module_admin .entity .vo .dict_vo import (
1415 DeleteDictDataModel ,
1516 DeleteDictTypeModel ,
2223from module_admin .service .dict_service import DictDataService , DictTypeService
2324from utils .common_util import bytes2file_response
2425from utils .log_util import logger
25- from utils .page_util import PageResponseModel
2626from utils .response_util import ResponseUtil
2727
2828dict_controller = APIRouter (prefix = '/system/dict' , dependencies = [PreAuthDependency ()])
2929
3030
3131@dict_controller .get (
3232 '/type/list' ,
33- response_model = PageResponseModel ,
33+ response_model = PageResponseModel [ DictTypeModel ] ,
3434 dependencies = [UserInterfaceAuthDependency ('system:dict:list' )],
3535)
3636async def get_system_dict_type_list (
@@ -49,6 +49,7 @@ async def get_system_dict_type_list(
4949
5050@dict_controller .post (
5151 '/type' ,
52+ response_model = ResponseBaseModel ,
5253 dependencies = [UserInterfaceAuthDependency ('system:dict:add' )],
5354)
5455@ValidateFields (validate_model = 'add_dict_type' )
@@ -71,6 +72,7 @@ async def add_system_dict_type(
7172
7273@dict_controller .put (
7374 '/type' ,
75+ response_model = ResponseBaseModel ,
7476 dependencies = [UserInterfaceAuthDependency ('system:dict:edit' )],
7577)
7678@ValidateFields (validate_model = 'edit_dict_type' )
@@ -91,6 +93,7 @@ async def edit_system_dict_type(
9193
9294@dict_controller .delete (
9395 '/type/refreshCache' ,
96+ response_model = ResponseBaseModel ,
9497 dependencies = [UserInterfaceAuthDependency ('system:dict:remove' )],
9598)
9699@Log (title = '字典类型' , business_type = BusinessType .UPDATE )
@@ -103,6 +106,7 @@ async def refresh_system_dict(request: Request, query_db: Annotated[AsyncSession
103106
104107@dict_controller .delete (
105108 '/type/{dict_ids}' ,
109+ response_model = ResponseBaseModel ,
106110 dependencies = [UserInterfaceAuthDependency ('system:dict:remove' )],
107111)
108112@Log (title = '字典类型' , business_type = BusinessType .DELETE )
@@ -118,7 +122,10 @@ async def delete_system_dict_type(
118122 return ResponseUtil .success (msg = delete_dict_type_result .message )
119123
120124
121- @dict_controller .get ('/type/optionselect' , response_model = list [DictTypeModel ])
125+ @dict_controller .get (
126+ '/type/optionselect' ,
127+ response_model = DataResponseModel [list [DictTypeModel ]],
128+ )
122129async def query_system_dict_type_options (
123130 request : Request , query_db : Annotated [AsyncSession , DBSessionDependency ()]
124131) -> Response :
@@ -132,7 +139,7 @@ async def query_system_dict_type_options(
132139
133140@dict_controller .get (
134141 '/type/{dict_id}' ,
135- response_model = DictTypeModel ,
142+ response_model = DataResponseModel [ DictTypeModel ] ,
136143 dependencies = [UserInterfaceAuthDependency ('system:dict:query' )],
137144)
138145async def query_detail_system_dict_type (
@@ -166,7 +173,10 @@ async def export_system_dict_type_list(
166173 return ResponseUtil .streaming (data = bytes2file_response (dict_type_export_result ))
167174
168175
169- @dict_controller .get ('/data/type/{dict_type}' )
176+ @dict_controller .get (
177+ '/data/type/{dict_type}' ,
178+ response_model = DataResponseModel [list [DictDataModel ]],
179+ )
170180async def query_system_dict_type_data (
171181 request : Request ,
172182 dict_type : Annotated [str , Path (description = '字典类型' )],
@@ -183,7 +193,7 @@ async def query_system_dict_type_data(
183193
184194@dict_controller .get (
185195 '/data/list' ,
186- response_model = PageResponseModel ,
196+ response_model = PageResponseModel [ DictDataModel ] ,
187197 dependencies = [UserInterfaceAuthDependency ('system:dict:list' )],
188198)
189199async def get_system_dict_data_list (
@@ -202,6 +212,7 @@ async def get_system_dict_data_list(
202212
203213@dict_controller .post (
204214 '/data' ,
215+ response_model = ResponseBaseModel ,
205216 dependencies = [UserInterfaceAuthDependency ('system:dict:add' )],
206217)
207218@ValidateFields (validate_model = 'add_dict_data' )
@@ -224,6 +235,7 @@ async def add_system_dict_data(
224235
225236@dict_controller .put (
226237 '/data' ,
238+ response_model = ResponseBaseModel ,
227239 dependencies = [UserInterfaceAuthDependency ('system:dict:edit' )],
228240)
229241@ValidateFields (validate_model = 'edit_dict_data' )
@@ -244,6 +256,7 @@ async def edit_system_dict_data(
244256
245257@dict_controller .delete (
246258 '/data/{dict_codes}' ,
259+ response_model = ResponseBaseModel ,
247260 dependencies = [UserInterfaceAuthDependency ('system:dict:remove' )],
248261)
249262@Log (title = '字典数据' , business_type = BusinessType .DELETE )
@@ -261,7 +274,7 @@ async def delete_system_dict_data(
261274
262275@dict_controller .get (
263276 '/data/{dict_code}' ,
264- response_model = DictDataModel ,
277+ response_model = DataResponseModel [ DictDataModel ] ,
265278 dependencies = [UserInterfaceAuthDependency ('system:dict:query' )],
266279)
267280async def query_detail_system_dict_data (
0 commit comments