PLC/SliceTags.py

   1 # $Id$
   2 # $URL$
   3 from PLC.Faults import *
   4 from PLC.Parameter import Parameter
   5 from PLC.Filter import Filter
   6 from PLC.Table import Row, Table
   7 # seems to cause import loops
   8 #from PLC.Slices import Slice, Slices
   9 from PLC.Nodes import Node, Nodes
  10 from PLC.NodeGroups import NodeGroup, NodeGroups
  11 from PLC.TagTypes import TagType, TagTypes
  12
  13 class SliceTag(Row):
  14     """
  15     Representation of a row in the slice_tag table. To use,
  16     instantiate with a dict of values.
  17     """
  18
  19     table_name = 'slice_tag'
  20     primary_key = 'slice_tag_id'
  21     fields = {
  22         'slice_tag_id': Parameter(int, "Slice tag identifier"),
  23         'slice_id': Parameter(int, "Slice identifier"),
  24         'name': Parameter(str, "Slice name"),
  25         'node_id': Node.fields['node_id'],
  26         'nodegroup_id': NodeGroup.fields['nodegroup_id'],
  27         'tag_type_id': TagType.fields['tag_type_id'],
  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         '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)