|
1 | 1 | import string
|
2 | 2 | import traceback
|
| 3 | +from datetime import datetime |
3 | 4 | from typing import Dict, List, Literal
|
4 | 5 |
|
5 | 6 | import argo_workflows
|
6 | 7 | from argo_workflows.api import (
|
7 | 8 | archived_workflow_service_api,
|
8 |
| - workflow_service_api, |
9 | 9 | artifact_service_api,
|
| 10 | + workflow_service_api, |
| 11 | +) |
| 12 | +from argo_workflows.exceptions import NotFoundException |
| 13 | +from argo_workflows.model.io_argoproj_workflow_v1alpha1_retry_archived_workflow_request import ( |
| 14 | + IoArgoprojWorkflowV1alpha1RetryArchivedWorkflowRequest, |
10 | 15 | )
|
11 | 16 | from argo_workflows.model.io_argoproj_workflow_v1alpha1_workflow_create_request import (
|
12 | 17 | IoArgoprojWorkflowV1alpha1WorkflowCreateRequest,
|
13 | 18 | )
|
14 |
| -from argo_workflows.model.io_argoproj_workflow_v1alpha1_workflow_terminate_request import ( |
15 |
| - IoArgoprojWorkflowV1alpha1WorkflowTerminateRequest, |
16 |
| -) |
17 | 19 | from argo_workflows.model.io_argoproj_workflow_v1alpha1_workflow_retry_request import (
|
18 | 20 | IoArgoprojWorkflowV1alpha1WorkflowRetryRequest,
|
19 | 21 | )
|
20 |
| -from argo_workflows.model.io_argoproj_workflow_v1alpha1_retry_archived_workflow_request import ( |
21 |
| - IoArgoprojWorkflowV1alpha1RetryArchivedWorkflowRequest, |
| 22 | +from argo_workflows.model.io_argoproj_workflow_v1alpha1_workflow_terminate_request import ( |
| 23 | + IoArgoprojWorkflowV1alpha1WorkflowTerminateRequest, |
22 | 24 | )
|
23 |
| -from argo_workflows.exceptions import NotFoundException |
24 | 25 |
|
25 | 26 | from argowrapper import logger
|
26 | 27 | from argowrapper.constants import (
|
27 | 28 | ARGO_HOST,
|
28 | 29 | ARGO_NAMESPACE,
|
29 |
| - WORKFLOW, |
30 |
| - GEN3_USER_METADATA_LABEL, |
| 30 | + GEN3_SUBMIT_TIMESTAMP_LABEL, |
31 | 31 | GEN3_TEAM_PROJECT_METADATA_LABEL,
|
| 32 | + GEN3_USER_METADATA_LABEL, |
32 | 33 | GEN3_WORKFLOW_PHASE_LABEL,
|
33 |
| - GEN3_SUBMIT_TIMESTAMP_LABEL, |
| 34 | + WORKFLOW, |
34 | 35 | )
|
35 | 36 | from argowrapper.engine.helpers import argo_engine_helper
|
36 | 37 | from argowrapper.engine.helpers.workflow_factory import WorkflowFactory
|
37 | 38 | from argowrapper.workflows.argo_workflows.gwas import GWAS
|
38 | 39 |
|
39 |
| -from datetime import datetime |
40 |
| - |
41 | 40 |
|
42 | 41 | class ArgoEngine:
|
43 | 42 | """
|
@@ -71,7 +70,6 @@ def __init__(self, dry_run: bool = False):
|
71 | 70 | self.artifact_api_instance = artifact_service_api.ArtifactServiceApi(api_client)
|
72 | 71 |
|
73 | 72 | def _get_workflow_details_dict(self, workflow_name: str) -> Dict:
|
74 |
| - |
75 | 73 | return self.api_instance.get_workflow(
|
76 | 74 | namespace=ARGO_NAMESPACE,
|
77 | 75 | name=workflow_name,
|
@@ -418,7 +416,9 @@ def get_user_workflows_for_current_month(self, auth_header: str) -> List[Dict]:
|
418 | 416 |
|
419 | 417 | def get_workflows_for_label_selector(self, label_selector: str) -> List[Dict]:
|
420 | 418 | try:
|
421 |
| - logger.info(f"ARGO_NAMESPACE: {ARGO_NAMESPACE}") |
| 419 | + logger.info( |
| 420 | + f"ARGO_NAMESPACE for get_workflows_for_label_selector: {ARGO_NAMESPACE}" |
| 421 | + ) |
422 | 422 | workflow_list_return = self.api_instance.list_workflows(
|
423 | 423 | namespace=ARGO_NAMESPACE,
|
424 | 424 | list_options_label_selector=label_selector,
|
|
0 commit comments