vortex-nwp 2.0.0b1__py3-none-any.whl → 2.1.0__py3-none-any.whl

vortex/nwp/util/usetnt.py

@@ -14,7 +14,7 @@ from vortex.util.roles import setrole
 
 logger = loggers.getLogger(__name__)
 
-tnt_checker = ExternalCodeImportChecker('thenamelisttool')
+tnt_checker = ExternalCodeImportChecker("thenamelisttool")
 with tnt_checker as tnt_register:
     import thenamelisttool
 
@@ -48,40 +48,54 @@ def compose_nam(options):
 
     :rtype: A file like object
     """
-    rhdict = options.get('rhandler', None)
-    source = rhdict['resource'].get('source', None)
+    rhdict = options.get("rhandler", None)
+    source = rhdict["resource"].get("source", None)
     if source is None:
         logger.error("Inapropriate type of resource. Got:\n%s", rhdict)
-        raise FunctionStoreCallbackError('Inapropriate type of resources.')
+        raise FunctionStoreCallbackError("Inapropriate type of resources.")
 
     t = sessions.current()
 
     def _get_pack_adress(role):
         role = setrole(role)
-        packlist = [sec.rh for sec in t.context.sequence.filtered_inputs(role=role)]
+        packlist = [
+            sec.rh for sec in t.context.sequence.filtered_inputs(role=role)
+        ]
         if len(packlist) != 1:
             logger.error("The number of namelist packs with role=%s is not 1.")
-            raise FunctionStoreCallbackError('Incorrect number of namelist packs.')
+            raise FunctionStoreCallbackError(
+                "Incorrect number of namelist packs."
+            )
         packrh = packlist[0]
-        if packrh.resource.realkind != 'namelistpack':
-            logger.error("Incorrect resource type for role %s. Resource handler:\n%s",
-                         role, packrh.icdard())
-            raise FunctionStoreCallbackError('Incorrect resource type.')
+        if packrh.resource.realkind != "namelistpack":
+            logger.error(
+                "Incorrect resource type for role %s. Resource handler:\n%s",
+                role,
+                packrh.icdard(),
+            )
+            raise FunctionStoreCallbackError("Incorrect resource type.")
         if not packrh.container.filled:
-            logger.error("The resource handler's container is not filled for role %s", role)
-            raise FunctionStoreCallbackError('RH container is not filled.')
+            logger.error(
+                "The resource handler's container is not filled for role %s",
+                role,
+            )
+            raise FunctionStoreCallbackError("RH container is not filled.")
         return packrh.container.abspath
 
-    nampack_path = _get_pack_adress(options.get('nampack', 'Main Namelist Pack'))
-    dirpack_role = options.get('dirpack', None)
-    dirpack_path = _get_pack_adress(dirpack_role) if dirpack_role else nampack_path
+    nampack_path = _get_pack_adress(
+        options.get("nampack", "Main Namelist Pack")
+    )
+    dirpack_role = options.get("dirpack", None)
+    dirpack_path = (
+        _get_pack_adress(dirpack_role) if dirpack_role else nampack_path
+    )
 
     out_io = io.BytesIO()
     thenamelisttool.util.compose_namelist(
-        t.sh.path.join(dirpack_path, source + '.yaml'),
+        t.sh.path.join(dirpack_path, source + ".yaml"),
         sourcenam_directory=nampack_path,
         sorting=thenamelisttool.namadapter.FIRST_ORDER_SORTING,
         squeeze=False,
-        fhoutput=codecs.getwriter('utf-8')(out_io)
+        fhoutput=codecs.getwriter("utf-8")(out_io),
     )
     return out_io

vortex/sessions.py

@@ -17,7 +17,7 @@ import footprints
 
 from vortex.tools.env import Environment
 
-from vortex import gloves  # @UnusedImport
+from vortex import gloves as gloves  # footprints import
 from vortex.layout import contexts
 
 #: No automatic export
@@ -28,9 +28,13 @@ logger = loggers.getLogger(__name__)
 
 # Module Interface
 
+
 def get(**kw):
     """Return actual session ticket object matching description."""
-    if kw.get('tag', 'current') == 'current' and Ticket.tag_focus() is not None:
+    if (
+        kw.get("tag", "current") == "current"
+        and Ticket.tag_focus() is not None
+    ):
         return current()
     else:
         return Ticket(**kw)
@@ -73,7 +77,7 @@ def getglove(**kw):
 
 def system(**kw):
     """Returns the system associated to the current ticket."""
-    return get(tag=kw.pop('tag', Ticket.tag_focus())).system(**kw)
+    return get(tag=kw.pop("tag", Ticket.tag_focus())).system(**kw)
 
 
 # noinspection PyShadowingBuiltins
@@ -94,17 +98,19 @@ class Ticket(getbytag.GetByTag):
     Default session ticket class, defined by tag.
     """
 
-    _tag_default = 'root'
-
-    def __init__(self,
-                 active=False,
-                 topenv=None,
-                 glove=None,
-                 context=None,
-                 datastore=None,
-                 prompt='Vortex:'):
+    _tag_default = "root"
+
+    def __init__(
+        self,
+        active=False,
+        topenv=None,
+        glove=None,
+        context=None,
+        datastore=None,
+        prompt="Vortex:",
+    ):
         self.prompt = prompt
-        self.line = "\n" + '-' * 100 + "\n"
+        self.line = "\n" + "-" * 100 + "\n"
 
         self._started = date.now()
         self._closed = 0
@@ -120,18 +126,24 @@ class Ticket(getbytag.GetByTag):
         else:
             self._glove = getglove()
 
-        logger.debug('New session system is %s', self.system())
+        logger.debug("New session system is %s", self.system())
 
         self._rundir = self.sh.getcwd()
 
-        logger.debug('Open session %s %s', self.tag, self)
+        logger.debug("Open session %s %s", self.tag, self)
 
         if datastore is None:
-            datastore = DataStore(default_picklefile='{:s}_session_datastore.pickled'.format(self.tag))
+            datastore = DataStore(
+                default_picklefile="{:s}_session_datastore.pickled".format(
+                    self.tag
+                )
+            )
         self._dstore = datastore
 
         if context is None:
-            context = contexts.Context(tag=self.tag, topenv=self._topenv, path=self.path)
+            context = contexts.Context(
+                tag=self.tag, topenv=self._topenv, path=self.path
+            )
         self._last_context = context
 
         if active:
@@ -144,12 +156,20 @@ class Ticket(getbytag.GetByTag):
     def _set_rundir(self, path):
         """Set a new default rundir for this session."""
         if self._rundir:
-            logger.warning('Session <%s> is changing its working directory <%s>', self.tag, self._rundir)
+            logger.warning(
+                "Session <%s> is changing its working directory <%s>",
+                self.tag,
+                self._rundir,
+            )
         if self.sh.path.isdir(path):
             self._rundir = path
-            logger.info('Session <%s> set rundir <%s>', self.tag, self._rundir)
+            logger.info("Session <%s> set rundir <%s>", self.tag, self._rundir)
         else:
-            logger.error('Try to change session <%s> to invalid path <%s>', self.tag, path)
+            logger.error(
+                "Try to change session <%s> to invalid path <%s>",
+                self.tag,
+                path,
+            )
 
     rundir = property(_get_rundir, _set_rundir)
 
@@ -210,11 +230,14 @@ class Ticket(getbytag.GetByTag):
         Returns the current OS handler used or set a new one according
         to ``kw`` dictionary-like arguments.
         """
-        refill = kw.pop('refill', False)
+        refill = kw.pop("refill", False)
         if not self._system or kw or refill:
             self._system = footprints.proxy.system(glove=self.glove, **kw)
             if not self._system:
-                logger.critical('Could not load a system object with description %s', str(kw))
+                logger.critical(
+                    "Could not load a system object with description %s",
+                    str(kw),
+                )
         return self._system
 
     def duration(self):
@@ -237,27 +260,31 @@ class Ticket(getbytag.GetByTag):
     def close(self):
         """Closes the current session."""
         if self.closed:
-            logger.warning('Session %s already closed at %s', self.tag, self.closed)
+            logger.warning(
+                "Session %s already closed at %s", self.tag, self.closed
+            )
         else:
             self._closed = date.now()
-            logger.debug('Close session %s ( time = %s )', self.tag, self.duration())
+            logger.debug(
+                "Close session %s ( time = %s )", self.tag, self.duration()
+            )
 
     @property
     def path(self):
-        return '/' + self.tag
+        return "/" + self.tag
 
     @property
     def subcontexts(self):
         """The current contexts binded to this session."""
-        rootpath = self.path + '/'
+        rootpath = self.path + "/"
         return [x for x in contexts.values() if x.path.startswith(rootpath)]
 
     def exit(self):
         """Exit from the current session."""
         ok = True
-        logger.debug('Exit session %s %s', self.tag, self)
+        logger.debug("Exit session %s %s", self.tag, self)
         for kid in self.subcontexts:
-            logger.debug('Exit from context %s', kid)
+            logger.debug("Exit from context %s", kid)
             ok = ok and kid.exit()
         if self.opened:
             self.close()
@@ -295,20 +322,26 @@ class Ticket(getbytag.GetByTag):
         """
         Returns the logging level.
         """
-        v_logger = loggers.getLogger('vortex')
+        v_logger = loggers.getLogger("vortex")
         return logging.getLevelName(v_logger.getEffectiveLevel())
 
-    def idcard(self, indent='+ '):
+    def idcard(self, indent="+ "):
         """Returns a printable description of the current session."""
-        card = "\n".join((
-            '{0}Name     = {1:s}',
-            '{0}Started  = {2!s}',
-            '{0}Opened   = {3!s}',
-            '{0}Duration = {4!s}',
-            '{0}Loglevel = {5:s}'
-        )).format(
+        card = "\n".join(
+            (
+                "{0}Name     = {1:s}",
+                "{0}Started  = {2!s}",
+                "{0}Opened   = {3!s}",
+                "{0}Duration = {4!s}",
+                "{0}Loglevel = {5:s}",
+            )
+        ).format(
             indent,
-            self.tag, self.started, self.opened, self.duration(), self.loglevel
+            self.tag,
+            self.started,
+            self.opened,
+            self.duration(),
+            self.loglevel,
         )
         return card
 
@@ -333,7 +366,7 @@ class Ticket(getbytag.GetByTag):
             obj.catch_focus()
             return obj
         else:
-            logger.error('Try to switch to an undefined session: %s', tag)
+            logger.error("Try to switch to an undefined session: %s", tag)
             return None
 
     def __del__(self):

vortex/syntax/__init__.py

@@ -6,4 +6,4 @@ The most important usage is done by :class:`FootprintBase` derivated objects.
 #: No automatic export
 __all__ = []
 
-__tocinfoline__ = 'VORTEX package where standard attributes are defined.'
+__tocinfoline__ = "VORTEX package where standard attributes are defined."