Django3.2自动获取项目中的全部URL

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


def check_url_exclude(url):
"""排除一些特定的url"""
for regex in settings.AUTO_DISCOVER_EXCLUDE: # AUTO_DISCOVER_EXCLUDE 具体自己定义后放在settings中
if re.match(regex,url):
return True



def recursion_urls(pre_namespace, pre_url, urlpatterns, url_ordered_dict):
"""递归去获取URL
:param pre_namespace: namespace前缀,以后用户拼接name
:param pre_url: url前缀,以后用于拼接url
:param urlpatterns: 路由关系列表
:param url_ordered_dict: 用于保存递归中获取的所有路由
"""
for item in urlpatterns:
if isinstance(item,URLPattern):
if not item.name:
continue

if pre_namespace:
name = "%s:%s" %(pre_namespace,item.name)
else:
name = item.name
if not item.name:
raise Exception('URL路由中必须设置name属性')
url = pre_url + item.pattern.regex.pattern
url = url.replace('^','').replace('$','')

if check_url_exclude(url):
continue
url_ordered_dict[name] = {'name': name, 'url': url}

elif isinstance(item,URLResolver): # 路由分发,递归操作
if pre_namespace:
if item.namespace:
namespace = "%s:%s" %(pre_namespace,item.namespace)
else:
namespace = item.namespace
else:
if item.namespace:
namespace = item.namespace
else:
namespace = None

recursion_urls(namespace, pre_url + item.pattern.regex.pattern, item.url_patterns, url_ordered_dict)


def get_all_url_dict():
"""
获取项目中所有的URL(必须有name别名)
"""
url_ordered_dict = OrderedDict()
md = import_string(settings.ROOT_URLCONF)
recursion_urls(None,'/',md.urlpatterns,url_ordered_dict) # 递归去获取所有的路由
return url_ordered_dict

posted on 2022-07-25 18:35  一先生94  阅读(195)  评论(0编辑  收藏  举报

导航