File tree 1 file changed +14
-9
lines changed
1 file changed +14
-9
lines changed Original file line number Diff line number Diff line change @@ -205,25 +205,30 @@ def get_instance_or_manager_resource_type(resource_instance_or_manager):
205
205
pass
206
206
207
207
208
+ def get_resource_type_from_model (model ):
209
+ # need to convert the verbose name which uses spaces to underscores
210
+ # to be propely parsed in format_relation_name
211
+ underscored_model_name = model ._meta .verbose_name .replace (" " , "_" )
212
+ return format_relation_name (underscored_model_name )
213
+
214
+
208
215
def get_resource_type_from_queryset (qs ):
209
- return format_relation_name (qs .model . _meta . model . __name__ )
216
+ return get_resource_type_from_model (qs .model )
210
217
211
218
212
219
def get_resource_type_from_instance (instance ):
213
- return format_relation_name (instance ._meta .model . __name__ )
220
+ return get_resource_type_from_model (instance ._meta .model )
214
221
215
222
216
223
def get_resource_type_from_manager (manager ):
217
- return format_relation_name (manager .model . __name__ )
224
+ return get_resource_type_from_model (manager .model )
218
225
219
226
220
227
def get_resource_type_from_serializer (serializer ):
221
- try :
222
- # Check the meta class for resource_name
223
- return serializer .Meta .resource_name
224
- except AttributeError :
225
- # Use the serializer model then pluralize and format
226
- return format_relation_name (serializer .Meta .model .__name__ )
228
+ return getattr (
229
+ serializer .Meta ,
230
+ 'resource_name' ,
231
+ get_resource_type_from_model (serializer .Meta .model ))
227
232
228
233
229
234
def extract_attributes (fields , resource ):
You can’t perform that action at this time.
0 commit comments