Lines Matching refs:query_set
332 def _add_attribute_join(self, query_set, join_condition,
336 return self.add_join(query_set, 'tko_test_attributes',
342 def _add_label_pivot_table_join(self, query_set, suffix, join_condition='',
344 return self.add_join(query_set, 'tko_test_labels_tests',
351 def _add_label_joins(self, query_set, suffix=''):
352 query_set = self._add_label_pivot_table_join(
353 query_set, suffix=suffix, force_left_join=True)
361 query_set.query.add_custom_join('tko_test_labels',
363 query_set.query.LOUTER,
365 return query_set
378 def _include_or_exclude_labels(self, query_set, label_names, exclude=False):
384 return self._add_label_pivot_table_join(query_set,
390 def _add_custom_select(self, query_set, select_name, select_sql):
391 return query_set.extra(select={select_name: select_sql})
394 def _add_select_value(self, query_set, alias):
395 return self._add_custom_select(query_set, alias,
399 def _add_select_ifnull(self, query_set, alias, non_null_value):
402 return self._add_custom_select(query_set, alias, select_sql)
405 def _join_test_label_column(self, query_set, label_name, label_id):
408 query_set = Test.objects.join_custom_field(query_set, label_query,
411 query_set = self._add_select_ifnull(query_set, alias, label_name)
412 return query_set
415 def _join_test_label_columns(self, query_set, label_names):
418 query_set = self._join_test_label_column(query_set, label_name,
420 return query_set
423 def _join_test_attribute(self, query_set, attribute, alias=None,
436 query_set = Test.objects.join_custom_field(query_set, attribute_query,
439 query_set = self._add_select_value(query_set, alias)
440 query_set
443 def _join_machine_label_columns(self, query_set, machine_label_names):
448 query_set = self._join_test_attribute(
449 query_set, 'host-labels',
451 query_set = self._add_select_ifnull(query_set, alias, label_name)
452 return query_set
455 def _join_one_iteration_key(self, query_set, result_key, first_alias=None):
466 query_set = Test.objects.join_custom_field(query_set, iteration_query,
469 query_set = self._add_select_value(query_set, alias)
472 query_set = self._add_custom_select(
473 query_set, 'iteration_index',
476 return query_set, alias
499 query_set, first_alias = self._join_one_iteration_key(
502 query_set, _ = self._join_one_iteration_key(query_set, result_key,
504 return query_set
507 def _join_job_keyvals(self, query_set, job_keyvals):
511 query_set = Job.objects.join_custom_field(query_set, keyval_query,
513 query_set = self._add_select_value(query_set, alias)
514 return query_set
517 def _join_iteration_attributes(self, query_set, iteration_attributes):
522 query_set = Test.objects.join_custom_field(query_set,
524 query_set = self._add_select_value(query_set, alias)
525 return query_set
569 query_set = self.get_query_set()
573 query_set = self._join_test_attribute(query_set, attribute)
576 query_set = self._join_test_label_columns(query_set, test_labels)
579 query_set = self._join_machine_label_columns(query_set, machine_labels)
582 query_set = self._join_iteration_results(query_set, iteration_keys)
585 query_set = self._join_job_keyvals(query_set, job_keyvals)
588 query_set = self._join_iteration_attributes(query_set,
597 query_set = self._add_label_joins(query_set)
603 query_set = self._include_or_exclude_labels(query_set,
607 query_set = self._include_or_exclude_labels(query_set,
617 query_set = self._add_attribute_join(
618 query_set,
622 query_set = self._add_attribute_join(
623 query_set,
632 query_set = self._add_attribute_join(
633 query_set, suffix='_host_labels',
637 return query_set
664 query_set = self.model.query_objects(filter_data)
665 query_set = self._add_label_joins(query_set, suffix='_list')
666 rows = self._custom_select_query(query_set, ['tko_test_labels_list.id'])
