获取Django项目的全部url

Posted 2020-10-12 名刀

在为一个项目添加权限时，遇到一个问题，就是为项目所有的url设置权限，但是一个一个手动输入太麻烦了，所以考虑用代码获取到一个项目所有的url

首先，考虑到项目最外层的urlpartterns，因为所有的url都要通过这里

urlpatterns = [
    # url(r‘^admin/‘, admin.site.urls),
    url(r‘^arya/‘, site.urls),
    url(r‘^index/‘, index),
]

先循环打印一下这个列表，看一下拿到的结果：

<RegexURLResolver <RegexURLPattern list> (arya:arya) ^arya/>
<RegexURLResolver <module ‘rbac.urls‘ from ‘C:\\Users\\zhangcan\\Desktop\\pro_crm\\pro_crm\\rbac\\urls.py‘> (None:None) ^rbac/>
<RegexURLPattern None ^index/>

可以看到，直接有对应函数的和通过文件导入的这种嵌套的，打印出来的类型是不一样的，导入一下这个类型

from django.urls.resolvers import RegexURLPattern
from django.urls.resolvers import RegexURLResolver

按住ctrl点进去看一下源码，找来找去看到有个

self._regex = regex

拿到代码中打印一下这个，拿到：

^arya/
^rbac/
^index/

哇，妙啊，但是这个只能是最后一个有用，上面两个还有嵌套，我们通过上面拿到的两种类型区分一下

对于这种有嵌套的，可以使用递归来一层一层的扒开它，直到拿到的是  RegexURLPattern  类型

而且，每扒开一层，它的url得加上之前的那个

我们可以通过   item.urlconf_name   拿到嵌套的里面一层的urlpartterns

这里面还涉及一个问题就是，用递归函数的话，如何保存每次得到的结果?

可以用一个全局变量，没有问题，其实还可以用一个Python的特性，就是如果参数的默认值为字典，列表这种可变数据类型的话，以后引用的都是同一个内存地址

所以这个全局变量还可以写成是一个默认是个空列表的参数,但是这样又出了新的问题，就是只要不重启项目，这个列表一直不清空，所以再用一个默认参数为False，第一次调用的时候把这个参数设置为True，在函数中判断如果这个参数为True，就清空列表。

最终代码可以写成这样：

from django.conf.urls import url,include
from arya.service.sites import site

from django.urls.resolvers import RegexURLPattern
from django.urls.resolvers import RegexURLResolver

from django.shortcuts import HttpResponse

def index(request):
    print(get_all_url(urlpatterns,prev=‘/‘))
    return HttpResponse(‘...‘)
def get_all_url(urlparrentens,prev,is_first=False,result=[]):
    if is_first:
        result.clear()
    for item in urlparrentens:
        v = item._regex.strip(‘^$‘)#去掉url中的^和$
        if isinstance(item,RegexURLPattern):
            result.append(prev + v)
        else:
            get_all_url(item.urlconf_name,prev + v)
    return result

urlpatterns = [
    url(r‘^arya/‘, site.urls),
    url(r‘^index/‘, index),
]

这样就拿到了这个项目所有的url

[‘/arya/login/‘, ‘/arya/logout/‘, ‘/arya/app01/department/‘, ‘/arya/app01/department/add/‘, ‘/arya/app01/department/(.+)/delete/‘, ‘/arya/app01/department/(.+)/change/‘, ‘/arya/app01/userinfo/‘, ‘/arya/app01/userinfo/add/‘, ‘/arya/app01/userinfo/(.+)/delete/‘, ‘/arya/app01/userinfo/(.+)/change/‘, ‘/arya/rbac/userinfo/‘, ‘/arya/rbac/userinfo/add/‘, ‘/arya/rbac/userinfo/(.+)/delete/‘, ‘/arya/rbac/userinfo/(.+)/change/‘, ‘/arya/rbac/role/‘, ‘/arya/rbac/role/add/‘, ‘/arya/rbac/role/(.+)/delete/‘, ‘/arya/rbac/role/(.+)/change/‘, ‘/arya/rbac/permission/‘, ‘/arya/rbac/permission/add/‘, ‘/arya/rbac/permission/(.+)/delete/‘, ‘/arya/rbac/permission/(.+)/change/‘, ‘/arya/rbac/menu/‘, ‘/arya/rbac/menu/add/‘, ‘/arya/rbac/menu/(.+)/delete/‘, ‘/arya/rbac/menu/(.+)/change/‘, ‘/index/‘]