1 # Copyright (C) 2021-2022 Wind River Systems, Inc.
3 # Licensed under the Apache License, Version 2.0 (the "License");
4 # you may not use this file except in compliance with the License.
5 # You may obtain a copy of the License at
7 # http://www.apache.org/licenses/LICENSE-2.0
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS,
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
15 from sqlalchemy.sql.elements import ColumnElement
16 from sqlalchemy import or_
18 from o2ims.domain.alarm_obj import AlarmEventRecord
19 from o2ims.domain.ocloud import Ocloud
21 from o2common.helper import o2logging
22 logger = o2logging.get_logger(__name__)
25 def gen_orm_filter(obj: ColumnElement, filter_str: str):
28 # filter_without_space = filter_str.replace(" ", "")
29 filter_without_space = filter_str.strip(' ()')
30 items = filter_without_space.split(';')
35 # i = i.replace("(", "")
37 # i = i.replace(")", "")
38 filter_expr = i.split(',')
39 if len(filter_expr) < 3:
41 filter_op = filter_expr[0].strip()
42 filter_key = filter_expr[1].strip()
43 filter_vals = filter_expr[2:]
44 filter_list.extend(toFilterArgs(
45 filter_op, obj, filter_key, filter_vals))
46 logger.debug('Filter list length: %d' % len(filter_list))
50 def toFilterArgs(operation: str, obj: ColumnElement, key: str, values: list):
51 # if not hasattr(obj, key):
52 # logger.warning('Filter attrName %s not in Object %s' %
55 # 'Filter attrName {} not in the Object'.format(key))
57 # if operation in ['eq', 'neq', 'gt', 'lt', 'gte', 'lte']:
58 # if len(values) != 1:
60 # 'Filter operation one {} is only support one value'.
62 # elif operation in ['in', 'nin', 'cont', 'ncont']:
63 # if len(values) == 0:
64 # raise KeyError('Filter operation {} value is needed'.
67 # raise KeyError('Filter operation {} not support'.format(operation))
68 key = transfer_filter_attr_name_in_special(obj, key)
73 if val.lower() == 'null':
75 ll.append(getattr(obj, key) == val)
76 elif operation == 'neq':
78 if val.lower() == 'null':
80 ll.append(getattr(obj, key) != val)
81 elif operation == 'gt':
83 ll.append(getattr(obj, key) > val)
84 elif operation == 'lt':
86 ll.append(getattr(obj, key) < val)
87 elif operation == 'gte':
89 ll.append(getattr(obj, key) >= val)
90 elif operation == 'lte':
92 ll.append(getattr(obj, key) <= val)
93 elif operation == 'in':
94 ll.append(getattr(obj, key).in_(values))
95 elif operation == 'nin':
96 ll.append(~getattr(obj, key).in_(values))
97 elif operation == 'cont':
100 val_list.append(getattr(obj, key).contains(val))
101 ll.append(or_(*val_list))
102 elif operation == 'ncont':
105 val_list.append(getattr(obj, key).contains(val))
106 ll.append(~or_(*val_list))
110 def transfer_filter_attr_name_in_special(obj: ColumnElement, filter_key: str):
111 if obj == AlarmEventRecord:
112 if filter_key == 'resourceTypeID':
113 filter_key = 'resourceTypeId'
114 elif filter_key == 'resourceID':
115 filter_key = 'resourceId'
116 elif filter_key == 'alarmDefinitionID':
117 filter_key = 'alarmDefinitionId'
118 elif filter_key == 'probableCauseID':
119 filter_key = 'probableCauseId'
121 if filter_key == 'globalcloudId':
122 filter_key = 'globalCloudId'