自动发现项目中的所有URL

我的rbac组件，是想用到任何一个，项目中的。 so 问题来了。
　　- 问题： 拿到一个项目。 怎样获取到，当前项目中， 所有的URL 以及 每个URL的别名name， 还有是有 namespace 命名空间。
　　- 实现思路：
　　　　1. 先要确定我们 根级 路由在哪里。  就是和项目文件同名的包中， 的 urls.py 中的 urlpatterns=[....]  这个路由的位置在我们的settings中是有进行配置的
　　　　　　ROOT_URLCONF = 'learn_formset.urls'  当然这个是可以进行修改的。

那个怎么才能获得，这个  urlpatterns  列表中的所有的对象呢？
直接导入的话，也是可以的。 但是更好的是 通过settings 中 ROOT_URLCONF 的字符串来进行导入。
Django中有一个模块就是来做这件事的， from django.utils.module_loading import import_string  这个模块就是用来根据一个字符串来导入相应的模块。
他的返回值，就是一个模块对象 <class 'module'> 。 可以通过 点 语法。获取到其中的 urlpatterns 列表。

看一看，打印 urlpatterns  里面的没一个数据， 能得到啥：

 

from django.conf import settings
from django.utils.module_loading import import_string

def get_all_url_dict():
    '''
    递归获取项目中，所有的url 保存到字典
    :return:
    '''
    md = import_string(settings.ROOT_URLCONF)  # <class 'module'>
    for item in md.urlpatterns:
   　　print(item)

#  <URLResolver <URLPattern list> (admin:admin) '^admin/'>

#  <URLResolver <module 'web.urls' from 'D:\\crm_learn\\web\\urls.py'> (None:None) '^'>

#  <URLResolver <module 'rbac.urls' from 'D:\\crm_learn\\rbac\\urls.py'> (rbac:rbac) '^rbac/'>

def multi_permissions(request):
    '''
    批量操作权限
    :param request:
    :return:
    '''
    # 获取项目中，所有的URL
    get_all_url_dict()
    return HttpResponse("OK")

可以看到的是，每一个item 都是一个  URLResolver 对象。这表示这是一个 路由分发的url。
但是  还有一种是  对应的视图函数的。URLPattern  对象。表示他对应的是一个 视图函数。

看代码吧！ 这是 Django 2.0 版本的。 和 1 版本有不小的改动。

from collections import OrderedDict
from django.conf import settings
from django.utils.module_loading import import_string
from django.urls.resolvers import URLResolver, URLPattern


def check_url_exclude(url):
    '''自定制，过滤一下。 以 xxx 为前缀的 url'''
    exclude_url = [
        "/admin/.*",
        "/login/",
    ]
    for regex in exclude_url:
        if re.match(regex, url):
            return True


def recursion_urls(pre_namespace, pre_url, urlpatterns, url_ordered_dict):
    '''
    递归获取，所有的url
    :param pre_namespace:  namespace前缀，用于拼接name (namespace:name)
    :param pre_url:  url的前缀， 用于拼接url
    :param urlpatterns:  路由关系列表
    :param url_ordered_dict:  用于保存递归中获取的所有的路由
    :return:
    '''
    for item in urlpatterns:
        if isinstance(item, URLPattern):  # 表示一个 非路由分发。将路由添加到字典中
            if not item.name:  # 判断这个url 有没有，name别名
                continue

　　　　　　　name = item.name
            if pre_namespace:  # 判断当前这个url是不是有namespace前缀。也就是:是否是某一个命名空间中的 name别名
                name = "%s:%s" % (pre_namespace, item.name)
            url = (pre_url + str(item.pattern)).replace("^", "").replace("$", "")
            if check_url_exclude(url):  # 在这里进行自定制的过滤。 过露出我不想要的 哪些url
                continue
            url_ordered_dict[name] = {"name": name, "url": url}

        elif isinstance(item, URLResolver):  # 表示这是一个路由分发。 这里就需要递归了
　　　　　　  namespace = pre_namespace
            if pre_namespace:  # 如果有前缀
                if item.namespace:  # 自己有没有namespace
                    namespace = "%s:%s" % (pre_namespace, item.namespace)# 把之前的pre_namespace 和当前的 item.namespace 拼接。 传给下一次的递归函数。继续进行拼接
            else:
                if item.namespace:
                    namespace = item.namespace
            recursion_urls(namespace, pre_url + str(item.pattern), item.url_patterns, url_ordered_dict)
　　　　　　　　# 进入下一次循环之前，pre_url + str(item.pattern) 要拼接上这一次循环的 url。
　　　　　　　　# item.url_patterns这一次是 URLResolver 对象的 url_patterns。 中间要加一个 _ 烦得很。 第一次是通过导入拿到的 模块对象。
　　　　　　　　# 但是 递归中的不是 模块对象。是一个URLResolver对象。 所以要加一个 _ 。下划线

def get_all_url_dict():
    '''
    获取项目中，所有的url 保存到字典（前提是，每个url必须有name别名）
    :return:
    '''
    url_ordered_dict = OrderedDict()
    md = import_string(settings.ROOT_URLCONF)

    recursion_urls(None, "/", md.urlpatterns, url_ordered_dict)
    # 递归的去获取所有的路由。
    # 第一次循环时，肯定是从 根路由开始， 所以没有前缀 传一个None.
    # "/" 也是因为，第一次循环时。 所有的url 都没有前导 的 "/" 手动的加上。
    # md.urlpatterns 要循环的这个列表。
    # url_ordered_dict 保存所有url 的字典。
    return url_ordered_dict


def multi_permissions(request):
    '''
    批量操作权限
    :param request:
    :return:
    '''
    # 获取项目中，所有的URL
    all_url_dict = get_all_url_dict()
    print(all_url_dict)
    return HttpResponse("OK")

 

 这里是1.0版本的。 主要是，几个关键的参数获取的位置上。 有些不同！

from collections import OrderedDict
from django.conf import settings
from django.utils.module_loading import import_string
from django.urls.resolvers import RegexURLResolver, RegexURLPattern


def recursion_urls(pre_namespace, pre_url, urlpatterns, url_ordered_dict):
    for item in urlpatterns:
        if isinstance(item, RegexURLPattern):  # 表示一个 非路由分发。将路由添加到字典中
            if not item.name:  # 判断这个url 有没有，name别名
                continue

            if pre_namespace:  # 判断当前这个url是不是有namespace前缀。如果有就说明是通过路由分发过来的，
　　　　　　　　　　　　　　　　　　　# 就要加上， 他上级的名称空间
                name = "%s:%s" % (pre_namespace, item.name)
            else:
                name = item.name
            url = pre_url + str(item._regex)
            url_ordered_dict[name] = {"name": name, "url": url.replace("^", "").replace("$", "")}

        elif isinstance(item, RegexURLResolver):  # 表示这是一个路由分发。 这里就需要递归了
            if pre_namespace:  # 如果有前缀
                if item.namespace:  # 自己有没有namespace
                    namespace = "%s:%s" % (pre_namespace, item.namespace)
                else:
                    namespace = pre_namespace
                # 把之前的pre_namespace 和当前的 item.namespace 拼接。 传给下一次的递归函数。继续进行拼接
            else:
                if item.namespace:
                    namespace = item.namespace
                else:
                    namespace = None
            recursion_urls(namespace, pre_url + str(item._regex.pattern), item.url_patterns, url_ordered_dict)


def get_all_url_dict():
    url_ordered_dict = OrderedDict()
    md = import_string(settings.ROOT_URLCONF)

    recursion_urls(None, "/", md.urlpatterns, url_ordered_dict)
    return url_ordered_dict


def multi_permissions(request):
    # 获取项目中，所有的URL
    all_url_dict = get_all_url_dict()
    print(all_url_dict)
    return HttpResponse("OK")

我的这个， 中间做了一个限制， 就是 必须要确保每一个。 url 必须要有一个别名。  没有别名的就直接跳过了。
因为，如果使用我的rbac 组件 我需要这个别名， 进行按钮粒度的控制。 必须要有。
但是我没有让他报错， 因为还是有一些，是不需要起别名的。

 然后就是把它放到，一个固定的地方了。 放到rbac这个app 哪里都行。 用的时候就导入  get_all_url_dict() 就行了