PLC/NetworkTypes.py

   1 #
   2 # Functions for interacting with the network_types table in the database
   3 #
   4 #
   5 from types import StringTypes
   6 from PLC.Faults import *
   7 from PLC.Parameter import Parameter
   8 from PLC.Storage.AlchemyObject import AlchemyObj
   9
  10 class NetworkType(AlchemyObj):
  11     """
  12     Representation of a row in the network_types table. To use,
  13     instantiate with a dict of values.
  14     """
  15
  16     tablename = 'network_types'
  17     fields = {
  18         'type': Parameter(str, "Network type", max = 20, primary_key=True),
  19         }
  20
  21     def validate_type(self, name):
  22         # Make sure name is not blank
  23         if not len(name):
  24             raise PLCInvalidArgument, "Network type must be specified"
  25
  26         # Make sure network type does not alredy exist
  27         conflicts = NetworkTypes(self.api, name)
  28         if conflicts:
  29             raise PLCInvalidArgument, "Network type name already in use"
  30
  31         return name
  32
  33     def sync(self, commit=True, validate=True):
  34         AlchemyObj.sync(self, commit=commit, validate=validate)
  35         AlchemyObj.insert(self, dict(self))
  36
  37     def delete(self, commit=True):
  38         assert 'type' in self
  39         AlchemyObj.delete(self, dict(self))
  40
  41 class NetworkTypes(list):
  42     """
  43     Representation of the network_types table in the database.
  44     """
  45
  46     def __init__(self, api, types = None):
  47         if not types:
  48             network_types = NetworkType().select()
  49         elif isinstance(types, StringTypes):
  50             network_types = NetworkType().select({'type': types})
  51         else:
  52             raise PLCInvalidArgument, "Wrong network type filter %r"%types
  53
  54         self.extend(network_types)