PLC/NodeTags.py

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