自动发现项目中的所有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 列表。

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 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

            if pre_namespace:  # 判断当前这个url是不是有namespace前缀。也就是:是否是某一个命名空间中的 name别名
                name = "%s:%s" % (pre_namespace, item.name)
            else:
                name = item.name
            url = pre_url + str(item.pattern)
            url_ordered_dict[name] = {"name": name, "url": url.replace("^", "").replace("$", "")}

        elif isinstance(item, URLResolver):  # 表示这是一个路由分发。 这里就需要递归了
            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.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")
未经过实际测试。不想用1版本的

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

 

猜你喜欢

转载自www.cnblogs.com/chengege/p/10720523.html
0条评论
添加一条新回复