views.py 16 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415
  1. import inspect
  2. from importlib import import_module
  3. from inspect import cleandoc
  4. from pathlib import Path
  5. from django.apps import apps
  6. from django.conf import settings
  7. from django.contrib import admin
  8. from django.contrib.admin.views.decorators import staff_member_required
  9. from django.contrib.admindocs import utils
  10. from django.contrib.admindocs.utils import (
  11. replace_named_groups, replace_unnamed_groups,
  12. )
  13. from django.core.exceptions import ImproperlyConfigured, ViewDoesNotExist
  14. from django.db import models
  15. from django.http import Http404
  16. from django.template.engine import Engine
  17. from django.urls import get_mod_func, get_resolver, get_urlconf
  18. from django.utils._os import safe_join
  19. from django.utils.decorators import method_decorator
  20. from django.utils.inspect import (
  21. func_accepts_kwargs, func_accepts_var_args, get_func_full_args,
  22. method_has_no_args,
  23. )
  24. from django.utils.translation import gettext as _
  25. from django.views.generic import TemplateView
  26. from .utils import get_view_name
  27. # Exclude methods starting with these strings from documentation
  28. MODEL_METHODS_EXCLUDE = ('_', 'add_', 'delete', 'save', 'set_')
  29. class BaseAdminDocsView(TemplateView):
  30. """
  31. Base view for admindocs views.
  32. """
  33. @method_decorator(staff_member_required)
  34. def dispatch(self, request, *args, **kwargs):
  35. if not utils.docutils_is_available:
  36. # Display an error message for people without docutils
  37. self.template_name = 'admin_doc/missing_docutils.html'
  38. return self.render_to_response(admin.site.each_context(request))
  39. return super().dispatch(request, *args, **kwargs)
  40. def get_context_data(self, **kwargs):
  41. return super().get_context_data(**{
  42. **kwargs,
  43. **admin.site.each_context(self.request),
  44. })
  45. class BookmarkletsView(BaseAdminDocsView):
  46. template_name = 'admin_doc/bookmarklets.html'
  47. class TemplateTagIndexView(BaseAdminDocsView):
  48. template_name = 'admin_doc/template_tag_index.html'
  49. def get_context_data(self, **kwargs):
  50. tags = []
  51. try:
  52. engine = Engine.get_default()
  53. except ImproperlyConfigured:
  54. # Non-trivial TEMPLATES settings aren't supported (#24125).
  55. pass
  56. else:
  57. app_libs = sorted(engine.template_libraries.items())
  58. builtin_libs = [('', lib) for lib in engine.template_builtins]
  59. for module_name, library in builtin_libs + app_libs:
  60. for tag_name, tag_func in library.tags.items():
  61. title, body, metadata = utils.parse_docstring(tag_func.__doc__)
  62. title = title and utils.parse_rst(title, 'tag', _('tag:') + tag_name)
  63. body = body and utils.parse_rst(body, 'tag', _('tag:') + tag_name)
  64. for key in metadata:
  65. metadata[key] = utils.parse_rst(metadata[key], 'tag', _('tag:') + tag_name)
  66. tag_library = module_name.split('.')[-1]
  67. tags.append({
  68. 'name': tag_name,
  69. 'title': title,
  70. 'body': body,
  71. 'meta': metadata,
  72. 'library': tag_library,
  73. })
  74. return super().get_context_data(**{**kwargs, 'tags': tags})
  75. class TemplateFilterIndexView(BaseAdminDocsView):
  76. template_name = 'admin_doc/template_filter_index.html'
  77. def get_context_data(self, **kwargs):
  78. filters = []
  79. try:
  80. engine = Engine.get_default()
  81. except ImproperlyConfigured:
  82. # Non-trivial TEMPLATES settings aren't supported (#24125).
  83. pass
  84. else:
  85. app_libs = sorted(engine.template_libraries.items())
  86. builtin_libs = [('', lib) for lib in engine.template_builtins]
  87. for module_name, library in builtin_libs + app_libs:
  88. for filter_name, filter_func in library.filters.items():
  89. title, body, metadata = utils.parse_docstring(filter_func.__doc__)
  90. title = title and utils.parse_rst(title, 'filter', _('filter:') + filter_name)
  91. body = body and utils.parse_rst(body, 'filter', _('filter:') + filter_name)
  92. for key in metadata:
  93. metadata[key] = utils.parse_rst(metadata[key], 'filter', _('filter:') + filter_name)
  94. tag_library = module_name.split('.')[-1]
  95. filters.append({
  96. 'name': filter_name,
  97. 'title': title,
  98. 'body': body,
  99. 'meta': metadata,
  100. 'library': tag_library,
  101. })
  102. return super().get_context_data(**{**kwargs, 'filters': filters})
  103. class ViewIndexView(BaseAdminDocsView):
  104. template_name = 'admin_doc/view_index.html'
  105. def get_context_data(self, **kwargs):
  106. views = []
  107. urlconf = import_module(settings.ROOT_URLCONF)
  108. view_functions = extract_views_from_urlpatterns(urlconf.urlpatterns)
  109. for (func, regex, namespace, name) in view_functions:
  110. views.append({
  111. 'full_name': get_view_name(func),
  112. 'url': simplify_regex(regex),
  113. 'url_name': ':'.join((namespace or []) + (name and [name] or [])),
  114. 'namespace': ':'.join(namespace or []),
  115. 'name': name,
  116. })
  117. return super().get_context_data(**{**kwargs, 'views': views})
  118. class ViewDetailView(BaseAdminDocsView):
  119. template_name = 'admin_doc/view_detail.html'
  120. @staticmethod
  121. def _get_view_func(view):
  122. urlconf = get_urlconf()
  123. if get_resolver(urlconf)._is_callback(view):
  124. mod, func = get_mod_func(view)
  125. try:
  126. # Separate the module and function, e.g.
  127. # 'mymodule.views.myview' -> 'mymodule.views', 'myview').
  128. return getattr(import_module(mod), func)
  129. except ImportError:
  130. # Import may fail because view contains a class name, e.g.
  131. # 'mymodule.views.ViewContainer.my_view', so mod takes the form
  132. # 'mymodule.views.ViewContainer'. Parse it again to separate
  133. # the module and class.
  134. mod, klass = get_mod_func(mod)
  135. return getattr(getattr(import_module(mod), klass), func)
  136. def get_context_data(self, **kwargs):
  137. view = self.kwargs['view']
  138. view_func = self._get_view_func(view)
  139. if view_func is None:
  140. raise Http404
  141. title, body, metadata = utils.parse_docstring(view_func.__doc__)
  142. title = title and utils.parse_rst(title, 'view', _('view:') + view)
  143. body = body and utils.parse_rst(body, 'view', _('view:') + view)
  144. for key in metadata:
  145. metadata[key] = utils.parse_rst(metadata[key], 'model', _('view:') + view)
  146. return super().get_context_data(**{
  147. **kwargs,
  148. 'name': view,
  149. 'summary': title,
  150. 'body': body,
  151. 'meta': metadata,
  152. })
  153. class ModelIndexView(BaseAdminDocsView):
  154. template_name = 'admin_doc/model_index.html'
  155. def get_context_data(self, **kwargs):
  156. m_list = [m._meta for m in apps.get_models()]
  157. return super().get_context_data(**{**kwargs, 'models': m_list})
  158. class ModelDetailView(BaseAdminDocsView):
  159. template_name = 'admin_doc/model_detail.html'
  160. def get_context_data(self, **kwargs):
  161. model_name = self.kwargs['model_name']
  162. # Get the model class.
  163. try:
  164. app_config = apps.get_app_config(self.kwargs['app_label'])
  165. except LookupError:
  166. raise Http404(_("App %(app_label)r not found") % self.kwargs)
  167. try:
  168. model = app_config.get_model(model_name)
  169. except LookupError:
  170. raise Http404(_("Model %(model_name)r not found in app %(app_label)r") % self.kwargs)
  171. opts = model._meta
  172. title, body, metadata = utils.parse_docstring(model.__doc__)
  173. title = title and utils.parse_rst(title, 'model', _('model:') + model_name)
  174. body = body and utils.parse_rst(body, 'model', _('model:') + model_name)
  175. # Gather fields/field descriptions.
  176. fields = []
  177. for field in opts.fields:
  178. # ForeignKey is a special case since the field will actually be a
  179. # descriptor that returns the other object
  180. if isinstance(field, models.ForeignKey):
  181. data_type = field.remote_field.model.__name__
  182. app_label = field.remote_field.model._meta.app_label
  183. verbose = utils.parse_rst(
  184. (_("the related `%(app_label)s.%(data_type)s` object") % {
  185. 'app_label': app_label, 'data_type': data_type,
  186. }),
  187. 'model',
  188. _('model:') + data_type,
  189. )
  190. else:
  191. data_type = get_readable_field_data_type(field)
  192. verbose = field.verbose_name
  193. fields.append({
  194. 'name': field.name,
  195. 'data_type': data_type,
  196. 'verbose': verbose or '',
  197. 'help_text': field.help_text,
  198. })
  199. # Gather many-to-many fields.
  200. for field in opts.many_to_many:
  201. data_type = field.remote_field.model.__name__
  202. app_label = field.remote_field.model._meta.app_label
  203. verbose = _("related `%(app_label)s.%(object_name)s` objects") % {
  204. 'app_label': app_label,
  205. 'object_name': data_type,
  206. }
  207. fields.append({
  208. 'name': "%s.all" % field.name,
  209. "data_type": 'List',
  210. 'verbose': utils.parse_rst(_("all %s") % verbose, 'model', _('model:') + opts.model_name),
  211. })
  212. fields.append({
  213. 'name': "%s.count" % field.name,
  214. 'data_type': 'Integer',
  215. 'verbose': utils.parse_rst(_("number of %s") % verbose, 'model', _('model:') + opts.model_name),
  216. })
  217. methods = []
  218. # Gather model methods.
  219. for func_name, func in model.__dict__.items():
  220. if inspect.isfunction(func) or isinstance(func, property):
  221. try:
  222. for exclude in MODEL_METHODS_EXCLUDE:
  223. if func_name.startswith(exclude):
  224. raise StopIteration
  225. except StopIteration:
  226. continue
  227. verbose = func.__doc__
  228. verbose = verbose and (
  229. utils.parse_rst(cleandoc(verbose), 'model', _('model:') + opts.model_name)
  230. )
  231. # Show properties and methods without arguments as fields.
  232. # Otherwise, show as a 'method with arguments'.
  233. if isinstance(func, property):
  234. fields.append({
  235. 'name': func_name,
  236. 'data_type': get_return_data_type(func_name),
  237. 'verbose': verbose or ''
  238. })
  239. elif method_has_no_args(func) and not func_accepts_kwargs(func) and not func_accepts_var_args(func):
  240. fields.append({
  241. 'name': func_name,
  242. 'data_type': get_return_data_type(func_name),
  243. 'verbose': verbose or '',
  244. })
  245. else:
  246. arguments = get_func_full_args(func)
  247. # Join arguments with ', ' and in case of default value,
  248. # join it with '='. Use repr() so that strings will be
  249. # correctly displayed.
  250. print_arguments = ', '.join([
  251. '='.join([arg_el[0], *map(repr, arg_el[1:])])
  252. for arg_el in arguments
  253. ])
  254. methods.append({
  255. 'name': func_name,
  256. 'arguments': print_arguments,
  257. 'verbose': verbose or '',
  258. })
  259. # Gather related objects
  260. for rel in opts.related_objects:
  261. verbose = _("related `%(app_label)s.%(object_name)s` objects") % {
  262. 'app_label': rel.related_model._meta.app_label,
  263. 'object_name': rel.related_model._meta.object_name,
  264. }
  265. accessor = rel.get_accessor_name()
  266. fields.append({
  267. 'name': "%s.all" % accessor,
  268. 'data_type': 'List',
  269. 'verbose': utils.parse_rst(_("all %s") % verbose, 'model', _('model:') + opts.model_name),
  270. })
  271. fields.append({
  272. 'name': "%s.count" % accessor,
  273. 'data_type': 'Integer',
  274. 'verbose': utils.parse_rst(_("number of %s") % verbose, 'model', _('model:') + opts.model_name),
  275. })
  276. return super().get_context_data(**{
  277. **kwargs,
  278. 'name': opts.label,
  279. 'summary': title,
  280. 'description': body,
  281. 'fields': fields,
  282. 'methods': methods,
  283. })
  284. class TemplateDetailView(BaseAdminDocsView):
  285. template_name = 'admin_doc/template_detail.html'
  286. def get_context_data(self, **kwargs):
  287. template = self.kwargs['template']
  288. templates = []
  289. try:
  290. default_engine = Engine.get_default()
  291. except ImproperlyConfigured:
  292. # Non-trivial TEMPLATES settings aren't supported (#24125).
  293. pass
  294. else:
  295. # This doesn't account for template loaders (#24128).
  296. for index, directory in enumerate(default_engine.dirs):
  297. template_file = Path(safe_join(directory, template))
  298. if template_file.exists():
  299. template_contents = template_file.read_text()
  300. else:
  301. template_contents = ''
  302. templates.append({
  303. 'file': template_file,
  304. 'exists': template_file.exists(),
  305. 'contents': template_contents,
  306. 'order': index,
  307. })
  308. return super().get_context_data(**{
  309. **kwargs,
  310. 'name': template,
  311. 'templates': templates,
  312. })
  313. ####################
  314. # Helper functions #
  315. ####################
  316. def get_return_data_type(func_name):
  317. """Return a somewhat-helpful data type given a function name"""
  318. if func_name.startswith('get_'):
  319. if func_name.endswith('_list'):
  320. return 'List'
  321. elif func_name.endswith('_count'):
  322. return 'Integer'
  323. return ''
  324. def get_readable_field_data_type(field):
  325. """
  326. Return the description for a given field type, if it exists. Fields'
  327. descriptions can contain format strings, which will be interpolated with
  328. the values of field.__dict__ before being output.
  329. """
  330. return field.description % field.__dict__
  331. def extract_views_from_urlpatterns(urlpatterns, base='', namespace=None):
  332. """
  333. Return a list of views from a list of urlpatterns.
  334. Each object in the returned list is a two-tuple: (view_func, regex)
  335. """
  336. views = []
  337. for p in urlpatterns:
  338. if hasattr(p, 'url_patterns'):
  339. try:
  340. patterns = p.url_patterns
  341. except ImportError:
  342. continue
  343. views.extend(extract_views_from_urlpatterns(
  344. patterns,
  345. base + str(p.pattern),
  346. (namespace or []) + (p.namespace and [p.namespace] or [])
  347. ))
  348. elif hasattr(p, 'callback'):
  349. try:
  350. views.append((p.callback, base + str(p.pattern), namespace, p.name))
  351. except ViewDoesNotExist:
  352. continue
  353. else:
  354. raise TypeError(_("%s does not appear to be a urlpattern object") % p)
  355. return views
  356. def simplify_regex(pattern):
  357. r"""
  358. Clean up urlpattern regexes into something more readable by humans. For
  359. example, turn "^(?P<sport_slug>\w+)/athletes/(?P<athlete_slug>\w+)/$"
  360. into "/<sport_slug>/athletes/<athlete_slug>/".
  361. """
  362. pattern = replace_named_groups(pattern)
  363. pattern = replace_unnamed_groups(pattern)
  364. # clean up any outstanding regex-y characters.
  365. pattern = pattern.replace('^', '').replace('$', '').replace('?', '')
  366. if not pattern.startswith('/'):
  367. pattern = '/' + pattern
  368. return pattern