路由分组
from wsgiref.simple_server import make_server import cgi from urllib import parse import webob from webob.dec import wsgify from webob.exc import HTTPNotFound import re class Route: ROUTETABLE = [] def __init__(self, prefix:str=''): self.__prefix = prefix.rstrip('/\\') self.__routetable = [] def route(self, pattern, *methods): def wraper(handler): self.__routetable.append((tuple(map(lambda x:x.upper(), methods)), re.compile(pattern), handler)) return handler return wraper def get(self, pattern): return self.route(pattern, 'GET') def post(self, pattern): return self.route(pattern, 'POST') def head(self, pattern): return self.route(pattern, 'HEAD') def match(self, request): if not request.path.startswith(self.__prefix): return None for methods, pattern, handler in self.__routetable: if not methods or request.method.upper() in methods: if pattern.match(request.path.replace(self.__prefix,'',1)): return handler(request) class App: __ROUTES = [] @classmethod def register(cls, *routers): for router in routers: cls.__ROUTES.append(router) @wsgify def __call__(self, request): for router in self.__ROUTES: response = router.match(request) if response: return response raise HTTPNotFound('<h1>你访问的页面被外星人劫持了</h1>') ind = Route() python = Route('/python') App.register(ind, python) @ind.route('^/$') def indexhandler(request): return 'index page' @python.get('^/python$') def pythonhandler(request): return 'python page' if __name__ == '__main__': ip = '127.0.0.1' port = 9999 server = make_server(ip, port, App()) try: server.serve_forever() except KeyboardInterrupt: server.shutdown() server.server_close()