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         'value': Parameter(str, "Slice attribute value"),
  32         }
  33
  34 class SliceTags(Table):
  35     """
  36     Representation of row(s) from the slice_tag table in the
  37     database.
  38     """
  39
  40     def __init__(self, api, slice_tag_filter = None, columns = None):
  41         Table.__init__(self, api, SliceTag, columns)
  42
  43         sql = "SELECT %s FROM view_slice_tags WHERE True" % \
  44               ", ".join(self.columns)
  45
  46         if slice_tag_filter is not None:
  47             if isinstance(slice_tag_filter, (list, tuple, set, int, long)):
  48                 slice_tag_filter = Filter(SliceTag.fields, {'slice_tag_id': slice_tag_filter})
  49             elif isinstance(slice_tag_filter, dict):
  50                 slice_tag_filter = Filter(SliceTag.fields, slice_tag_filter)
  51             else:
  52                 raise PLCInvalidArgument, "Wrong slice tag filter %r"%slice_tag_filter
  53             sql += " AND (%s) %s" % slice_tag_filter.sql(api)
  54
  55         self.selectall(sql)