Skip to content
Open
Show file tree
Hide file tree
Changes from 1 commit
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
4 changes: 2 additions & 2 deletions src/radical/utils/flux/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,6 @@
from .flux_module import FluxModule, spec_from_command, spec_from_dict

_fm = FluxModule()
if _fm.version == 1: FluxHelper = _FluxHelperV1
else : FluxHelper = _FluxHelperV0
if _fm.mode == 1: FluxHelper = _FluxHelperV1
else : FluxHelper = _FluxHelperV0

2 changes: 2 additions & 0 deletions src/radical/utils/flux/flux_helper_v0.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,6 +44,8 @@ def __init__(self, uri : str,
self._cbacks = list() # list of callbacks

self._fm.verify()
for line in self._fm.version.splitlines():
self._log.info('flux version: %s', line)


# --------------------------------------------------------------------------
Expand Down
2 changes: 2 additions & 0 deletions src/radical/utils/flux/flux_helper_v1.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,6 +54,8 @@ def __init__(self, uri : str,
self._cbacks = list() # list of callbacks

self._fm.verify()
for line in self._fm.version.splitlines():
self._log.info('flux version: %s', line)


# --------------------------------------------------------------------------
Expand Down
28 changes: 20 additions & 8 deletions src/radical/utils/flux/flux_module.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@ class FluxModule(object):
_flux_job = None
_flux_exc = None
_flux_v = None
_flux_mode = None


# --------------------------------------------------------------------------
Expand All @@ -32,18 +33,24 @@ def __init__(self):
if self._flux_core or self._flux_job or self._flux_exc:
return

flux = None
flux_job = None
flux_exc = None
flux_v = None
flux = None
flux_job = None
flux_exc = None
flux_v = None
flux_mode = None

try:
out, err, ret = sh_callout('flux --version')
assert not ret, ['flux --version', err]

flux_v = out.strip()

flux = import_module('flux')
flux_job = import_module('flux.job')
if 'JournalConsumer' in dir(flux_job):
flux_v = 1
flux_mode = 1
else:
flux_v = 0
flux_mode = 0

except Exception as e:
flux_exc = e
Expand All @@ -67,9 +74,9 @@ def __init__(self):
flux = import_module('flux')
flux_job = import_module('flux.job')
if 'JournalConsumer' in dir(flux_job):
flux_v = 1
flux_mode = 1
else:
flux_v = 0
flux_mode = 0

except Exception as e:
flux_exc = e
Expand All @@ -81,6 +88,7 @@ def __init__(self):
self._flux_job = flux_job
self._flux_exc = flux_exc
self._flux_v = flux_v
self._flux_mode = flux_mode
self._flux_exe = which('flux')


Expand All @@ -103,6 +111,10 @@ def verify(self) -> None:

# --------------------------------------------------------------------------
#
@property
def mode(self) -> int:
return self._flux_mode

@property
def version(self) -> int:
return self._flux_v
Expand Down