X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=PLC%2FNodeNetworks.py;h=75dddf6792ef831a1ac41c8ec8ce94af1e05f14c;hb=7f71c47a861497aba265bdbc9edb212ed79f08f2;hp=0bdf8c493ab70287bf86ce41f49908bdb4df32dc;hpb=f6f0164afcbada9498622842964024f6fb3e5dfa;p=plcapi.git diff --git a/PLC/NodeNetworks.py b/PLC/NodeNetworks.py index 0bdf8c4..75dddf6 100644 --- a/PLC/NodeNetworks.py +++ b/PLC/NodeNetworks.py @@ -4,7 +4,7 @@ # Mark Huang # Copyright (C) 2006 The Trustees of Princeton University # -# $Id: NodeNetworks.py,v 1.10 2006/10/24 20:02:22 mlhuang Exp $ +# $Id$ # from types import StringTypes @@ -13,6 +13,7 @@ import struct from PLC.Faults import * from PLC.Parameter import Parameter +from PLC.Filter import Filter from PLC.Debug import profile from PLC.Table import Row, Table from PLC.NetworkTypes import NetworkType, NetworkTypes @@ -47,40 +48,47 @@ class NodeNetwork(Row): table_name = 'nodenetworks' primary_key = 'nodenetwork_id' + join_tables = ['nodenetwork_setting'] fields = { 'nodenetwork_id': Parameter(int, "Node interface identifier"), 'method': Parameter(str, "Addressing method (e.g., 'static' or 'dhcp')"), 'type': Parameter(str, "Address type (e.g., 'ipv4')"), - 'ip': Parameter(str, "IP address"), - 'mac': Parameter(str, "MAC address"), - 'gateway': Parameter(str, "IP address of primary gateway"), - 'network': Parameter(str, "Subnet address"), - 'broadcast': Parameter(str, "Network broadcast address"), - 'netmask': Parameter(str, "Subnet mask"), - 'dns1': Parameter(str, "IP address of primary DNS server"), - 'dns2': Parameter(str, "IP address of secondary DNS server"), - 'bwlimit': Parameter(int, "Bandwidth limit", min = 0), - 'hostname': Parameter(str, "(Optional) Hostname"), + 'ip': Parameter(str, "IP address", nullok = True), + 'mac': Parameter(str, "MAC address", nullok = True), + 'gateway': Parameter(str, "IP address of primary gateway", nullok = True), + 'network': Parameter(str, "Subnet address", nullok = True), + 'broadcast': Parameter(str, "Network broadcast address", nullok = True), + 'netmask': Parameter(str, "Subnet mask", nullok = True), + 'dns1': Parameter(str, "IP address of primary DNS server", nullok = True), + 'dns2': Parameter(str, "IP address of secondary DNS server", nullok = True), + 'bwlimit': Parameter(int, "Bandwidth limit", min = 0, nullok = True), + 'hostname': Parameter(str, "(Optional) Hostname", nullok = True), 'node_id': Parameter(int, "Node associated with this interface"), 'is_primary': Parameter(bool, "Is the primary interface for this node"), + 'nodenetwork_setting_ids' : Parameter([int], "List of nodenetwork settings"), } def validate_method(self, method): - if method not in NetworkMethods(self.api): - raise PLCInvalidArgument, "Invalid addressing method" + network_methods = [row['method'] for row in NetworkMethods(self.api)] + if method not in network_methods: + raise PLCInvalidArgument, "Invalid addressing method %s"%method return method def validate_type(self, type): - if type not in NetworkTypes(self.api): - raise PLCInvalidArgument, "Invalid address type" + network_types = [row['type'] for row in NetworkTypes(self.api)] + if type not in network_types: + raise PLCInvalidArgument, "Invalid address type %s"%type return type def validate_ip(self, ip): if ip and not valid_ip(ip): - raise PLCInvalidArgument, "Invalid IP address " + ip + raise PLCInvalidArgument, "Invalid IP address %s"%ip return ip def validate_mac(self, mac): + if not mac: + return mac + try: bytes = mac.split(":") if len(bytes) < 6: @@ -92,7 +100,7 @@ class NodeNetwork(Row): bytes[i] = "%02x" % byte mac = ":".join(bytes) except: - raise PLCInvalidArgument, "Invalid MAC address" + raise PLCInvalidArgument, "Invalid MAC address %s"%mac return mac @@ -103,20 +111,29 @@ class NodeNetwork(Row): validate_dns1 = validate_ip validate_dns2 = validate_ip + def validate_bwlimit(self, bwlimit): + if not bwlimit: + return bwlimit + + if bwlimit < 500000: + raise PLCInvalidArgument, 'Minimum bw is 500 kbs' + + return bwlimit + def validate_hostname(self, hostname): # Optional if not hostname: return hostname if not PLC.Nodes.valid_hostname(hostname): - raise PLCInvalidArgument, "Invalid hostname" + raise PLCInvalidArgument, "Invalid hostname %s"%hostname return hostname def validate_node_id(self, node_id): nodes = PLC.Nodes.Nodes(self.api, [node_id]) if not nodes: - raise PLCInvalidArgument, "No such node" + raise PLCInvalidArgument, "No such node %d"%node_id return node_id @@ -126,15 +143,16 @@ class NodeNetwork(Row): """ if is_primary: - nodes = PLC.Nodes.Nodes(self.api, [self['node_id']]).values() + nodes = PLC.Nodes.Nodes(self.api, [self['node_id']]) if not nodes: - raise PLCInvalidArgument, "No such node" + raise PLCInvalidArgument, "No such node %d"%node_id node = nodes[0] if node['nodenetwork_ids']: conflicts = NodeNetworks(self.api, node['nodenetwork_ids']) - for nodenetwork_id, nodenetwork in conflicts.iteritems(): - if ('nodenetwork_id' not in self or self['nodenetwork_id'] != nodenetwork_id) and \ + for nodenetwork in conflicts: + if ('nodenetwork_id' not in self or \ + self['nodenetwork_id'] != nodenetwork['nodenetwork_id']) and \ nodenetwork['is_primary']: raise PLCInvalidArgument, "Can only set one primary interface per node" @@ -164,7 +182,10 @@ class NodeNetwork(Row): # DHCP! elif method == "static": - for key in ['ip', 'gateway', 'network', 'broadcast', 'netmask', 'dns1']: + if 'gateway' not in self or self['gateway'] is None: + if 'is_primary' in self and self['is_primary'] is True: + raise PLCInvalidArgument, "For static method primary network, gateway is required" + for key in ['ip', 'network', 'broadcast', 'netmask', 'dns1']: if key not in self or not self[key]: raise PLCInvalidArgument, "For static method, %s is required" % key globals()[key] = self[key] @@ -174,7 +195,7 @@ class NodeNetwork(Row): if not in_same_network(broadcast, network, netmask): raise PLCInvalidArgument, "Broadcast address %s is inconsistent with network %s/%s" % \ (broadcast, network, netmask) - if not in_same_network(ip, gateway, netmask): + if 'gateway' in globals() and not in_same_network(ip, gateway, netmask): raise PLCInvalidArgument, "Gateway %s is not reachable from %s/%s" % \ (gateway, ip, netmask) @@ -188,26 +209,21 @@ class NodeNetworks(Table): database. """ - def __init__(self, api, nodenetwork_id_or_ip_list = None): - self.api = api - - sql = "SELECT %s FROM nodenetworks" % \ - ", ".join(NodeNetwork.fields) - - if nodenetwork_id_or_ip_list: - # Separate the list into integers and strings - nodenetwork_ids = filter(lambda nodenetwork_id: isinstance(nodenetwork_id, (int, long)), - nodenetwork_id_or_ip_list) - ips = filter(lambda ip: isinstance(ip, StringTypes), - nodenetwork_id_or_ip_list) - sql += " WHERE (False" - if nodenetwork_ids: - sql += " OR nodenetwork_id IN (%s)" % ", ".join(map(str, nodenetwork_ids)) - if ips: - sql += " OR ip IN (%s)" % ", ".join(api.db.quote(ips)).lower() - sql += ")" - - rows = self.api.db.selectall(sql) - - for row in rows: - self[row['nodenetwork_id']] = NodeNetwork(api, row) + def __init__(self, api, nodenetwork_filter = None, columns = None): + Table.__init__(self, api, NodeNetwork, columns) + + sql = "SELECT %s FROM view_nodenetworks WHERE True" % \ + ", ".join(self.columns) + + if nodenetwork_filter is not None: + if isinstance(nodenetwork_filter, (list, tuple, set)): + nodenetwork_filter = Filter(NodeNetwork.fields, {'nodenetwork_id': nodenetwork_filter}) + elif isinstance(nodenetwork_filter, dict): + nodenetwork_filter = Filter(NodeNetwork.fields, nodenetwork_filter) + elif isinstance(nodenetwork_filter, int): + nodenetwork_filter = Filter(NodeNetwork.fields, {'nodenetwork_id': [nodenetwork_filter]}) + else: + raise PLCInvalidArgument, "Wrong node network filter %r"%nodenetwork_filter + sql += " AND (%s) %s" % nodenetwork_filter.sql(api) + + self.selectall(sql)