Skip to content

Commit 99a3f47

Browse files
committed
add fohm and gsaid to mutant uploadAPI
1 parent 8b7573b commit 99a3f47

File tree

2 files changed

+15
-3
lines changed

2 files changed

+15
-3
lines changed

cg/meta/upload/fohm/fohm.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -368,9 +368,13 @@ def parse_and_write_pangolin_report(self) -> list[FohmPangolinReport]:
368368
self.create_pangolin_report(sars_cov_pangolin_reports)
369369
return sars_cov_pangolin_reports
370370

371-
def aggregate_delivery(self, cases: list[str]) -> None:
372-
"""Aggregate and hardlink reports."""
373-
self.set_cases_to_aggregate(cases)
371+
def aggregate_delivery(self, case_ids: list[str]) -> None:
372+
"""
373+
Aggregate and hardlink reports.
374+
args:
375+
case_ids: The internal ids for cases to aggregate.
376+
"""
377+
self.set_cases_to_aggregate(case_ids)
374378
self.create_daily_delivery_folders()
375379
sars_cov_complementary_reports: list[FohmComplementaryReport] = (
376380
self.parse_and_write_complementary_report()

cg/meta/upload/mutant/mutant.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
11
from click import Context
22

3+
from cg.meta.upload.fohm.fohm import FOHMUploadAPI
4+
from cg.meta.upload.gisaid import GisaidAPI
35
from cg.meta.upload.upload_api import UploadAPI
46
from cg.meta.workflow.mutant import MutantAnalysisAPI
57
from cg.models.cg_config import CGConfig
@@ -10,10 +12,16 @@ class MutantUploadAPI(UploadAPI):
1012

1113
def __init__(self, config: CGConfig):
1214
self.analysis_api: MutantAnalysisAPI = MutantAnalysisAPI(config)
15+
self.fohm_api = FOHMUploadAPI(config)
16+
self.gsaid_api = GisaidAPI(config)
17+
1318
super().__init__(config=config, analysis_api=self.analysis_api)
1419

1520
def upload(self, ctx: Context, case: Case, restart: bool) -> None:
1621
latest_analysis: Analysis = case.analyses[0]
1722
self.update_upload_started_at(latest_analysis)
1823
self.upload_files_to_customer_inbox(case)
24+
self.gsaid_api.upload(case.internal_id)
25+
self.fohm_api.aggregate_delivery(case_ids=[case.internal_id])
26+
self.fohm_api.sync_files_sftp()
1927
self.update_uploaded_at(latest_analysis)

0 commit comments

Comments
 (0)