# Mark Huang <mlhuang@cs.princeton.edu>
# Copyright (C) 2006 The Trustees of Princeton University
#
-# $Id: PCUs.py,v 1.6 2006/10/25 14:29:13 mlhuang Exp $
-#
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.NodeNetworks import valid_ip, NodeNetwork, NodeNetworks
+from PLC.Interfaces import valid_ip, Interface, Interfaces
from PLC.Nodes import Node, Nodes
class PCU(Row):
'password': Parameter(str, "PCU username", max = 254, nullok = True),
'notes': Parameter(str, "Miscellaneous notes", max = 254, nullok = True),
'model': Parameter(str, "PCU model string", max = 32, nullok = True),
- 'node_ids': Parameter([int], "List of nodes that this PCU controls", ro = True),
- 'ports': Parameter([int], "List of the port numbers that each node is connected to", ro = True),
+ 'node_ids': Parameter([int], "List of nodes that this PCU controls"),
+ 'ports': Parameter([int], "List of the port numbers that each node is connected to"),
+ 'last_updated': Parameter(int, "Date and time when node entry was created", ro = True),
}
def validate_ip(self, ip):
if not valid_ip(ip):
- raise PLCInvalidArgument, "Invalid IP address " + ip
+ raise PLCInvalidArgument("Invalid IP address " + ip)
return ip
+ validate_last_updated = Row.validate_timestamp
+
+ def update_timestamp(self, col_name, commit = True):
+ """
+ Update col_name field with current time
+ """
+
+ assert 'pcu_id' in self
+ assert self.table_name
+
+ self.api.db.do("UPDATE %s SET %s = CURRENT_TIMESTAMP " % (self.table_name, col_name) + \
+ " where pcu_id = %d" % (self['pcu_id']) )
+ self.sync(commit)
+
+ def update_last_updated(self, commit = True):
+ self.update_timestamp('last_updated', commit)
+
def add_node(self, node, port, commit = True):
"""
Add node to existing PCU.
assert 'pcu_id' in self
assert isinstance(node, Node)
- assert isinstance(port, (int, long))
+ assert isinstance(port, int)
assert 'node_id' in node
pcu_id = self['pcu_id']
database.
"""
- def __init__(self, api, pcu_ids = None):
- self.api = api
-
- # N.B.: Node IDs returned may be deleted.
- sql = "SELECT %s FROM view_pcus" % \
- ", ".join(PCU.fields)
+ def __init__(self, api, pcu_filter = None, columns = None):
+ Table.__init__(self, api, PCU, columns)
- if pcu_ids:
- sql += " WHERE pcu_id IN (%s)" % ", ".join(map(str, pcu_ids))
+ sql = "SELECT %s FROM view_pcus WHERE True" % \
+ ", ".join(self.columns)
- rows = self.api.db.selectall(sql)
+ if pcu_filter is not None:
+ if isinstance(pcu_filter, (list, tuple, set, int)):
+ pcu_filter = Filter(PCU.fields, {'pcu_id': pcu_filter})
+ elif isinstance(pcu_filter, dict):
+ pcu_filter = Filter(PCU.fields, pcu_filter)
+ else:
+ raise PLCInvalidArgument("Wrong pcu filter %r"%pcu_filter)
+ sql += " AND (%s) %s" % pcu_filter.sql(api)
- for row in rows:
- self[row['pcu_id']] = pcu = PCU(api, row)
- for aggregate in ['node_ids', 'ports']:
- if not pcu.has_key(aggregate) or pcu[aggregate] is None:
- pcu[aggregate] = []
- else:
- pcu[aggregate] = map(int, pcu[aggregate].split(','))
+ self.selectall(sql)