Development of an internal social media platform with personalised dashboards for students
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

views.py 16KB

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