在為一個(gè)項(xiàng)目添加權(quán)限時(shí),遇到一個(gè)問題,就是為項(xiàng)目所有的url設(shè)置權(quán)限,但是一個(gè)一個(gè)手動(dòng)輸入太麻煩了,所以考慮用代碼獲取到一個(gè)項(xiàng)目所有的url
首先,考慮到項(xiàng)目最外層的urlpartterns,因?yàn)樗械膗rl都要通過這里
urlpatterns = [ # url(r'^admin/', admin.site.urls), url(r'^arya/', site.urls), url(r'^index/', index),]
先循環(huán)打印一下這個(gè)列表,看一下拿到的結(jié)果:
<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/>
可以看到,直接有對(duì)應(yīng)函數(shù)的和通過文件導(dǎo)入的這種嵌套的,打印出來的類型是不一樣的,導(dǎo)入一下這個(gè)類型
from django.urls.resolvers import RegexURLPatternfrom django.urls.resolvers import RegexURLResolver
按住ctrl點(diǎn)進(jìn)去看一下源碼,找來找去看到有個(gè)
self._regex = regex
拿到代碼中打印一下這個(gè),拿到:
^arya/^rbac/^index/
哇,妙啊,但是這個(gè)只能是最后一個(gè)有用,上面兩個(gè)還有嵌套,我們通過上面拿到的兩種類型區(qū)分一下
對(duì)于這種有嵌套的,可以使用遞歸來一層一層的扒開它,直到拿到的是 RegexURLPattern 類型
而且,每扒開一層,它的url得加上之前的那個(gè)
我們可以通過 item.urlconf_name 拿到嵌套的里面一層的urlpartterns
這里面還涉及一個(gè)問題就是,用遞歸函數(shù)的話,如何保存每次得到的結(jié)果?
可以用一個(gè)全局變量,沒有問題,其實(shí)還可以用一個(gè)Python的特性,就是如果參數(shù)的默認(rèn)值為字典,列表這種可變數(shù)據(jù)類型的話,以后引用的都是同一個(gè)內(nèi)存地址
所以這個(gè)全局變量還可以寫成是一個(gè)默認(rèn)是個(gè)空列表的參數(shù),但是這樣又出了新的問題,就是只要不重啟項(xiàng)目,這個(gè)列表一直不清空,所以再用一個(gè)默認(rèn)參數(shù)為False,第一次調(diào)用的時(shí)候把這個(gè)參數(shù)設(shè)置為True,在函數(shù)中判斷如果這個(gè)參數(shù)為True,就清空列表。
最終代碼可以寫成這樣:
from django.conf.urls import url,includefrom arya.service.sites import sitefrom django.urls.resolvers import RegexURLPatternfrom django.urls.resolvers import RegexURLResolverfrom django.shortcuts import HttpResponsedef 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 resulturlpatterns = [ url(r'^arya/', site.urls), url(r'^index/', index),]
新聞熱點(diǎn)
疑難解答
圖片精選