from functools import partial as _partial from django.conf import settings as _settings from django.core.cache import cache as _cache from django.db.models import Prefetch as _Prefetch from django.urls import reverse from pythonpro.modules.models import ( Chapter as _Chapter, Module as _Module, Section as _Section, Topic as _Topic, ) __all__ = [ 'get_topic_model', 'get_all_modules', 'get_module_with_contents', 'get_section_with_contents', 'get_chapter_with_contents', 'get_topic_with_contents', 'get_topic_with_contents_by_id', 'get_entire_content_forest', 'get_tree', 'topics_user_interacted_queryset', 'get_tree_by_module_slug', 'add_modules_purchase_link' ] def get_topic_model(): return _Topic def get_all_modules(): """ Search all modules on database sorted by order :return: tuple of Module """ lazy_all_modules = _partial(tuple, _Module.objects.order_by('order')) return _cache.get_or_set('ALL_MODULES', lazy_all_modules, _settings.CACHE_TTL) def get_module_with_contents(slug): """ Search for a module with respective sections and chapters :param slug: module's slug :return: Module with respective section on attribute sections """ return _Module.objects.filter(slug=slug).prefetch_related( _Prefetch( 'section_set', queryset=_Section.objects.order_by('order').prefetch_related( _Prefetch( 'chapter_set', queryset=_Chapter.objects.order_by('order').prefetch_related( _Prefetch( 'topic_set', queryset=_Topic.objects.order_by( 'order'), to_attr='topics') ), to_attr='chapters' ) ), to_attr='sections') ).get() def get_section_with_contents(slug): """ Search for a section with respective module and chapters :param slug: section's slug :return: Section """ return _Section.objects.filter(slug=slug).select_related('module').prefetch_related( _Prefetch( 'chapter_set', queryset=_Chapter.objects.order_by('order').prefetch_related( _Prefetch( 'topic_set', queryset=_Topic.objects.order_by( 'order'), to_attr='topics') ), to_attr='chapters' ) ).get() def get_chapter_with_contents(slug): """ Search for a chapter respective to slug with it's module, section and topics :param slug: chapter's slug :return: Chapter """ return _Chapter.objects.filter(slug=slug).select_related('section').select_related( 'section__module').prefetch_related( _Prefetch( 'topic_set', queryset=_Topic.objects.order_by('order'), to_attr='topics' )).get() def get_topic_with_contents(slug): """ Search for a topic respective to slug with it's module, section and chapter :param slug: topic's slug :return: Topic """ return _Topic.objects.filter(slug=slug).select_related('chapter').select_related('chapter__section').select_related( 'chapter__section__module').get() def get_topic_memberkit_url(slug): """ Search for a topic respective to slug with it's module, section and chapter :param slug: topic's slug :return: Topic """ return _Topic.objects.only('memberkit_url').get(slug=slug).memberkit_url def get_topic_with_contents_by_id(id: int) -> _Topic: """ Search for a topic respective to slug with it's module, section and chapter :param id: topic's id :return: Topic """ return _Topic.objects.filter(id=id).select_related('chapter').select_related('chapter__section').select_related( 'chapter__section__module').get() def get_entire_content_forest(): """ Return a list of modules with the entire content on it :return: """ return list(_Module.objects.all().prefetch_related( _Prefetch( 'section_set', queryset=_Section.objects.order_by('order').prefetch_related( _Prefetch( 'chapter_set', queryset=_Chapter.objects.order_by('order').prefetch_related( _Prefetch( 'topic_set', queryset=_Topic.objects.order_by('order'), to_attr='topics' ) ), to_attr='chapters' ) ), to_attr='sections') )) def get_tree(module): """ Return a list of modules with the entire content on it :return: """ sections = list(_Section.objects.filter(module=module).order_by('order').prefetch_related( _Prefetch( 'chapter_set', queryset=_Chapter.objects.order_by('order').prefetch_related( _Prefetch( 'topic_set', queryset=_Topic.objects.order_by( 'order'), to_attr='topics') ), to_attr='chapters'))) module.sections = sections return sections def topics_user_interacted_queryset(user): return _Topic.objects.filter( topicinteraction__user=user ).select_related('chapter').select_related('chapter__section').select_related( 'chapter__section__module' ) def add_modules_purchase_link(modules): """ Add purchase link to modules :param modules - a list of modules :return modules - a list of modules with a purchase link """ link = reverse('checkout:bootcamp_lp') for module in modules: module.purchase_link = link return modules def get_tree_by_module_slug(module_slug: str): module = _Module.objects.get(slug=module_slug) module.sections = get_tree(module) return module