@@ -47,24 +47,25 @@ def is_mapped_instance(cls):
47
47
48
48
def to_type_name (name ):
49
49
"""Convert the given name to a GraphQL type name."""
50
- return '' .join (part [:1 ].upper () + part [1 :] for part in name .split ('_' ))
50
+ return "" .join (part [:1 ].upper () + part [1 :] for part in name .split ("_" ))
51
51
52
52
53
- _re_enum_value_name_1 = re .compile (' (.)([A-Z][a-z]+)' )
54
- _re_enum_value_name_2 = re .compile (' ([a-z0-9])([A-Z])' )
53
+ _re_enum_value_name_1 = re .compile (" (.)([A-Z][a-z]+)" )
54
+ _re_enum_value_name_2 = re .compile (" ([a-z0-9])([A-Z])" )
55
55
56
56
57
57
def to_enum_value_name (name ):
58
58
"""Convert the given name to a GraphQL enum value name."""
59
59
return _re_enum_value_name_2 .sub (
60
- r'\1_\2' , _re_enum_value_name_1 .sub (r'\1_\2' , name )).upper ()
60
+ r"\1_\2" , _re_enum_value_name_1 .sub (r"\1_\2" , name )
61
+ ).upper ()
61
62
62
63
63
64
def default_symbol_name (column_name , is_asc ):
64
65
return to_enum_value_name (column_name ) + ("_ASC" if is_asc else "_DESC" )
65
66
66
67
67
- def plain_symbol_name (column_name , is_asc ):
68
+ def plain_symbol_name (column_name , is_asc ): # pragma: no cover
68
69
return column_name + ("_asc" if is_asc else "_desc" )
69
70
70
71
@@ -80,7 +81,8 @@ def __init__(self, str_value, value):
80
81
81
82
82
83
def create_sort_enum_for_model (
83
- cls , name = None , symbol_name = default_symbol_name , registry = None ):
84
+ cls , name = None , symbol_name = default_symbol_name , registry = None
85
+ ):
84
86
"""Create a Graphene Enum type for defining a sort order for the given model class.
85
87
86
88
The created Enum type and sort order will then be registered for that class.
@@ -104,6 +106,7 @@ def create_sort_enum_for_model(
104
106
name = cls .__name__ + "SortEnum"
105
107
if registry is None :
106
108
from .registry import get_global_registry
109
+
107
110
registry = get_global_registry ()
108
111
members = OrderedDict ()
109
112
default_sort = []
@@ -134,18 +137,23 @@ def get_sort_enum_for_model(cls, registry=None):
134
137
"""
135
138
if registry is None :
136
139
from .registry import get_global_registry
140
+
137
141
registry = get_global_registry ()
138
142
sort_params = registry .get_sort_params_for_model (cls )
139
143
if not sort_params :
140
144
sort_params = create_sort_enum_for_model (cls , registry = registry )
141
145
return sort_params [0 ]
142
146
143
147
144
- def sort_enum_for_model (cls , name = None , symbol_name = plain_symbol_name ):
148
+ def sort_enum_for_model (
149
+ cls , name = None , symbol_name = plain_symbol_name
150
+ ): # pragma: no cover
145
151
warnings .warn (
146
152
"sort_argument_for_model() is deprecated;"
147
153
" use get_sort_argument_for_model() and create_sort_argument_for_model()" ,
148
- DeprecationWarning , stacklevel = 2 )
154
+ DeprecationWarning ,
155
+ stacklevel = 2 ,
156
+ )
149
157
if not name and not symbol_name :
150
158
return get_sort_enum_for_model (cls )
151
159
sort_params = create_sort_enum_for_model (cls , name , symbol_name )
@@ -161,6 +169,7 @@ def get_sort_argument_for_model(cls, has_default=True, registry=None):
161
169
"""
162
170
if registry is None :
163
171
from .registry import get_global_registry
172
+
164
173
registry = get_global_registry ()
165
174
sort_params = registry .get_sort_params_for_model (cls )
166
175
if not sort_params :
@@ -171,8 +180,10 @@ def get_sort_argument_for_model(cls, has_default=True, registry=None):
171
180
return Argument (List (enum ), default_value = default )
172
181
173
182
174
- def sort_argument_for_model (cls , has_default = True ):
183
+ def sort_argument_for_model (cls , has_default = True ): # pragma: no cover
175
184
warnings .warn (
176
185
"sort_argument_for_model() is deprecated; use get_sort_argument_for_model()." ,
177
- DeprecationWarning , stacklevel = 2 )
186
+ DeprecationWarning ,
187
+ stacklevel = 2 ,
188
+ )
178
189
return get_sort_argument_for_model (cls , has_default = has_default )
0 commit comments