装饰器实现路由控制

part 1

def application(env, start_response):
    print(env['PATH_INFO'])
    status = '200 OK'
    response_headers = [('Content-Type', 'text/html')]
    start_response(status, response_headers)
    return ['您请求的路径为：{}'.format(env['PATH_INFO']).encode('utf8')]

from webapplication import application as app
from wsgiref.simple_server import make_server


server = make_server('', 8080, app)
server.serve_forever()

请求： http://localhost:8080/home

/home路径

请求： http://localhost:8080

网站根目录

part 2

先前博主的那个装饰器实现的路由控制只能在Python2.7上运行，到了3就不好使了。然后本人又修改了一下，实现可以在Python3上运行。

# coding: utf8
import sys
from wsgiref.simple_server import make_server


class Router(object):

    def __init__(self):
        self.routes = {}

    def route(self, path):
        def decorator(func):
            self.routes[path] = func
            print(self.routes)
            return func
        return decorator

    def __call__(self, env, start_response):
        path = env['PATH_INFO']
        if path in self.routes:
            status = '200 OK'
            headers = [('Content-Type', 'text/html;charset=UTF-8')]
            start_response(status, headers)
            # 然后可以通过route绑定的函数来处理相关的星球，然后返回给客户端
            # return ['您请求的路径为：{}'.format(path).encode('utf8')]
            return self.routes[path]()
        else:
            status = '404 Not Found'
            headers = [('Content-Type', 'text/html;charset=UTF-8')]
            start_response(status, headers)
            return [b'Bad Request']


app = Router()

@app.route('/')
def index():
    print('首页内容')
    yield 'root path'.encode('utf8')

@app.route('/home')
def home():
    print('home页面内容')
    yield 'home页面内容'.encode('utf8')

server = make_server('', 8080, app)
server.serve_forever()