2 # Functions for interacting with the nodes table in the database
4 # Mark Huang <mlhuang@cs.princeton.edu>
5 # Copyright (C) 2006 The Trustees of Princeton University
7 # $Id: Nodes.py,v 1.12 2006/10/20 17:51:32 mlhuang Exp $
10 from types import StringTypes
13 from PLC.Faults import *
14 from PLC.Parameter import Parameter
15 from PLC.Debug import profile
16 from PLC.Table import Row, Table
17 from PLC.NodeNetworks import NodeNetwork, NodeNetworks
18 from PLC.BootStates import BootStates
20 def valid_hostname(hostname):
21 # 1. Each part begins and ends with a letter or number.
22 # 2. Each part except the last can contain letters, numbers, or hyphens.
23 # 3. Each part is between 1 and 64 characters, including the trailing dot.
24 # 4. At least two parts.
25 # 5. Last part can only contain between 2 and 6 letters.
26 good_hostname = r'^([a-z0-9]([a-z0-9-]{0,61}[a-z0-9])?\.)+' \
29 re.match(good_hostname, hostname, re.IGNORECASE)
33 Representation of a row in the nodes table. To use, optionally
34 instantiate with a dict of values. Update as you would a
35 dict. Commit to the database with sync().
39 primary_key = 'node_id'
41 'node_id': Parameter(int, "Node identifier"),
42 'hostname': Parameter(str, "Fully qualified hostname", max = 255, optional = False),
43 'site_id': Parameter(int, "Site at which this node is located", optional = False),
44 'boot_state': Parameter(str, "Boot state", max = 20, optional = False),
45 'model': Parameter(str, "Make and model of the actual machine", max = 255),
46 'boot_nonce': Parameter(str, "(Admin only) Random value generated by the node at last boot", max = 128),
47 'version': Parameter(str, "Apparent Boot CD version", max = 64),
48 'ssh_rsa_key': Parameter(str, "Last known SSH host key", max = 1024),
49 'date_created': Parameter(str, "Date and time when node entry was created", ro = True),
50 'last_updated': Parameter(str, "Date and time when node entry was created", ro = True),
51 'key': Parameter(str, "(Admin only) Node key", max = 256),
52 'session': Parameter(str, "(Admin only) Node session value", max = 256),
53 'nodenetwork_ids': Parameter([int], "List of network interfaces that this node has", ro = True),
54 'nodegroup_ids': Parameter([int], "List of node groups that this node is in", ro = True),
55 'conf_file_ids': Parameter([int], "List of configuration files specific to this node", ro = True),
56 # 'root_person_ids': Parameter([int], "(Admin only) List of people who have root access to this node", ro = True),
57 'slice_ids': Parameter([int], "List of slices on this node", ro = True),
58 'pcu_ids': Parameter([int], "List of PCUs that control this node", ro = True),
59 'ports': Parameter([int], "List of PCU ports that this node is connected to", ro = True),
62 def validate_hostname(self, hostname):
63 if not valid_hostname(hostname):
64 raise PLCInvalidArgument, "Invalid hostname"
66 conflicts = Nodes(self.api, [hostname])
67 for node_id, node in conflicts.iteritems():
68 if 'node_id' not in self or self['node_id'] != node_id:
69 raise PLCInvalidArgument, "Hostname already in use"
73 def validate_boot_state(self, boot_state):
74 if boot_state not in BootStates(self.api):
75 raise PLCInvalidArgument, "Invalid boot state"
79 def delete(self, commit = True):
84 assert 'node_id' in self
86 # Delete all nodenetworks
87 nodenetworks = NodeNetworks(self.api, self['nodenetwork_ids'])
88 for nodenetwork in nodenetworks.values():
89 nodenetwork.delete(commit = False)
91 # Clean up miscellaneous join tables
92 for table in ['nodegroup_node', 'slice_node', 'slice_attribute']:
93 self.api.db.do("DELETE FROM %s" \
94 " WHERE node_id = %d" % \
95 (table, self['node_id']))
98 self['deleted'] = True
103 Representation of row(s) from the nodes table in the
107 def __init__(self, api, node_id_or_hostname_list = None):
110 sql = "SELECT %s FROM view_nodes WHERE deleted IS False" % \
111 ", ".join(Node.fields)
113 if node_id_or_hostname_list:
114 # Separate the list into integers and strings
115 node_ids = filter(lambda node_id: isinstance(node_id, (int, long)),
116 node_id_or_hostname_list)
117 hostnames = filter(lambda hostname: isinstance(hostname, StringTypes),
118 node_id_or_hostname_list)
121 sql += " OR node_id IN (%s)" % ", ".join(map(str, node_ids))
123 sql += " OR hostname IN (%s)" % ", ".join(api.db.quote(hostnames)).lower()
126 rows = self.api.db.selectall(sql)
129 self[row['node_id']] = node = Node(api, row)
130 for aggregate in ['nodenetwork_ids', 'nodegroup_ids',
131 'conf_file_ids', 'root_person_ids', 'slice_ids',
133 if not node.has_key(aggregate) or node[aggregate] is None:
136 node[aggregate] = map(int, node[aggregate].split(','))