Matteo Scandolo | d2044a4 | 2017-08-07 16:08:28 -0700 | [diff] [blame] | 1 | # Copyright 2017-present Open Networking Foundation |
| 2 | # |
| 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 |
| 6 | # |
| 7 | # http://www.apache.org/licenses/LICENSE-2.0 |
| 8 | # |
| 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. |
| 14 | |
Zack Williams | 9a42f87 | 2019-02-15 17:56:04 -0700 | [diff] [blame] | 15 | from __future__ import absolute_import |
| 16 | from .base import xproto_string_type, unquote |
Matteo Scandolo | 292cc2a | 2017-07-31 19:02:12 -0700 | [diff] [blame] | 17 | |
Zack Williams | 045b63d | 2019-01-22 16:30:57 -0700 | [diff] [blame] | 18 | |
Matteo Scandolo | 292cc2a | 2017-07-31 19:02:12 -0700 | [diff] [blame] | 19 | def xproto_type_to_ui_type(f): |
| 20 | try: |
Zack Williams | 045b63d | 2019-01-22 16:30:57 -0700 | [diff] [blame] | 21 | content_type = f["options"]["content_type"] |
Matteo Scandolo | 292cc2a | 2017-07-31 19:02:12 -0700 | [diff] [blame] | 22 | content_type = eval(content_type) |
Zack Williams | 045b63d | 2019-01-22 16:30:57 -0700 | [diff] [blame] | 23 | except BaseException: |
Matteo Scandolo | 292cc2a | 2017-07-31 19:02:12 -0700 | [diff] [blame] | 24 | content_type = None |
| 25 | pass |
| 26 | |
Zack Williams | 045b63d | 2019-01-22 16:30:57 -0700 | [diff] [blame] | 27 | if "choices" in f["options"]: |
| 28 | return "select" |
| 29 | elif content_type == "date": |
| 30 | return "date" |
| 31 | elif f["type"] == "bool": |
| 32 | return "boolean" |
| 33 | elif f["type"] == "string": |
| 34 | return xproto_string_type(f["options"]) |
| 35 | elif f["type"] in ["int", "uint32", "int32"] or "link" in f: |
| 36 | return "number" |
| 37 | elif f["type"] in ["double", "float"]: |
| 38 | return "string" |
| 39 | |
Matteo Scandolo | 292cc2a | 2017-07-31 19:02:12 -0700 | [diff] [blame] | 40 | |
| 41 | def xproto_options_choices_to_dict(choices): |
| 42 | list = [] |
| 43 | |
| 44 | for c in eval(choices): |
Zack Williams | 045b63d | 2019-01-22 16:30:57 -0700 | [diff] [blame] | 45 | list.append({"id": c[0], "label": c[1]}) |
Matteo Scandolo | 292cc2a | 2017-07-31 19:02:12 -0700 | [diff] [blame] | 46 | if len(list) > 0: |
| 47 | return list |
| 48 | else: |
| 49 | return None |
| 50 | |
Zack Williams | 045b63d | 2019-01-22 16:30:57 -0700 | [diff] [blame] | 51 | |
Zack Williams | 9a42f87 | 2019-02-15 17:56:04 -0700 | [diff] [blame] | 52 | def xproto_dict_to_sorted_string(d): |
| 53 | """ |
| 54 | sorts the dict by key and returns a string representation, which makes |
| 55 | for better consistency when testing |
| 56 | """ |
| 57 | ft = [] # formatted tuples |
| 58 | for k, v in sorted(d.items(), key=lambda t: t[0]): # sorted by key |
| 59 | if v is not None: |
| 60 | ft.append("'%s': '%s'" % (k, v)) |
| 61 | else: |
| 62 | ft.append("'%s': None" % k) |
| 63 | return "{%s}" % ", ".join(ft) |
| 64 | |
| 65 | |
Matteo Scandolo | 292cc2a | 2017-07-31 19:02:12 -0700 | [diff] [blame] | 66 | def xproto_validators(f): |
Matteo Scandolo | 8ef264f | 2019-05-01 15:26:39 -0700 | [diff] [blame] | 67 | |
Matteo Scandolo | 292cc2a | 2017-07-31 19:02:12 -0700 | [diff] [blame] | 68 | # To be cleaned up when we formalize validation in xproto |
| 69 | validators = [] |
| 70 | |
| 71 | # bound-based validators |
Zack Williams | 045b63d | 2019-01-22 16:30:57 -0700 | [diff] [blame] | 72 | bound_validators = [("max_length", "maxlength"), ("min", "min"), ("max", "max")] |
Matteo Scandolo | 292cc2a | 2017-07-31 19:02:12 -0700 | [diff] [blame] | 73 | |
| 74 | for v0, v1 in bound_validators: |
| 75 | try: |
Zack Williams | 045b63d | 2019-01-22 16:30:57 -0700 | [diff] [blame] | 76 | validators.append({"name": v1, "int_value": int(f["options"][v0])}) |
Matteo Scandolo | 292cc2a | 2017-07-31 19:02:12 -0700 | [diff] [blame] | 77 | except KeyError: |
| 78 | pass |
| 79 | |
| 80 | # validators based on content_type |
Zack Williams | 045b63d | 2019-01-22 16:30:57 -0700 | [diff] [blame] | 81 | content_type_validators = ["ip", "url", "email"] |
Matteo Scandolo | 292cc2a | 2017-07-31 19:02:12 -0700 | [diff] [blame] | 82 | |
| 83 | for v in content_type_validators: |
Zack Williams | 045b63d | 2019-01-22 16:30:57 -0700 | [diff] [blame] | 84 | # if f['name']=='ip': pdb.set_trace() |
Matteo Scandolo | 292cc2a | 2017-07-31 19:02:12 -0700 | [diff] [blame] | 85 | try: |
Zack Williams | 045b63d | 2019-01-22 16:30:57 -0700 | [diff] [blame] | 86 | val = unquote(f["options"]["content_type"]) == v |
Matteo Scandolo | 292cc2a | 2017-07-31 19:02:12 -0700 | [diff] [blame] | 87 | if not val: |
| 88 | raise KeyError |
| 89 | |
Zack Williams | 045b63d | 2019-01-22 16:30:57 -0700 | [diff] [blame] | 90 | validators.append({"name": v, "bool_value": True}) |
Matteo Scandolo | 292cc2a | 2017-07-31 19:02:12 -0700 | [diff] [blame] | 91 | except KeyError: |
| 92 | pass |
| 93 | |
| 94 | # required validator |
| 95 | try: |
Matteo Scandolo | 8ef264f | 2019-05-01 15:26:39 -0700 | [diff] [blame] | 96 | required = f["modifier"] == "required" |
Matteo Scandolo | 292cc2a | 2017-07-31 19:02:12 -0700 | [diff] [blame] | 97 | if required: |
Zack Williams | 045b63d | 2019-01-22 16:30:57 -0700 | [diff] [blame] | 98 | validators.append({"name": "required", "bool_value": required}) |
Matteo Scandolo | 292cc2a | 2017-07-31 19:02:12 -0700 | [diff] [blame] | 99 | except KeyError: |
| 100 | pass |
| 101 | |
| 102 | return validators |
| 103 | |
Zack Williams | 045b63d | 2019-01-22 16:30:57 -0700 | [diff] [blame] | 104 | |
Matteo Scandolo | 292cc2a | 2017-07-31 19:02:12 -0700 | [diff] [blame] | 105 | def is_number(s): |
| 106 | try: |
| 107 | float(s) |
| 108 | return True |
| 109 | except ValueError: |
| 110 | return False |
| 111 | |
Zack Williams | 045b63d | 2019-01-22 16:30:57 -0700 | [diff] [blame] | 112 | |
Matteo Scandolo | 292cc2a | 2017-07-31 19:02:12 -0700 | [diff] [blame] | 113 | def xproto_default_to_gui(default): |
Scott Baker | 4839dec | 2019-02-27 16:50:37 -0800 | [diff] [blame] | 114 | # TODO: Using `eval` here is potentially dangerous as it may allow code injection |
Matteo Scandolo | 292cc2a | 2017-07-31 19:02:12 -0700 | [diff] [blame] | 115 | val = "null" |
Scott Baker | 4839dec | 2019-02-27 16:50:37 -0800 | [diff] [blame] | 116 | try: |
| 117 | if is_number(default): |
| 118 | val = str(default) |
| 119 | elif eval(default) is True: |
| 120 | val = "true" |
| 121 | elif eval(default) is False: |
| 122 | val = "false" |
| 123 | elif eval(default) is None: |
| 124 | val = "null" |
| 125 | else: |
| 126 | val = str(default) |
| 127 | except NameError: |
| 128 | # val was a function call, and we can't pass those to the GUI |
Zack Williams | 045b63d | 2019-01-22 16:30:57 -0700 | [diff] [blame] | 129 | val = "null" |
Scott Baker | 4839dec | 2019-02-27 16:50:37 -0800 | [diff] [blame] | 130 | |
Matteo Scandolo | 1f826a4 | 2017-08-02 12:02:02 -0700 | [diff] [blame] | 131 | return val |
| 132 | |
| 133 | |
| 134 | def xproto_links_to_modeldef_relations(llst): |
| 135 | outlist = [] |
| 136 | seen = [] |
| 137 | for l in llst: |
| 138 | try: |
Zack Williams | 045b63d | 2019-01-22 16:30:57 -0700 | [diff] [blame] | 139 | t = l["link_type"] |
| 140 | except KeyError as e: |
Matteo Scandolo | 1f826a4 | 2017-08-02 12:02:02 -0700 | [diff] [blame] | 141 | raise e |
| 142 | |
Zack Williams | 045b63d | 2019-01-22 16:30:57 -0700 | [diff] [blame] | 143 | if l["peer"]["fqn"] not in seen and t != "manytomany": |
| 144 | on_field = "null" |
| 145 | if l["link_type"] == "manytoone": |
| 146 | on_field = l["src_port"] |
| 147 | elif l["link_type"] == "onetomany": |
| 148 | on_field = l["dst_port"] |
| 149 | outlist.append( |
Zack Williams | be5ee1c | 2019-03-18 15:33:07 -0700 | [diff] [blame] | 150 | "{model: %s, type: %s, on_field: %s}" |
Zack Williams | 045b63d | 2019-01-22 16:30:57 -0700 | [diff] [blame] | 151 | % (l["peer"]["name"], l["link_type"], on_field) |
| 152 | ) |
| 153 | seen.append(l["peer"]) |
Matteo Scandolo | 1f826a4 | 2017-08-02 12:02:02 -0700 | [diff] [blame] | 154 | |
Sapan Bhatia | bfb233a | 2018-02-09 14:53:09 -0800 | [diff] [blame] | 155 | return outlist |