1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
# Copyright (c) 2016 Mirantis, Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
YAQLIZATION_ATTR = '__yaqlization__'
def yaqlize(class_or_object=None, yaqlize_attributes=True,
yaqlize_methods=True, yaqlize_indexer=True,
auto_yaqlize_result=False, whitelist=None, blacklist=None,
attribute_remapping=None, blacklist_remapped_attributes=True):
def func(something):
if not hasattr(something, YAQLIZATION_ATTR):
setattr(something, YAQLIZATION_ATTR, build_yaqlization_settings(
yaqlize_attributes=yaqlize_attributes,
yaqlize_methods=yaqlize_methods,
yaqlize_indexer=yaqlize_indexer,
auto_yaqlize_result=auto_yaqlize_result,
whitelist=whitelist,
blacklist=blacklist,
attribute_remapping=attribute_remapping,
))
return something
if class_or_object is None:
return func
else:
return func(class_or_object)
def get_yaqlization_settings(class_or_object):
return getattr(class_or_object, YAQLIZATION_ATTR, None)
def is_yaqlized(class_or_object):
return hasattr(class_or_object, YAQLIZATION_ATTR)
def build_yaqlization_settings(
yaqlize_attributes=True, yaqlize_methods=True, yaqlize_indexer=True,
auto_yaqlize_result=False, whitelist=None, blacklist=None,
attribute_remapping=None, blacklist_remapped_attributes=True):
whitelist = set(whitelist or [])
blacklist = set(blacklist or [])
attribute_remapping = attribute_remapping or {}
if blacklist_remapped_attributes:
for value in attribute_remapping.values():
if not isinstance(value, str):
name = value[0]
else:
name = value
blacklist.add(name)
return {
'yaqlizeAttributes': yaqlize_attributes,
'yaqlizeMethods': yaqlize_methods,
'yaqlizeIndexer': yaqlize_indexer,
'autoYaqlizeResult': auto_yaqlize_result,
'whitelist': whitelist,
'blacklist': blacklist,
'attributeRemapping': attribute_remapping
}
|
