1 #
2 # Copyright (C) 2007 Stefan Seefeld
3 # All rights reserved.
4 # For license terms see the file COPYING.txt.
5 #
7 from roundup import hyperdb
8 from roundup.cgi.exceptions import *
9 from roundup.exceptions import UsageError
10 from roundup.date import Date, Range, Interval
11 from roundup import actions
12 from SimpleXMLRPCServer import *
13 from xmlrpclib import Binary
15 def translate(value):
16 """Translate value to becomes valid for XMLRPC transmission."""
18 if isinstance(value, (Date, Range, Interval)):
19 return repr(value)
20 elif type(value) is list:
21 return [translate(v) for v in value]
22 elif type(value) is tuple:
23 return tuple([translate(v) for v in value])
24 elif type(value) is dict:
25 return dict([[translate(k), translate(value[k])] for k in value])
26 else:
27 return value
30 def props_from_args(db, cl, args, itemid=None):
31 """Construct a list of properties from the given arguments,
32 and return them after validation."""
34 props = {}
35 for arg in args:
36 if isinstance(arg, Binary):
37 arg = arg.data
38 try :
39 key, value = arg.split('=', 1)
40 except ValueError :
41 raise UsageError, 'argument "%s" not propname=value'%arg
42 if isinstance(key, unicode):
43 try:
44 key = key.encode ('ascii')
45 except UnicodeEncodeError:
46 raise UsageError, 'argument %r is no valid ascii keyword'%key
47 if isinstance(value, unicode):
48 value = value.encode('utf-8')
49 if value:
50 try:
51 props[key] = hyperdb.rawToHyperdb(db, cl, itemid,
52 key, value)
53 except hyperdb.HyperdbValueError, message:
54 raise UsageError, message
55 else:
56 props[key] = None
58 return props
60 class RoundupInstance:
61 """The RoundupInstance provides the interface accessible through
62 the Python XMLRPC mapping."""
64 def __init__(self, db, actions, translator):
66 self.db = db
67 self.actions = actions
68 self.translator = translator
70 def schema(self):
71 s = {}
72 for c in self.db.classes:
73 cls = self.db.classes[c]
74 props = [(n,repr(v)) for n,v in cls.properties.items()]
75 s[c] = props
76 return s
78 def list(self, classname, propname=None):
79 cl = self.db.getclass(classname)
80 if not propname:
81 propname = cl.labelprop()
82 result = [cl.get(itemid, propname)
83 for itemid in cl.list()
84 if self.db.security.hasPermission('View', self.db.getuid(),
85 classname, propname, itemid)
86 ]
87 return result
89 def filter(self, classname, search_matches, filterspec,
90 sort=[], group=[]):
91 cl = self.db.getclass(classname)
92 result = cl.filter(search_matches, filterspec, sort=sort, group=group)
93 return result
95 def display(self, designator, *properties):
96 classname, itemid = hyperdb.splitDesignator(designator)
97 cl = self.db.getclass(classname)
98 props = properties and list(properties) or cl.properties.keys()
99 props.sort()
100 for p in props:
101 if not self.db.security.hasPermission('View', self.db.getuid(),
102 classname, p, itemid):
103 raise Unauthorised('Permission to view %s of %s denied'%
104 (p, designator))
105 result = [(prop, cl.get(itemid, prop)) for prop in props]
106 return dict(result)
108 def create(self, classname, *args):
110 if not self.db.security.hasPermission('Create', self.db.getuid(), classname):
111 raise Unauthorised('Permission to create %s denied'%classname)
113 cl = self.db.getclass(classname)
115 # convert types
116 props = props_from_args(self.db, cl, args)
118 # check for the key property
119 key = cl.getkey()
120 if key and not props.has_key(key):
121 raise UsageError, 'you must provide the "%s" property.'%key
123 for key in props:
124 if not self.db.security.hasPermission('Create', self.db.getuid(),
125 classname, property=key):
126 raise Unauthorised('Permission to create %s.%s denied'%(classname, key))
128 # do the actual create
129 try:
130 result = cl.create(**props)
131 self.db.commit()
132 except (TypeError, IndexError, ValueError), message:
133 raise UsageError, message
134 return result
136 def set(self, designator, *args):
138 classname, itemid = hyperdb.splitDesignator(designator)
139 cl = self.db.getclass(classname)
140 props = props_from_args(self.db, cl, args, itemid) # convert types
141 for p in props.iterkeys():
142 if not self.db.security.hasPermission('Edit', self.db.getuid(),
143 classname, p, itemid):
144 raise Unauthorised('Permission to edit %s of %s denied'%
145 (p, designator))
146 try:
147 result = cl.set(itemid, **props)
148 self.db.commit()
149 except (TypeError, IndexError, ValueError), message:
150 raise UsageError, message
151 return result
154 builtin_actions = {'retire': actions.Retire}
156 def action(self, name, *args):
157 """Execute a named action."""
159 if name in self.actions:
160 action_type = self.actions[name]
161 elif name in self.builtin_actions:
162 action_type = self.builtin_actions[name]
163 else:
164 raise Exception('action "%s" is not supported %s' % (name, ','.join(self.actions.keys())))
165 action = action_type(self.db, self.translator)
166 return action.execute(*args)
169 class RoundupDispatcher(SimpleXMLRPCDispatcher):
170 """RoundupDispatcher bridges from cgi.client to RoundupInstance.
171 It expects user authentication to be done."""
173 def __init__(self, db, actions, translator,
174 allow_none=False, encoding=None):
176 try:
177 # python2.5 and beyond
178 SimpleXMLRPCDispatcher.__init__(self, allow_none, encoding)
179 except TypeError:
180 # python2.4
181 SimpleXMLRPCDispatcher.__init__(self)
182 self.register_instance(RoundupInstance(db, actions, translator))
185 def dispatch(self, input):
186 return self._marshaled_dispatch(input)
188 def _dispatch(self, method, params):
190 retn = SimpleXMLRPCDispatcher._dispatch(self, method, params)
191 retn = translate(retn)
192 return retn