Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat(refactor): get_families_with_samples #3927

Closed
wants to merge 2 commits into from
Closed
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
40 changes: 30 additions & 10 deletions cg/store/crud/read.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,11 +12,19 @@
from cg.exc import CaseNotFoundError, CgError, OrderNotFoundError, SampleNotFoundError
from cg.models.orders.constants import OrderType
from cg.server.dto.orders.orders_request import OrdersRequest
from cg.server.dto.samples.collaborator_samples_request import CollaboratorSamplesRequest
from cg.server.dto.samples.collaborator_samples_request import (
CollaboratorSamplesRequest,
)
from cg.store.base import BaseHandler
from cg.store.exc import EntryNotFoundError
from cg.store.filters.status_analysis_filters import AnalysisFilter, apply_analysis_filter
from cg.store.filters.status_application_filters import ApplicationFilter, apply_application_filter
from cg.store.filters.status_analysis_filters import (
AnalysisFilter,
apply_analysis_filter,
)
from cg.store.filters.status_application_filters import (
ApplicationFilter,
apply_application_filter,
)
from cg.store.filters.status_application_limitations_filters import (
ApplicationLimitationsFilter,
apply_application_limitations_filter,
Expand All @@ -26,14 +34,23 @@
apply_application_versions_filter,
)
from cg.store.filters.status_bed_filters import BedFilter, apply_bed_filter
from cg.store.filters.status_bed_version_filters import BedVersionFilter, apply_bed_version_filter
from cg.store.filters.status_bed_version_filters import (
BedVersionFilter,
apply_bed_version_filter,
)
from cg.store.filters.status_case_filters import CaseFilter, apply_case_filter
from cg.store.filters.status_case_sample_filters import CaseSampleFilter, apply_case_sample_filter
from cg.store.filters.status_case_sample_filters import (
CaseSampleFilter,
apply_case_sample_filter,
)
from cg.store.filters.status_collaboration_filters import (
CollaborationFilter,
apply_collaboration_filter,
)
from cg.store.filters.status_customer_filters import CustomerFilter, apply_customer_filter
from cg.store.filters.status_customer_filters import (
CustomerFilter,
apply_customer_filter,
)
from cg.store.filters.status_illumina_flow_cell_filters import (
IlluminaFlowCellFilter,
apply_illumina_flow_cell_filters,
Expand All @@ -52,7 +69,10 @@
OrderTypeApplicationFilter,
apply_order_type_application_filter,
)
from cg.store.filters.status_organism_filters import OrganismFilter, apply_organism_filter
from cg.store.filters.status_organism_filters import (
OrganismFilter,
apply_organism_filter,
)
from cg.store.filters.status_pacbio_smrt_cell_filters import (
PacBioSMRTCellFilter,
apply_pac_bio_smrt_cell_filters,
Expand Down Expand Up @@ -1015,7 +1035,7 @@ def get_families_with_analyses(self) -> Query:
"""Return all cases in the database with an analysis."""
return self._get_outer_join_cases_with_analyses_query()

def get_families_with_samples(self) -> Query:
def get_cases_with_samples(self) -> Query:
"""Return all cases in the database with samples."""
return self._get_join_cases_with_samples_query()

Expand Down Expand Up @@ -1211,7 +1231,7 @@ def observations_to_upload(self, workflow: Workflow = None) -> Query:
CaseFilter.WITH_LOQUSDB_SUPPORTED_SEQUENCING_METHOD,
]
records: Query = apply_case_filter(
cases=self.get_families_with_samples(),
cases=self.get_cases_with_samples(),
filter_functions=case_filter_functions,
workflow=workflow,
)
Expand All @@ -1222,7 +1242,7 @@ def observations_to_upload(self, workflow: Workflow = None) -> Query:
def observations_uploaded(self, workflow: Workflow = None) -> Query:
"""Return observations that have been uploaded."""
records: Query = apply_case_filter(
cases=self.get_families_with_samples(),
cases=self.get_cases_with_samples(),
filter_functions=[CaseFilter.WITH_LOQUSDB_SUPPORTED_WORKFLOW],
workflow=workflow,
)
Expand Down
Loading