70 lines
2.4 KiB
Python
70 lines
2.4 KiB
Python
from drf_spectacular.openapi import AutoSchema
|
|
|
|
from pluralizer import Pluralizer
|
|
|
|
import re
|
|
|
|
|
|
class CustomAutoSchema(AutoSchema):
|
|
method_mapping = {
|
|
'get': 'get',
|
|
'post': 'create',
|
|
'put': 'update',
|
|
'patch': 'partial_update',
|
|
'delete': 'delete',
|
|
}
|
|
|
|
pluralizer = Pluralizer()
|
|
|
|
|
|
def get_operation_id(self):
|
|
# Modified operation id getter from
|
|
# https://github.com/tfranzel/drf-spectacular/blob/6973aa48f4ff08f7f33799d50c288fcc79ea8076/drf_spectacular/openapi.py#L424-L441
|
|
|
|
tokenized_path = self._tokenize_path()
|
|
|
|
# replace dashes as they can be problematic later in code generation
|
|
tokenized_path = [t.replace('-', '_') for t in tokenized_path]
|
|
|
|
# replace plural forms with singular forms
|
|
tokenized_path = [self.pluralizer.singular(t) for t in tokenized_path]
|
|
|
|
if not tokenized_path:
|
|
tokenized_path.append('root')
|
|
|
|
model = tokenized_path.pop()
|
|
|
|
if self.method == 'GET' and self._is_list_view():
|
|
action = 'get'
|
|
model = self.pluralizer.plural(model)
|
|
else:
|
|
action = self.method_mapping[self.method.lower()]
|
|
|
|
if re.search(r'<drf_format_suffix\w*:\w+>', self.path_regex):
|
|
tokenized_path.append('formatted')
|
|
|
|
# rename `get_radio_radio_track` to `get_radio_track`
|
|
if len(tokenized_path) > 1 and tokenized_path[1] == 'radio' and tokenized_path[1] == 'radio':
|
|
tokenized_path.pop(0)
|
|
|
|
# rename `get_manage_channel` to `admin_get_channel`
|
|
elif len(tokenized_path) > 0 and tokenized_path[0] == 'manage':
|
|
tokenized_path.pop(0)
|
|
|
|
# rename `get_manage_library_album` to `admin_get_album`
|
|
if len(tokenized_path) > 0 and tokenized_path[0] == 'library':
|
|
tokenized_path.pop(0)
|
|
|
|
# rename `get_manage_user_users` to `admin_get_users`
|
|
elif len(tokenized_path) > 0 and tokenized_path[0] == 'user':
|
|
tokenized_path.pop(0)
|
|
|
|
# rename `get_manage_moderation_note` to `moderation_get_note`
|
|
elif len(tokenized_path) > 0 and tokenized_path[0] == 'moderation':
|
|
tokenized_path.pop(0)
|
|
return '_'.join(['moderation', action] + tokenized_path + [model])
|
|
|
|
return '_'.join(['admin', action] + tokenized_path + [model])
|
|
|
|
return '_'.join([action] + tokenized_path + [model])
|