X-Git-Url: http://dolda2000.com/gitweb/?p=wrw.git;a=blobdiff_plain;f=wrw%2Futil.py;h=7209ae42d9d96bc39f1b01594b49d1d4afb3db2b;hp=7c42eaed6c25bc620edc0f114b364c1056798f3c;hb=HEAD;hpb=62113fc6dabec39f06a1f86f4ea6f73975f0acdf diff --git a/wrw/util.py b/wrw/util.py index 7c42eae..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): @@ -267,6 +280,6 @@ class specdirty(sessiondata, metaclass=specclass): def datecheck(req, mtime): if "If-Modified-Since" in req.ihead: rtime = proto.phttpdate(req.ihead["If-Modified-Since"]) - if rtime >= math.floor(mtime): + if rtime is not None and rtime >= math.floor(mtime): raise resp.unmodified() req.ohead["Last-Modified"] = proto.httpdate(mtime)