Differential D8942 Diff 29033 search/serializers/training_search.py

Changeset View

Standalone View

search/serializers/training_search.py

	from django.db.models.expressions import F, Value
	from django.db.models.fields import CharField
	from taggit.models import Tag			from taggit.models import Tag

	from films.models import Asset, AssetCategory
	from search.serializers.base import BaseSearchSerializer			from search.serializers.base import BaseSearchSerializer
	from training.models import Training, Section, TrainingStatus			from training.models import Training, TrainingStatus


	class TrainingSearchSerializer(BaseSearchSerializer):			class TrainingSearchSerializer(BaseSearchSerializer):
	"""Prepare database objects to be indexed in the training search index."""			"""Prepare database objects to be indexed in the training search index."""

	models_to_index = [Training, Section, Asset]			models_to_index = [Training]
	filter_params = {			filter_params = {
	Training: {'status': TrainingStatus.published},			Training: {'status': TrainingStatus.published},
	Section: {'chapter__training__status': TrainingStatus.published},
	Asset: {
	'is_published': True,
	'film__is_published': True,
	'category': AssetCategory.production_lesson,
	},
	}			}
	annotations = {			annotations = {
	Training: {},			Training: {},
	Section: {
	'project': F('chapter__training__name'),
	'chapter_name': F('chapter__name'),
	'description': F('text'),
	},
	Asset: {
	'project': F('film__title'),
	'type': Value(AssetCategory.production_lesson, output_field=CharField()),
	},
	}			}
	additional_fields = {			additional_fields = {
	Training: {			Training: {
	'tags': lambda instance: [tag.name for tag in instance.tags.all()],			'tags': lambda instance: [tag.name for tag in instance.tags.all()],
	'secondary_tags': lambda instance: [			'secondary_tags': lambda instance: [
	tag.name			tag.name
	for tag in Tag.objects.filter(section__chapter__training__pk=instance.pk).distinct()			for tag in Tag.objects.filter(section__chapter__training__pk=instance.pk).distinct()
	],			],
	'favorite_url': lambda instance: instance.favorite_url,			'favorite_url': lambda instance: instance.favorite_url,
	},			},
	Section: {
	'tags': lambda instance: [tag.name for tag in instance.tags.all()],
	'secondary_tags': (
	lambda instance: [tag.name for tag in instance.chapter.training.tags.all()]
	),
	},
	Asset: {'tags': lambda instance: [tag.name for tag in instance.tags.all()]},
	}			}