PLC/Parameter.py

   1 #
   2 # Shared type definitions
   3 #
   4 # Mark Huang <mlhuang@cs.princeton.edu>
   5 # Copyright (C) 2006 The Trustees of Princeton University
   6 #
   7
   8 from types import *
   9 from PLC.Faults import *
  10
  11 class Parameter:
  12     """
  13     Typed value wrapper. Use in accepts and returns to document method
  14     parameters. Set the optional and default attributes for
  15     sub-parameters (i.e., dict fields).
  16     """
  17
  18     def __init__(self, typeval, doc = "",
  19                  min = None, max = None,
  20                  optional = None,
  21                  ro = False,
  22                  nullok = False):
  23         # Basic type of the parameter. Must be a builtin type
  24         # that can be marshalled by XML-RPC.
  25         self.type = typeval
  26
  27         # Documentation string for the parameter
  28         self.doc = doc
  29
  30         # Basic value checking. For numeric types, the minimum and
  31         # maximum possible values, inclusive. For string types, the
  32         # minimum and maximum possible UTF-8 encoded byte lengths.
  33         self.min = min
  34         self.max = max
  35
  36         # Whether the sub-parameter is optional or not. If None,
  37         # unknown whether it is optional.
  38         self.optional = optional
  39
  40         # Whether the DB field is read-only.
  41         self.ro = ro
  42
  43         # Whether the DB field can be NULL.
  44         self.nullok = nullok
  45
  46     def type(self):
  47         return self.type
  48
  49     def __repr__(self):
  50         return repr(self.type)
  51
  52 class Mixed(tuple):
  53     """
  54     A list (technically, a tuple) of types. Use in accepts and returns
  55     to document method parameters that may return mixed types.
  56     """
  57
  58     def __new__(cls, *types):
  59         return tuple.__new__(cls, types)
  60
  61 def python_type(arg):
  62     """
  63     Returns the Python type of the specified argument, which may be a
  64     Python type, a typed value, or a Parameter.
  65     """
  66
  67     if isinstance(arg, Parameter):
  68         arg = arg.type
  69
  70     if isinstance(arg, type):
  71         return arg
  72     else:
  73         return type(arg)
  74
  75 def xmlrpc_type(arg):
  76     """
  77     Returns the XML-RPC type of the specified argument, which may be a
  78     Python type, a typed value, or a Parameter.
  79     """
  80
  81     arg_type = python_type(arg)
  82
  83     if arg_type == NoneType:
  84         return "nil"
  85     elif arg_type == IntType or arg_type == LongType:
  86         return "int"
  87     elif arg_type == bool:
  88         return "boolean"
  89     elif arg_type == FloatType:
  90         return "double"
  91     elif arg_type in StringTypes:
  92         return "string"
  93     elif arg_type == ListType or arg_type == TupleType:
  94         return "array"
  95     elif arg_type == DictType:
  96         return "struct"
  97     elif arg_type == Mixed:
  98         # Not really an XML-RPC type but return "mixed" for
  99         # documentation purposes.
 100         return "mixed"
 101     else:
 102         raise PLCAPIError, "XML-RPC cannot marshal %s objects" % arg_type