zoukankan      html  css  js  c++  java
  • 自动发现项目中的所有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")
    View Code

    可以看到的是,每一个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")
    未经过实际测试。不想用1版本的

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

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

  • 相关阅读:
    洛谷P1656 炸铁路 题解 并查集
    洛谷P1455 搭配购买 题解 并查集+01背包
    欧拉计划第16题题解
    欧拉计划第15题题解
    欧拉计划第14题题解
    scrapy 爬虫保存数据
    FileNotFoundError: File b'score.xlsx' does not exist
    图片截取
    图像过滤器
    如何查看字符编码?
  • 原文地址:https://www.cnblogs.com/chengege/p/10720523.html
Copyright © 2011-2022 走看看