X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=csit%2Flibraries%2FSxp.py;h=4aaf3b094f86c25a85ec6356351daf94163d699a;hb=43c30c4829b5f75d106a79d12029db2190b6ff5d;hp=9701cbc794e64d7630653152cebc68dbf7611320;hpb=ce2fd5ce86f724715c839ebe0878eb0e7806239b;p=integration%2Ftest.git diff --git a/csit/libraries/Sxp.py b/csit/libraries/Sxp.py index 9701cbc794..4aaf3b094f 100644 --- a/csit/libraries/Sxp.py +++ b/csit/libraries/Sxp.py @@ -1,18 +1,102 @@ import json -from ipaddr import IPAddress +from netaddr import IPAddress from string import Template +def get_active_controller_from_json(resp, service): + """Gets index of active controller running specified service + + :param resp: JSON formatted response from EOS + :type resp: str + :param service: EOS Service to look for + :type service: str + :return: Index of controller + """ + entities = json.loads(resp)['entity-owners']['entity-type'] + for entity in entities: + if entity['type'] == "org.opendaylight.mdsal.ServiceEntityType": + for instance in entity['entity']: + if service in instance['id']: + return int(instance['owner'][-1:]) + return 0 + + def mod(num, base): + """Gets modulo of number + + :param num: Number to be used + :type num: str + :param base: Base used + :type base: str + :returns: Int representing modulo of specified numbers. + + """ return int(num) % int(base) -def get_ip_from_number(n): - ip = IPAddress(2130706432 + n) +def get_average_of_items(items): + """Gets average of items in provided list + + :param items: To be proceed + :return: Average value + + """ + return sum(items) / len(items) + + +def get_opposing_mode(mode): + """Generate string representing opposing SXP peer mode + + :param mode: SXP peer mode + :type mode: str + :returns: String with opposing SXP peer mode. + + """ + if 'speaker' == mode: + return 'listener' + elif 'listener' == mode: + return 'speaker' + return 'both' + + +def get_ip_from_number(n, base=2130706432): + """Generate string representing Ipv4 from specified number plus base value + + :param n: Number to be converted + :type n: int + :param base: Starting index + :type base: int + :returns: String containing Ipv4. + + """ + ip = IPAddress(int(base) + n) + return str(ip) + + +def get_ip_from_number_and_ip(n, ip_address): + """Generate string representing Ipv4 from specified number and IPAddress + + :param n: Number to be converted + :type n: int + :param ip_address: Base address + :type ip_address: str + :returns: String containing Ipv4. + + """ + ip = IPAddress(int(IPAddress(ip_address)) + n) return str(ip) def lower_version(ver1, ver2): + """Generate xml containing SGT mach data + + :param ver1: Version of SXP protocol for compare + :type ver1: str + :param ver2: Version of SXP protocol for compare + :type ver2: str + :returns: String containing lower from those two specified versions. + + """ v1 = int(ver1[-1:]) v2 = int(ver2[-1:]) if v1 <= v2: @@ -21,19 +105,345 @@ def lower_version(ver1, ver2): return ver2 -def parse_connections(input): - data = json.loads(input) - connections = data['output']['connections'] +def get_filter_entry(seq, entry_type, sgt="", esgt="", acl="", eacl="", pl="", epl="", ps=""): + """Generate xml containing FilterEntry data + + :param seq: Sequence of entry + :type seq: str + :param entry_type: Type of entry (permit/deny) + :type entry_type: str + :param sgt: SGT matches to be added to entry + :type sgt: str + :param esgt: SGT ranges match to be added to entry + :type esgt: str + :param acl: ACL matches to be added to entry + :type acl: str + :param eacl: EACL matches to be added to entry + :type eacl: str + :param pl: PrefixList matches to be added to entry + :type pl: str + :param epl: ExtendedPrefixList matches to be added to entry + :type epl: str + :param ps: PeerSequence matches to be added to entry + :type ps: str + :returns: String containing xml data for request + + """ + entries = "" + # Generate XML request containing combination of Matches of different types + if sgt: + args = sgt.split(',') + entries += add_sgt_matches_xml(args) + elif esgt: + args = esgt.split(',') + entries += add_sgt_range_xml(args[0], args[1]) + if pl: + entries += add_pl_entry_xml(pl) + elif epl: + args = epl.split(',') + entries += add_epl_entry_xml(args[0], args[1], args[2]) + if acl: + args = acl.split(',') + entries += add_acl_entry_xml(args[0], args[1]) + elif eacl: + args = eacl.split(',') + entries += add_eacl_entry_xml(args[0], args[1], args[2], args[3]) + if ps: + args = ps.split(',') + entries += add_ps_entry_xml(args[0], args[1]) + # Wrap entries in ACL/PrefixList according to specified values + if pl or epl: + return add_pl_entry_default_xml(seq, entry_type, entries) + elif ps: + return add_ps_entry_default_xml(seq, entry_type, entries) + return add_acl_entry_default_xml(seq, entry_type, entries) + + +def add_peers(*args): + """Generate xml containing Peer mach data + + :param args: Peers data + :type args: dict + :returns: String containing xml data for request + + """ + templ = Template(''' + + $ip + ''') + peers = "" + for count, value in enumerate(args): + peers += templ.substitute({'ip': value}) + return peers + + +def add_domains(*args): + """Generate xml containing Domain mach data + + :param args: Domain data + :type args: dict + :returns: String containing xml data for request + + """ + templ = Template(''' + + $name + ''') + peers = "" + for count, value in enumerate(args): + peers += templ.substitute({'name': value}) + return peers + + +def add_sgt_matches_xml(sgt_entries): + """Generate xml containing SGT mach data + + :param sgt_entries: SGT matches + :type sgt_entries: str + :returns: String containing xml data for request + + """ + templ = Template(''' + $sgt''') + matches = "" + for sgt in sgt_entries: + matches += templ.substitute({'sgt': sgt}) + return matches + + +def add_sgt_range_xml(start, end): + """Generate xml containing SGT RangeMach data + + :param start: Start range of SGT + :type start: str + :param end: End range of SGT + :type end: str + :returns: String containing xml data for request + + """ + templ = Template(''' + $start + $end''') + match = templ.substitute({'start': start, 'end': end}) + return match + + +def add_acl_entry_default_xml(seq, entry_type, acl_entries): + """Generate xml containing AccessList data + + :param seq: Sequence of PrefixList entry + :type seq: str + :param entry_type: Entry type (permit/deny) + :type entry_type: str + :param acl_entries: XML data containing AccessList entries + :type acl_entries: str + :returns: String containing xml data for request + + """ + templ = Template(''' + + $entry_type + $seq$acl_entries + ''') + matches = templ.substitute( + {'seq': seq, 'entry_type': entry_type, 'acl_entries': acl_entries}) + return matches + + +def add_acl_entry_xml(ip, mask): + """Generate xml containing AccessList data + + :param ip: Ipv4/6 address + :type ip: str + :param mask: Ipv4/6 wildcard mask + :type mask: str + :returns: String containing xml data for request + + """ + templ = Template(''' + + $ip + $mask + ''') + return templ.substitute({'ip': ip, 'mask': mask}) + + +def add_eacl_entry_xml(ip, mask, amask, wmask): + """Generate xml containing ExtendedAccessList data + + :param ip: Ipv4/6 address + :type ip: str + :param mask: Ipv4/6 wildcard mask + :type mask: str + :param amask: Ipv4/6 address mask + :type amask: str + :param wmask: Ipv4/6 address wildcard mask + :type wmask: str + :returns: String containing xml data for request + + """ + templ = Template(''' + + $ip + $mask + + $amask + $wmask + + ''') + return templ.substitute({'ip': ip, 'mask': mask, 'amask': amask, 'wmask': wmask}) + + +def add_ps_entry_default_xml(seq, entry_type, ps_entries): + """Generate xml containing PeerSequence data + + :param seq: Sequence of PrefixList entry + :type seq: str + :param entry_type: Entry type (permit/deny) + :type entry_type: str + :param ps_entries: XML data containing PeerSequence entries + :type ps_entries: str + :returns: String containing xml data for request + + """ + templ = Template(''' + + $entry_type + $seq$ps_entries + ''') + return templ.substitute({'seq': seq, 'entry_type': entry_type, 'ps_entries': ps_entries}) + + +def add_pl_entry_default_xml(seq, entry_type, pl_entries): + """Generate xml containing PrefixList data + + :param seq: Sequence of PrefixList entry + :type seq: str + :param entry_type: Entry type (permit/deny) + :type entry_type: str + :param pl_entries: XML data containing PrefixList entries + :type pl_entries: str + :returns: String containing xml data for request + + """ + templ = Template(''' + + $entry_type + $seq$pl_entries + ''') + return templ.substitute({'seq': seq, 'entry_type': entry_type, 'pl_entries': pl_entries}) + + +def add_pl_entry_xml(prefix): + """Generate xml containing PrefixList data + + :param prefix: Ipv4/6 prefix + :type prefix: str + :returns: String containing xml data for request + + """ + templ = Template(''' + + $prefix + ''') + return templ.substitute({'prefix': prefix}) + + +def add_epl_entry_xml(prefix, op, mask): + """Generate xml containing Extended PrefixList data + + :param prefix: Ipv4/6 prefix + :type prefix: str + :param op: PrefixList option (ge/le/eq) + :type op: str + :param mask: Ipv4/6 Mask + :type mask: str + :returns: String containing xml data for request + + """ + templ = Template(''' + + $prefix + + $op + $mask + + ''') + return templ.substitute({'prefix': prefix, 'mask': mask, 'op': op}) + + +def add_ps_entry_xml(op, length): + """Generate xml containing Extended PrefixList data + + :param op: PrefixList option (ge/le/eq) + :type op: str + :param length: PeerSequence length + :type length: str + :returns: String containing xml data for request + + """ + templ = Template(''' + $length + $op + ''') + return templ.substitute({'length': length, 'op': op}) + + +def parse_peer_groups(groups_json): + """Parse JSON string into Array of PeerGroups + + :param groups_json: JSON containing PeerGroups + :type groups_json: str + :returns: Array containing PeerGroups. + + """ + data = json.loads(groups_json) + groups = data['output'] output = [] - for list in connections.values(): - output = output + list + for group in groups.values(): + output += group return output -def find_connection(input, version, mode, ip, port, state): - for connection in parse_connections(input): - if (connection['peer-address'] == ip and connection['tcp-port'] == int(port) and connection['mode'] == mode and - connection['version'] == version): +def parse_connections(connections_json): + """Parse JSON string into Array of Connections + + :param connections_json: JSON containing Connections + :type connections_json: str + :returns: Array containing Connections. + + """ + data = json.loads(connections_json) + output = data['output'] + result = [] + if output: + connections = output['connections'] + for connection in connections.values(): + result += connection + return result + + +def find_connection(connections_json, version, mode, ip, port, state): + """Test if Connection with specified values is contained in JSON + + :param connections_json: JSON containing Connections + :type connections_json: str + :param version: Version of SXP protocol (version1/2/3/4) + :type version: str + :param mode: Mode of SXP peer (speaker/listener/both) + :type mode: str + :param ip: Ipv4/6 address of remote peer + :type ip: str + :param port: Port on with remote peer listens + :type port: str + :param state: State of connection (on/off/pendingOn/deleteHoldDown) + :type state: str + :returns: True if Connection with specified params was found, otherwise False. + + """ + for connection in parse_connections(connections_json): + if (connection['peer-address'] == ip and connection['tcp-port'] == int(port) and ( + mode.strip() == 'any' or connection['mode'] == mode) and connection['version'] == version): if state == 'none': return True elif connection['state'] == state: @@ -41,21 +451,81 @@ def find_connection(input, version, mode, ip, port, state): return False -def parse_prefix_groups(input, source_): - data = json.loads(input) +def parse_bindings(bindings_json): + """Parse JSON string into Array of Bindings + + :param bindings_json: JSON containing Bindings + :type bindings_json: str + :returns: Array containing Bindings. + + """ + data = json.loads(bindings_json) + output = [] + for bindings_json in data['output'].values(): + for binding in bindings_json: + output.append(binding) + return output + + +def find_binding(bindings, sgt, prefix): + """Test if Binding with specified values is contained in JSON + + :param bindings: JSON containing Bindings + :type bindings: str + :param sgt: Source Group Tag + :type sgt: str + :param prefix: Ipv4/6 prefix + :type prefix: str + :returns: True if Binding with specified params was found, otherwise False. + + """ + for binding in parse_bindings(bindings): + if binding['sgt'] == int(sgt): + for ip_prefix in binding['ip-prefix']: + if ip_prefix == prefix: + return True + return False + + +def parse_prefix_groups(prefix_groups_json, source_): + """Parse JSON string into Array of PrefixGroups + + :param prefix_groups_json: JSON containing PrefixGroups + :type prefix_groups_json: str + :param source_: Source of PrefixGroups (sxp/local) + :type source_: str + :returns: Array containing PrefixGroups. + + """ + data = json.loads(prefix_groups_json) bindings = data['sxp-node:master-database'] output = [] for binding in bindings.values(): for binding_source in binding: - if binding_source['binding-source'] == source_: + if source_ == "any" or binding_source['binding-source'] == source_: for prefix_group in binding_source['prefix-group']: output.append(prefix_group) return output -def find_binding(input, sgt, prefix, source_, action): +def find_binding_legacy(prefix_groups_json, sgt, prefix, source_, action): + """Test if Binding with specified values is contained in JSON + + :param prefix_groups_json: JSON containing Bindings and PrefixGroups + :type prefix_groups_json: str + :param sgt: Source Group Tag + :type sgt: str + :param prefix: Ipv4/6 prefix + :type prefix: str + :param source_: Source of binding (local/sxp) + :type source_: str + :param action: Action for binding (add/delete) + :type action: str + :returns: True if Binding with specified params was found, otherwise False. + + """ found = False - for prefixgroup in parse_prefix_groups(input, source_): + for prefixgroup in parse_prefix_groups(prefix_groups_json, source_): if prefixgroup['sgt'] == int(sgt): for binding in prefixgroup['binding']: if binding['ip-prefix'] == prefix and binding['action'] == action: @@ -63,35 +533,33 @@ def find_binding(input, sgt, prefix, source_, action): return found -def find_binding_with_peer_sequence(input, sgt, prefix, source_, action, node_id, peer_seq): - correct_sequence = False - found_source = False - for prefixgroup in parse_prefix_groups(input, source_): - if prefixgroup['sgt'] == int(sgt): - for binding in prefixgroup['binding']: - if binding['ip-prefix'] == prefix and binding['action'] == action: - for peer in binding['peer-sequence']['peer']: - if peer['seq'] == int(peer_seq) and peer['node-id'] == node_id: - correct_sequence = True - for peer_source in binding['sources']['source']: - if peer_source == node_id: - found_source = True - return found_source and correct_sequence - - -def add_entry_xml(sgt, prefix, ip): - templ = Template(''' - $ip - $sgt - $prefix -''') - data = templ.substitute({'sgt': sgt, 'prefix': prefix, 'ip': ip}) - return data +def add_connection_xml(version, mode, ip, port, node, password_, domain_name, bindings_timeout=0, security_mode=''): + """Generate xml for Add Connection request + :param version: Version of SXP protocol (version1/2/3/4) + :type version: str + :param mode: Mode of SXP peer (speaker/listener/both) + :type mode: str + :param ip: Ipv4/6 address of remote peer + :type ip: str + :param port: Port on with remote peer listens + :type port: str + :param node: Ipv4 address of node + :type node: str + :param password_: Password type (none/default) + :type password_: str + :param domain_name: Name of Domain + :type domain_name: str + :param security_mode: Default/TSL security + :type security_mode: str + :param bindings_timeout: Specifies DHD and Reconciliation timers + :type bindings_timeout: int + :returns: String containing xml data for request -def add_connection_xml(version, mode, ip, port, ip_, password_): + """ templ = Template(''' - $ip_ + $node + $domain $ip @@ -100,58 +568,100 @@ def add_connection_xml(version, mode, ip, port, ip_, password_): $mode $version Connection to ISR-G2 + $security_type 45 30 - 120 + $timeout + $timeout + 90 + 180 + 90 ''') data = templ.substitute( - {'ip': ip, 'port': port, 'mode': mode, 'version': version, 'ip_': ip_, 'password_': password_}) + {'ip': ip, 'port': port, 'mode': mode, 'version': version, 'node': node, + 'password_': password_, 'domain': get_domain_name(domain_name), 'timeout': bindings_timeout, + 'security_type': '' + security_mode + '' if security_mode else ''}) return data -def delete_connections_xml(address, port, node): +def delete_connections_xml(address, port, node, domain_name): + """Generate xml for Delete Connection request + + :param address: Ipv4/6 address of remote peer + :type address: str + :param port: Port on with remote peer listens + :type port: str + :param node: Ipv4 address of node + :type node: str + :param domain_name: Name of Domain + :type domain_name: str + :returns: String containing xml data for request + + """ templ = Template(''' $node + $domain $address $port ''') - data = templ.substitute({'address': address, 'port': port, 'node': node}) + data = templ.substitute({'address': address, 'port': port, 'node': node, 'domain': get_domain_name(domain_name)}) return data -def update_binding_xml(sgt0, prefix0, sgt1, prefix1, ip): +def add_peer_group_xml(name, peers, ip): + """Generate xml for Add PeerGroups request + + :param name: Name of PeerGroup + :type name: str + :param peers: XML formatted peers that will be added to group + :type peers: str + :param ip: Ipv4 address of node + :type ip: str + :returns: String containing xml data for request + + """ templ = Template(''' $ip - - $sgt0 - $prefix0 - - - $sgt1 - $prefix1 - + + $name + $peers + ''') - data = templ.substitute( - {'sgt0': sgt0, 'sgt1': sgt1, 'prefix0': prefix0, 'prefix1': prefix1, 'ip': ip}) + data = templ.substitute({'name': name, 'peers': peers, 'ip': ip}) return data -def delete_binding_xml(sgt, prefix, ip): +def delete_peer_group_xml(name, ip): + """Generate xml for Delete PeerGroup request + + :param name: Name of PeerGroup + :type name: str + :param ip: Ipv4 address of node + :type ip: str + :returns: String containing xml data for request + + """ templ = Template(''' $ip - $sgt - $prefix + $name ''') - data = templ.substitute({'sgt': sgt, 'prefix': prefix, 'ip': ip}) + data = templ.substitute({'name': name, 'ip': ip}) return data -def get_connections_from_node_xml(ip): +def get_peer_groups_from_node_xml(ip): + """Generate xml for Get PeerGroups request + + :param ip: Ipv4 address of node + :type ip: str + :returns: String containing xml data for request + + """ templ = Template(''' $ip ''') @@ -159,9 +669,610 @@ def get_connections_from_node_xml(ip): return data -def get_bindings_from_node_xml(ip): +def add_filter_xml(group, filter_type, entries, ip, policy=None): + """Generate xml for Add Filter request + + :param group: Name of group containing filter + :type group: str + :param filter_type: Type of filter + :type filter_type: str + :param entries: XML formatted entries that will be added in filter + :type entries: str + :param ip: Ipv4 address of node + :type ip: str + :param policy: Policy of filter update mechanism + :type policy: str + :returns: String containing xml data for request + + """ + if policy: + policy = "" + policy + "" + else: + policy = "" templ = Template(''' $ip + $group + + $filter_policy + $filter_type$entries + ''') - data = templ.substitute({'ip': ip}) + data = templ.substitute( + {'group': group, 'filter_type': filter_type, 'ip': ip, 'entries': entries, 'filter_policy': policy}) + return data + + +def add_domain_filter_xml(domain, domains, entries, ip, filter_name=None): + """Generate xml for Add Domain Filter request + + :param domain: Name of Domain containing filter + :type domain: str + :param domains: Domains on which filter will be applied + :type domains: str + :param entries: XML formatted entries that will be added in filter + :type entries: str + :param ip: Ipv4 address of node + :type ip: str + :param filter_name: Name of filter + :type filter_name: str + :returns: String containing xml data for request + + """ + if filter_name: + filter_name = "" + filter_name + "" + templ = Template(''' + $ip + $domain + + $filter_name + $domains + $entries + +''') + data = templ.substitute( + {'domain': domain, 'domains': domains, 'ip': ip, 'entries': entries, 'filter_name': filter_name}) + return data + + +def delete_filter_xml(group, filter_type, ip): + """Generate xml for Delete Filter request + + :param group: Name of group containing filter + :type group: str + :param filter_type: Type of filter + :type filter_type: str + :param ip: Ipv4 address of node + :type ip: str + :returns: String containing xml data for request + + """ + templ = Template(''' + $ip + $group + $filter_type +''') + data = templ.substitute( + {'group': group, 'filter_type': filter_type, 'ip': ip}) + return data + + +def delete_domain_filter_xml(domain, ip, filter_name=None): + """Generate xml for Delete Filter request + + :param domain: Name of Domain containing filter + :type domain: str + :param ip: Ipv4 address of node + :type ip: str + :param filter_name: Name of filter + :type filter_name: str + :returns: String containing xml data for request + + """ + if filter_name: + filter_name = '' + filter_name + "" + templ = Template(''' + $ip + $domain + $filter_name +''') + data = templ.substitute( + {'domain': domain, 'ip': ip, 'filter_name': filter_name}) + return data + + +def get_connections_from_node_xml(ip, domain_name): + """Generate xml for Get Connections request + + :param ip: Ipv4 address of node + :type ip: str + :param domain_name: Name of Domain + :type domain_name: str + :returns: String containing xml data for request + + """ + templ = Template(''' + $ip + $domain +''') + data = templ.substitute({'ip': ip, 'domain': get_domain_name(domain_name)}) + return data + + +def get_bindings_from_node_xml(ip, binding_range, domain_name): + """Generate xml for Get Bindings request + + :param binding_range: All or only Local bindings + :type binding_range: str + :param ip: Ipv4 address of node + :type ip: str + :param domain_name: Name of Domain + :type domain_name: str + :returns: String containing xml data for request + + """ + templ = Template(''' + $ip + $range + $domain +''') + data = templ.substitute({'ip': ip, 'range': binding_range, 'domain': get_domain_name(domain_name)}) + return data + + +def add_node_xml(node_id, port, password, version, node_ip=None, expansion=0, bindings_timeout=0, keystores=None, + retry_open_timer=1): + """Generate xml for Add Node request + + :param node_id: Ipv4 address formatted node id + :type node_id: str + :param node_ip: Ipv4 address of node + :type node_ip: strl + :param port: Node port number + :type port: int + :param password: TCP-MD5 password + :type password: str + :param version: Sxp device version + :type version: str + :param expansion: Bindings expansion + :type expansion: int + :param bindings_timeout: Specifies DHD and Reconciliation timers + :type bindings_timeout: int + :param keystores: SSL keystore and truststore specification + :type keystores: dict + :returns: String containing xml data for request + + """ + tls = '' + if keystores: + tls = Template(''' + + + $keystore + JKS + PATH + $passwd + + + $truststore + JKS + PATH + $passwd + + $passwd + + ''').substitute( + {'keystore': keystores['keystore'], 'truststore': keystores['truststore'], 'passwd': keystores['password']}) + + templ = Template(''' + $id + + $retry_open_timer + 120 + $timeout + 90 + $timeout + 90 + 180 + 30 + + $expansion + + $tls + $password + + $port + $version + ODL SXP Controller + $ip +''') + data = templ.substitute( + {'ip': node_ip or node_id, 'id': node_id, 'port': port, 'password': password, + 'version': version, 'expansion': expansion, 'timeout': bindings_timeout, 'tls': tls, + 'retry_open_timer': retry_open_timer}) return data + + +def delete_node_xml(node_id): + """Generate xml for Delete node request + + :param node_id: Ipv4 address formatted node id + :type node_id: str + :returns: String containing xml data for request + + """ + templ = Template(''' + $id +''') + data = templ.substitute({'id': node_id}) + return data + + +def add_domain_xml_fluorine(node_id, name, sgt, prefixes, origin): + """Generate xml for Add Domain request (Fluorine version: bindings with origin) + + :param node_id: Id of node + :type node_id: str + :param name: Name of Domain + :type name: str + :param sgt: Security group + :type sgt: int + :param prefixes: List of ip-prefixes + :type prefixes: str + :param origin: Origin of added bindings + :type origin: str + :returns: String containing xml data for request + + """ + master_database = '' + if prefixes != 'None': + xml_prefixes = '' + for prefix in prefixes.split(','): + xml_prefixes += '\n' + '' + prefix + '' + if xml_prefixes: + master_database += ''' + + $sgt + $xml_prefixes + + ''' + master_database = Template(master_database).substitute(({'sgt': sgt, 'xml_prefixes': xml_prefixes})) + + templ = Template(''' + $id + $name + $origin + $master_database +''') + + data = templ.substitute({'name': name, 'id': node_id, 'origin': origin, 'master_database': master_database}) + return data + + +def add_domain_xml_oxygen(node_id, name, sgt, prefixes): + """Generate xml for Add Domain request (Oxygen version: bindings without origin) + + :param node_id: Id of node + :type node_id: str + :param name: Name of Domain + :type name: str + :param sgt: Security group + :type sgt: int + :param prefixes: List of ip-prefixes + :type prefixes: str + :returns: String containing xml data for request + + """ + master_database = '' + if prefixes != 'None': + xml_prefixes = '' + for prefix in prefixes.split(','): + xml_prefixes += '\n' + '' + prefix + '' + if xml_prefixes: + master_database += ''' + + $sgt + $xml_prefixes + + ''' + master_database = Template(master_database).substitute(({'sgt': sgt, 'xml_prefixes': xml_prefixes})) + + templ = Template(''' + $id + $name + $master_database +''') + + data = templ.substitute({'name': name, 'id': node_id, 'master_database': master_database}) + return data + + +def delete_domain_xml(node_id, name): + """Generate xml for Remove Domain request + + :param node_id: Id of node + :type node_id: str + :param name: Name of Domain + :type name: str + :returns: String containing xml data for request + + """ + templ = Template(''' + $node_id + $name +''') + + data = templ.substitute({'node_id': node_id, 'name': name}) + return data + + +def get_domain_name(domain_name): + """Generate xml for Get Bindings request + + :param domain_name: Name of Domain + :type domain_name: str + :returns: String containing xml data for request + + """ + if domain_name == 'global': + return '' + else: + return '' + domain_name + '' + + +def add_bindings_xml_fluorine(node_id, domain, sgt, prefixes, origin): + """Generate xml for Add Bindings request (Fluorine version with origin type) + + :param node_id: Id of node + :type node_id: str + :param domain: Name of Domain + :type domain: str + :param sgt: Security group + :type sgt: int + :param prefixes: List of ip-prefixes + :type prefixes: str + :param origin: Origin of added bindings + :type origin: str + :returns: String containing xml data for request + + """ + xml_prefixes = '' + for prefix in prefixes.split(','): + xml_prefixes += '\n' + '' + prefix + '' + templ = Template(''' + $id + $name + $origin + + + $sgt + $xml_prefixes + + +''') + data = templ.substitute({'name': domain, 'id': node_id, 'sgt': sgt, 'xml_prefixes': xml_prefixes, 'origin': origin}) + return data + + +def add_bindings_xml_oxygen(node_id, domain, sgt, prefixes): + """Generate xml for Add Bindings request (Oxygen version without origin type) + + :param node_id: Id of node + :type node_id: str + :param domain: Name of Domain + :type domain: str + :param sgt: Security group + :type sgt: int + :param prefixes: List of ip-prefixes + :type prefixes: str + :returns: String containing xml data for request + + """ + xml_prefixes = '' + for prefix in prefixes.split(','): + xml_prefixes += '\n' + '' + prefix + '' + templ = Template(''' + $id + $name + + $sgt + $xml_prefixes + +''') + data = templ.substitute({'name': domain, 'id': node_id, 'sgt': sgt, 'xml_prefixes': xml_prefixes}) + return data + + +def delete_bindings_xml(node_id, domain, sgt, prefixes): + """Generate xml for Remove Bindings request + + :param node_id: Id of node + :type node_id: str + :param domain: Name of Domain + :type domain: str + :param sgt: Security group + :type sgt: int + :param prefixes: Comma separated list of ip-prefixes + :type prefixes: str + :returns: String containing xml data for request + + """ + xml_prefixes = '' + for prefix in prefixes.split(','): + xml_prefixes += '\n' + '' + prefix + '' + templ = Template(''' + $id + $name + + $sgt + $xml_prefixes + +''') + data = templ.substitute({'name': domain, 'id': node_id, 'sgt': sgt, 'xml_prefixes': xml_prefixes}) + return data + + +def prefix_range(start, end): + """Generate and concatenate ip-prefixes + + :param start: Start index + :type start: str + :param end: End index + :type end: str + :returns: String containing concatenated ip-prefixes + + """ + start = int(start) + end = int(end) + index = 0 + prefixes = '' + while index < end: + prefixes += get_ip_from_number(index + start) + '/32' + index += 1 + if index < end: + prefixes += ',' + return prefixes + + +def route_definition_xml(virtual_ip, net_mask, interface): + """Generate xml for Add Bindings request + + :param interface: Network interface name + :type interface: str + :param net_mask: NetMask of virtual ip + :type net_mask: str + :param virtual_ip: Virtual ip + :type virtual_ip: str + :returns: String containing xml data for request + + """ + templ = Template(''' + + $vip + $interface + $mask + + ''') + data = templ.substitute({'mask': net_mask, 'vip': virtual_ip, 'interface': interface}) + return data + + +def route_definitions_xml(routes, old_routes=None): + """Generate xml for Add Bindings request + + :param routes: XML formatted data containing RouteDefinitions + :type routes: str + :param old_routes: Routes add to request that needs to persist + :type old_routes: str + :returns: String containing xml data for request + + """ + if old_routes and "" in old_routes: + templ = Template(old_routes.replace("", "$routes")) + else: + templ = Template(''' + $routes + + ''') + data = templ.substitute({'routes': routes}) + return data + + +def add_binding_origin_xml(origin, priority): + """Generate xml for Add Binding Origin request + + :param origin: Origin type + :type origin: str + :param priority: Origin priority + :type priority: str + :returns: String containing xml data for request + + """ + templ = Template(''' + $origin + $priority +''') + data = templ.substitute({'origin': origin, 'priority': priority}) + return data + + +def update_binding_origin_xml(origin, priority): + """Generate xml for Update Binding Origin request + + :param origin: Origin type + :type origin: str + :param priority: Origin priority + :type priority: str + :returns: String containing xml data for request + + """ + templ = Template(''' + $origin + $priority +''') + data = templ.substitute({'origin': origin, 'priority': priority}) + return data + + +def delete_binding_origin_xml(origin): + """Generate xml for Delete Binding Origin request + + :param origin: Origin type + :type origin: str + :returns: String containing xml data for request + + """ + templ = Template(''' + $origin +''') + data = templ.substitute({'origin': origin}) + return data + + +def find_binding_origin(origins_json, origin): + """Test if Binding origin of specified value is contained in JSON + + :param origins_json: JSON containing Binding origins + :type origins_json: str + :param origin: Origin to be found + :type origin: str + :returns: True if Binding origin of specified origin type was found, otherwise False. + + """ + for json_origin in parse_binding_origins(origins_json): + if json_origin['origin'] == origin: + return True + return False + + +def find_binding_origin_with_priority(origins_json, origin, priority): + """Test if Binding origin of specified value and priority is contained in JSON + + :param origins_json: JSON containing Binding origins + :type origins_json: str + :param origin: Origin to be found + :type origin: str + :param priority: desired priority of origin + :type priority: str + :returns: True if Binding origin of specified origin type with desired priority was found, otherwise False. + + """ + for json_origin in parse_binding_origins(origins_json): + if json_origin['origin'] == origin: + if json_origin['priority'] == int(priority): + return True + return False + + +def parse_binding_origins(origins_json): + """Parse JSON string into Array of Binding origins + + :param origins_json: JSON containing Binding origins + :type origins_json: str + :returns: Array containing Binding origins. + + """ + output = [] + for origins in origins_json['binding-origins'].values(): + for origin in origins: + output.append(origin) + return output