X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=o2common%2Fviews%2Fview.py;fp=o2common%2Fviews%2Fview.py;h=d3906342f85414fc90634a61f0a1b7fb44cd0b07;hb=356dc5488966c011e8a49fc87a8b824fba39b18b;hp=2b0502794fa29f718ea14c64cc0d049833db59cd;hpb=5f6c39c6bfb90907a289be7d02210be7f2ea22dd;p=pti%2Fo2.git diff --git a/o2common/views/view.py b/o2common/views/view.py index 2b05027..d390634 100644 --- a/o2common/views/view.py +++ b/o2common/views/view.py @@ -12,7 +12,7 @@ # See the License for the specific language governing permissions and # limitations under the License. -import re +# import re from sqlalchemy.sql.elements import ColumnElement from o2common.views.route_exception import BadRequestException @@ -44,33 +44,49 @@ def gen_filter(obj: ColumnElement, filter_str: str): def check_filter(obj: ColumnElement, filter_str: str): if not filter_str: return - pattern = r'^(\((eq|neq|gt|lt|gte|lte){1},\w+,[\w -\.]+\)\;?|' +\ - r'\((in|nin|cont|ncont){1},\w*(,[\w -\.]*)*\)\;?)+' - result = re.match(pattern, filter_str) - logger.warning('filter: {} match result is {}'.format(filter_str, result)) - if not result: - raise BadRequestException( - 'filter value formater not correct.') + # pattern = r'^(\((eq|neq|gt|lt|gte|lte){1},\w+,[\w -\.]+\)\;?|' +\ + # r'\((in|nin|cont|ncont){1},\w*(,[\w -\.]*)*\)\;?)+' + # result = re.match(pattern, filter_str) + # logger.debug('filter: {} match result is {}'.format(filter_str, result)) + # if not result: + # raise BadRequestException( + # 'filter value format is invalid') check_filter_attribute(obj, filter_str) def check_filter_attribute(obj: ColumnElement, filter_str: str): - filter_without_space = filter_str.replace(" ", "") + # filter_without_space = filter_str.replace(" ", "") + filter_without_space = filter_str.strip(' ()') + logger.debug( + f"filter_str: {filter_str}, stripped: {filter_without_space}") items = filter_without_space.split(';') for i in items: - if '(' in i: - i = i.replace("(", "") - if ')' in i: - i = i.replace(")", "") + # if '(' in i: + # i = i.replace("(", "") + # if ')' in i: + # i = i.replace(")", "") filter_expr = i.split(',') if len(filter_expr) < 3: raise BadRequestException( - 'Filter {} formater not correct.'.format(i)) + 'ignore invalid filter {}'.format(i)) continue - # filter_op = filter_expr[0] - filter_key = filter_expr[1] - # filter_vals = filter_expr[2:] + filter_op = filter_expr[0].strip() + filter_key = filter_expr[1].strip() + filter_vals = filter_expr[2:] + if filter_op in ["eq", "neq", "gt", "lt", "gte", "lte"]: + if len(filter_vals) != 1: + raise BadRequestException( + "Found {} values: {} while only single value" + " is allowed for operation {}".format( + len(filter_vals), filter_vals, filter_op) + ) + elif filter_op not in ["in", "nin", "cont", "ncont"]: + raise BadRequestException( + 'Filter operation {} is invalid'.format(filter_op) + ) + else: + pass if not hasattr(obj, filter_key): raise BadRequestException( - 'Filter attrName {} not in the Object'.format(filter_key)) + 'Filter attrName {} is invalid'.format(filter_key))