注册路由的简易实现

时间 2019-12-07

原文原文链接

首先是由于今天很早就醒了，并且还睡不着，我就只好起床了，而后无聊就看了一下之前刚学python时候跟着写的代码，如今理解的更全面点而后也正好总结一下其中的一些点，好比说注册路由。python

注册路由

代码

resigter_handlers.py

# -*- coding: utf-8 -*-

import functools


def get(path):
    def _get(func):
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            return func(*args, **kwargs)
        wrapper.__method__ = "GET"
        wrapper.__route__ = path
        return wrapper
    return _get


class App(object):

    def register(self, method, path, handler):
        if self.__class__:
            pass
        print(f"method: {method}\n"
              f"path: {path}\n"
              f"handler: {handler}"
              )


app = App()


def add_routes(module_name):
    dot = module_name.rfind(".")
    if dot == -1:
        mod = __import__(module_name, globals(), locals())
    else:
        name = module_name[dot+1:]
        mod = __import__(name, globals(), locals())

    for item in dir(mod):

        if item.startswith("_"):
            continue

        func = getattr(mod, item)
        if callable(func):
            path = getattr(func, "__route__", None)
            method = getattr(func, "__method__", None)
            if path and method:
                add_route(method, path, func)


def add_route(method, path, func):
    app.register(method, path, RequestHandlers(func))


class RequestHandlers(object):

    def __init__(self, func):
        self.func = func

    def __call__(self, request):
        kwargs = request.match_info.items()

        response = self.func(**kwargs)

        return response


if __name__ == "__main__":
    add_routes("handlers")

handlers.py

# -*- coding: utf-8 -*-

from _http.src.request_handlers.register_handlers import get


@get("/api/register")
def create_user():
    return "create"


@get("api/user")
def get_user():
    return "get"


@get("/api/update")
def update_user():
    return "update"


@get("/api/delete")
def delete_user():
    return "delete"

思路

首先是建立了一个add_route的方法，用于在app对象中注册路由，其中的ReqestHandlers类是用来对web请求的参数进行解析的而后传递给视图函数处理的，这里就简单写了一下；web

而后是实现了一个装饰器，用来指定__method__和__route__的，便于后面的func的筛选以及app中视图的注册；api

接着就是编写add_routes函数，先经过文件名将该文件下所全部变量都导入到mod变量当中，而后便利mod变量中的全部属性，经过一个简单的筛选条件，将可调用而且具有__method__和__route__属性的函数进行注册。app

基本就是上述这些步骤了，其实中间还有不少对于参数的处理和条件的判断，在简易实现这里就不作多的扩展了，更多的仍是提供给本身一个简单的思路。函数