PLC/SliceTags.py

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