Skip to content
Open
Show file tree
Hide file tree
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
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 or '').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 or '').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