airflow-unicore-integration 0.1.5__py3-none-any.whl → 0.1.6__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (7) hide show
  1. airflow_unicore_integration/executors/unicore_executor.py +6 -2
  2. {airflow_unicore_integration-0.1.5.dist-info → airflow_unicore_integration-0.1.6.dist-info}/METADATA +1 -1
  3. {airflow_unicore_integration-0.1.5.dist-info → airflow_unicore_integration-0.1.6.dist-info}/RECORD +7 -7
  4. {airflow_unicore_integration-0.1.5.dist-info → airflow_unicore_integration-0.1.6.dist-info}/WHEEL +0 -0
  5. {airflow_unicore_integration-0.1.5.dist-info → airflow_unicore_integration-0.1.6.dist-info}/entry_points.txt +0 -0
  6. {airflow_unicore_integration-0.1.5.dist-info → airflow_unicore_integration-0.1.6.dist-info}/licenses/LICENSE +0 -0
  7. {airflow_unicore_integration-0.1.5.dist-info → airflow_unicore_integration-0.1.6.dist-info}/top_level.txt +0 -0
airflow_unicore_integration/executors/unicore_executor.py CHANGED
@@ -39,6 +39,12 @@ STATE_MAPPINGS: Dict[uc_client.JobStatus, TaskInstanceState] = {
39
39
 
40
40
  class UnicoreExecutor(BaseExecutor):
41
41
 
42
+ EXECUTOR_CONFIG_UNICORE_CONN_KEY = (
43
+ "unicore_connection_id" # alternative connection id for the Unicore connection to use
44
+ )
45
+ EXECUTOR_CONFIG_UNICORE_SITE_KEY = "unicore_site" # alternative Unicore site to run at, only required if different than connection default
46
+ EXECUTOR_CONFIG_UNICORE_CREDENTIAL_KEY = "unicore_credential" # alternative unicore credential to use for the job, only required if different than connection default
47
+
42
48
  def start(self):
43
49
  self.active_jobs: Dict[TaskInstanceKey, uc_client.Job] = {}
44
50
  self.uc_conn = unicore_hooks.UnicoreHook().get_conn()
@@ -67,9 +73,7 @@ class UnicoreExecutor(BaseExecutor):
67
73
  return []
68
74
 
69
75
  def _get_unicore_client(self, executor_config: dict | None = {}):
70
- # TODO fix this only temporary solution
71
76
  return self.uc_conn
72
- # END TODO fix this
73
77
  # include client desires from executor_config
74
78
  unicore_conn_id = executor_config.get( # type: ignore
75
79
  UnicoreExecutor.EXECUTOR_CONFIG_UNICORE_CONN_KEY,
{airflow_unicore_integration-0.1.5.dist-info → airflow_unicore_integration-0.1.6.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: airflow-unicore-integration
3
- Version: 0.1.5
3
+ Version: 0.1.6
4
4
  Summary: Running Unicore Jobs from airflow DAGs.
5
5
  Author-email: Christian Böttcher <c.boettcher@fz-juelich.de>
6
6
  License-Expression: BSD-3-Clause
{airflow_unicore_integration-0.1.5.dist-info → airflow_unicore_integration-0.1.6.dist-info}/RECORD RENAMED
@@ -1,16 +1,16 @@
1
1
  airflow_unicore_integration/__init__.py,sha256=sibcOlQKBCUKfTbZlAl0L7r1d2nxf0nGAQys612vzeA,429
2
2
  airflow_unicore_integration/executors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
3
3
  airflow_unicore_integration/executors/run_task_via_supervisor.py,sha256=3ErgPf-Oy3B4Di5yNXhhPkaojIJykvCxMZ9MlKSYPI8,2756
4
- airflow_unicore_integration/executors/unicore_executor.py,sha256=f2PKj66tZlWb39xF5YacEBoZ2-eZ_JR9pnKQwNHAybc,5619
4
+ airflow_unicore_integration/executors/unicore_executor.py,sha256=EhGsGksSRd1T_dxzQxpGhuCDadkCqvZodbwK1XWVVnM,6001
5
5
  airflow_unicore_integration/hooks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
6
6
  airflow_unicore_integration/hooks/unicore_hooks.py,sha256=JjcjogWtN1xveagpkraQuYOdXjkp2lSnEdQc0waqhU4,1662
7
7
  airflow_unicore_integration/operators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
8
8
  airflow_unicore_integration/operators/unicore_operators.py,sha256=xYX1t_QzjpqHZDoEu6jJNNXVmBEnjIlApFWvgYoYoB0,16790
9
9
  airflow_unicore_integration/policies/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
10
10
  airflow_unicore_integration/util/job.py,sha256=BSP-XuWL5-TRGACX__MavIcmPAA52gYLdwWL236cVz8,6703
11
- airflow_unicore_integration-0.1.5.dist-info/licenses/LICENSE,sha256=hZ5ouAedeNr8ClHrQE-RLsgMsARcmv3kSZz7tE2BTJE,1526
12
- airflow_unicore_integration-0.1.5.dist-info/METADATA,sha256=ZtH2sUzOSHxsDmVSIgb-yJ61i7bi9Zc2olzdPhaeMHE,11188
13
- airflow_unicore_integration-0.1.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
14
- airflow_unicore_integration-0.1.5.dist-info/entry_points.txt,sha256=PzEfCLYLSawjiYR-HNBzw8-YGfJxs1nPBULevgBQjoY,147
15
- airflow_unicore_integration-0.1.5.dist-info/top_level.txt,sha256=j45X-uIuOk3oL78iwlpHakMWtUkg__B7zUlJLwmZx6w,28
16
- airflow_unicore_integration-0.1.5.dist-info/RECORD,,
11
+ airflow_unicore_integration-0.1.6.dist-info/licenses/LICENSE,sha256=hZ5ouAedeNr8ClHrQE-RLsgMsARcmv3kSZz7tE2BTJE,1526
12
+ airflow_unicore_integration-0.1.6.dist-info/METADATA,sha256=_V_fEL6Rp-V2T7tUp6R6DRkNjWUrXB_eElcTsAP0Jj8,11188
13
+ airflow_unicore_integration-0.1.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
14
+ airflow_unicore_integration-0.1.6.dist-info/entry_points.txt,sha256=PzEfCLYLSawjiYR-HNBzw8-YGfJxs1nPBULevgBQjoY,147
15
+ airflow_unicore_integration-0.1.6.dist-info/top_level.txt,sha256=j45X-uIuOk3oL78iwlpHakMWtUkg__B7zUlJLwmZx6w,28
16
+ airflow_unicore_integration-0.1.6.dist-info/RECORD,,