X-Git-Url: http://dolda2000.com/gitweb/?p=wrw.git;a=blobdiff_plain;f=wrw%2Futil.py;h=6fa4155b6a423e0cc9f7501d1461df3fad76f806;hp=557b25d684c8a57e861e479fd3002ce0aaae4fff;hb=HEAD;hpb=438b6207201da45a2f23e6ff207a61a0f57b20f3 diff --git a/wrw/util.py b/wrw/util.py index 557b25d..6fa4155 100644 --- a/wrw/util.py +++ b/wrw/util.py @@ -8,18 +8,27 @@ def wsgiwrap(callable): return wrapper def formparams(callable): - spec = inspect.getargspec(callable) + sig = inspect.signature(callable) + haskw = inspect.Parameter.VAR_KEYWORD in (par.kind for par in sig.parameters.values()) def wrapper(req): - data = form.formdata(req) - args = dict(data.items()) - args["req"] = req - if not spec.keywords: - for arg in list(args): - if arg not in spec.args: - del args[arg] - for i in range(len(spec.args) - (len(spec.defaults) if spec.defaults else 0)): - if spec.args[i] not in args: - raise resp.httperror(400, "Missing parameter", ("The query parameter `", resp.h.code(spec.args[i]), "' is required but not supplied.")) + try: + data = dict(form.formdata(req).items()) + except IOError: + raise resp.httperror(400, "Invalid request", "Form data was incomplete") + + data["req"] = req + if haskw: + args = data + else: + args = {} + for par in sig.parameters.values(): + if par.name in data: + args[par.name] = data[par.name] + for par in sig.parameters.values(): + if par.kind is inspect.Parameter.VAR_KEYWORD: + continue + if par.default is inspect.Parameter.empty and par.name not in args: + raise resp.httperror(400, "Missing parameter", ("The query parameter `", resp.h.code(par.name), "' is required but not supplied.")) return callable(**args) wrapper.__wrapped__ = callable return wrapper @@ -38,6 +47,8 @@ class funplex(object): def __call__(self, req): if req.pathinfo == "": + if "__root__" in self.dir: + return self.dir["__root__"](req) raise resp.redirect(req.uriname + "/") if req.pathinfo[:1] != "/": raise resp.notfound() @@ -49,7 +60,9 @@ class funplex(object): p = p.partition("/")[0] bi = len(p) + 1 if p in self.dir: - return self.dir[p](req.shift(bi)) + sreq = req.shift(bi) + sreq.selfpath = req.pathinfo[1:] + return self.dir[p](sreq) raise resp.notfound() def add(self, fun):