PLC/NodeTags.py

   1 # $Id$
   2 #
   3 # Thierry Parmentelat - INRIA
   4 #
   5 # $Revision: 9423 $
   6 #
   7 from PLC.Faults import *
   8 from PLC.Parameter import Parameter
   9 from PLC.Filter import Filter
  10 from PLC.Table import Row, Table
  11 from PLC.Nodes import Node, Nodes
  12 from PLC.TagTypes import TagType, TagTypes
  13
  14 class NodeTag(Row):
  15     """
  16     Representation of a row in the node_tag.
  17     To use, instantiate with a dict of values.
  18     """
  19
  20     table_name = 'node_tag'
  21     primary_key = 'node_tag_id'
  22     fields = {
  23         'node_tag_id': Parameter(int, "Node tag identifier"),
  24         'node_id': Node.fields['node_id'],
  25         'hostname' : Node.fields['hostname'],
  26         'tag_type_id': TagType.fields['tag_type_id'],
  27         'value': Parameter(str, "Node tag value"),
  28         'tagname': TagType.fields['tagname'],
  29         'description': TagType.fields['description'],
  30         'category': TagType.fields['category'],
  31         'min_role_id': TagType.fields['min_role_id'],
  32         }
  33
  34 class NodeTags(Table):
  35     """
  36     Representation of row(s) from the node_tag table in the
  37     database.
  38     """
  39
  40     def __init__(self, api, node_tag_filter = None, columns = None):
  41         Table.__init__(self, api, NodeTag, columns)
  42
  43         sql = "SELECT %s FROM view_node_tags WHERE True" % \
  44               ", ".join(self.columns)
  45
  46         if node_tag_filter is not None:
  47             if isinstance(node_tag_filter, (list, tuple, set)):
  48                 node_tag_filter = Filter(NodeTag.fields, {'node_tag_id': node_tag_filter})
  49             elif isinstance(node_tag_filter, dict):
  50                 node_tag_filter = Filter(NodeTag.fields, node_tag_filter)
  51             elif isinstance(node_tag_filter, int):
  52                 node_tag_filter = Filter(NodeTag.fields, {'node_tag_id': [node_tag_filter]})
  53             else:
  54                 raise PLCInvalidArgument, "Wrong node tag filter %r"%node_tag_filter
  55             sql += " AND (%s) %s" % node_tag_filter.sql(api)
  56
  57
  58         self.selectall(sql)