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

vortex/nwp/tools/igastuff.py

@@ -11,56 +11,70 @@ __all__ = []
 fuzzystr = dict(
     histfix=dict(
         historic=dict(
-            pearp='prev', arome='AROM', arpege='arpe', arp_court='arpe',
-            aearp='arpe', aladin='ALAD', surfex='SURF'
+            pearp="prev",
+            arome="AROM",
+            arpege="arpe",
+            arp_court="arpe",
+            aearp="arpe",
+            aladin="ALAD",
+            surfex="SURF",
         )
     ),
     prefix=dict(
         # LFM 2016/12/30: It was dble='PA' but apparently it's wrong. No idea why...
-        gridpoint=dict(oper='PE', dble='PE', mirr='PE', hycom_grb='vent'),
-        historic=dict(hycom='s_init0_', mfwam='BLS_', mfwam_BLS='BLS_', mfwam_LAW='LAW_'),
-        analysis=dict(hycom='s_init0_', mfwam='LAW_')
+        gridpoint=dict(oper="PE", dble="PE", mirr="PE", hycom_grb="vent"),
+        historic=dict(
+            hycom="s_init0_", mfwam="BLS_", mfwam_BLS="BLS_", mfwam_LAW="LAW_"
+        ),
+        analysis=dict(hycom="s_init0_", mfwam="LAW_"),
     ),
     suffix=dict(
-        bgstderr=dict(input='in', output='out'),
-        analysis=dict(hycom_hycom='.gz', hycom_surcotes='.gz', hycom_surcotes_oi='.gz'),
-        historic=dict(surfex_arpege='.sfx', surfex_aearp='.sfx',
-                      hycom_hycom='.gz', hycom_surcotes='.gz', hycom_surcotes_oi='.gz'),
-        gridpoint=dict(hycom_grb='grb'),
+        bgstderr=dict(input="in", output="out"),
+        analysis=dict(
+            hycom_hycom=".gz", hycom_surcotes=".gz", hycom_surcotes_oi=".gz"
+        ),
+        historic=dict(
+            surfex_arpege=".sfx",
+            surfex_aearp=".sfx",
+            hycom_hycom=".gz",
+            hycom_surcotes=".gz",
+            hycom_surcotes_oi=".gz",
+        ),
+        gridpoint=dict(hycom_grb="grb"),
     ),
     term0003=dict(
-        bgstderr=dict(input='', output='_assim'),
+        bgstderr=dict(input="", output="_assim"),
     ),
     term0009=dict(
-        bgstderr=dict(input='', output='_production'),
+        bgstderr=dict(input="", output="_production"),
     ),
     term0012=dict(
-        bgstderr=dict(input='_production_dsbscr', output='_production_dsbscr'),
+        bgstderr=dict(input="_production_dsbscr", output="_production_dsbscr"),
     ),
     varbcarpege=dict(
-        varbc=dict(input='.cycle_arp', output='.cycle'),
+        varbc=dict(input=".cycle_arp", output=".cycle"),
     ),
     varbcaladin=dict(
-        varbc=dict(input='.cycle_alad', output='.cycle'),
+        varbc=dict(input=".cycle_alad", output=".cycle"),
     ),
     varbcarome=dict(
-        varbc=dict(input='.cycle_aro', output='.cycle'),
+        varbc=dict(input=".cycle_aro", output=".cycle"),
     ),
     surf0000=dict(
-        histsurf=dict(input='INIT_SURF', output='INIT_SURF'),
-        historic=dict(input='INIT_SURF', output='INIT_SURF'),
+        histsurf=dict(input="INIT_SURF", output="INIT_SURF"),
+        historic=dict(input="INIT_SURF", output="INIT_SURF"),
     ),
     surf0003=dict(
-        histsurf=dict(input='PREP', output='AROMOUT_.0003'),
-        historic=dict(input='PREP', output='AROMOUT_.0003'),
+        histsurf=dict(input="PREP", output="AROMOUT_.0003"),
+        historic=dict(input="PREP", output="AROMOUT_.0003"),
     ),
     surf0006=dict(
-        histsurf=dict(input='PREP', output='AROMOUT_.0006'),
-        historic=dict(input='PREP', output='AROMOUT_.0006'),
+        histsurf=dict(input="PREP", output="AROMOUT_.0006"),
+        historic=dict(input="PREP", output="AROMOUT_.0006"),
     ),
 )
 
-arpcourt_vconf = ('courtfr', 'frcourt', 'court')
+arpcourt_vconf = ("courtfr", "frcourt", "court")
 
 
 def fuzzyname(entry, realkind, key, default=None):
@@ -81,35 +95,21 @@ def archive_suffix(model, cutoff, date, vconf=None):
     for h in hh:
         hrange.append("%02d" % h)
 
-    if cutoff == 'assim':
-        rr = dict(
-            zip(
-                zip(
-                    (cutoff,) * len(hrange),
-                    hh
-                ),
-                hrange
-            )
-        )
+    if cutoff == "assim":
+        rr = dict(zip(zip((cutoff,) * len(hrange), hh), hrange))
     else:
-        if re.search(r'court|arome', model) or vconf in arpcourt_vconf:
+        if re.search(r"court|arome", model) or vconf in arpcourt_vconf:
             rr = dict(
                 zip(
-                    zip(
-                        (cutoff,) * len(hrange),
-                        hh
-                    ),
-                    ('CM', 'TR', 'SX', 'NF', 'PM', 'QZ', 'DH', 'VU')
+                    zip((cutoff,) * len(hrange), hh),
+                    ("CM", "TR", "SX", "NF", "PM", "QZ", "DH", "VU"),
                 )
             )
         else:
             rr = dict(
                 zip(
-                    zip(
-                        (cutoff,) * len(hrange),
-                        hh
-                    ),
-                    ('AM', 'TR', 'SX', 'NF', 'PM', 'QZ', 'DH', 'VU')
+                    zip((cutoff,) * len(hrange), hh),
+                    ("AM", "TR", "SX", "NF", "PM", "QZ", "DH", "VU"),
                 )
             )
 
@@ -123,7 +123,7 @@ class _BaseIgakeyFactory(str):
     Needs to be subclassed !
     """
 
-    _re_appconf = re.compile(r'^(\w+)/([\w@]+)$')
+    _re_appconf = re.compile(r"^(\w+)/([\w@]+)$")
     _keymap = {}
 
     def __new__(cls, value):
@@ -134,9 +134,9 @@ class _BaseIgakeyFactory(str):
         """
         val_split = cls._re_appconf.match(value)
         if val_split:
-            value = cls._keymap.get(val_split.group(1),
-                                    {}).get(val_split.group(2),
-                                            val_split.group(1))
+            value = cls._keymap.get(val_split.group(1), {}).get(
+                val_split.group(2), val_split.group(1)
+            )
         return str.__new__(cls, value)
 
 
@@ -145,52 +145,65 @@ class IgakeyFactoryArchive(_BaseIgakeyFactory):
     Given the vapp/vconf, returns a default value for the igakey attribute
     """
 
-    _keymap = {'arpege': {'4dvarfr': 'arpege',
-                          '4dvar': 'arpege',
-                          'pearp': 'pearp',
-                          'aearp': 'aearp',
-                          'courtfr': 'arpege',
-                          'frcourt': 'arpege',
-                          'court': 'arpege', },
-               'mocage': {'camsfcst': 'macc',
-                          'camsassim': 'macc', },
-               'arome': {'3dvarfr': 'arome',
-                         'france': 'arome',
-                         'pegase': 'pegase', },
-               'aladin': {'antiguy': 'antiguy',
-                          'caledonie': 'caledonie',
-                          'nc': 'caledonie',
-                          'polynesie': 'polynesie',
-                          'reunion': 'reunion', },
-               'hycom': {'atl@anarp': 'surcotes',
-                         'med@anarp': 'surcotes',
-                         'atl@fcarp': 'surcotes',
-                         'med@fcarp': 'surcotes',
-                         'atl@anaro': 'surcotes',
-                         'med@anaro': 'surcotes',
-                         'atl@fcaro': 'surcotes',
-                         'med@fcaro': 'surcotes',
-                         'atl@fcaoc': 'surcotes',
-                         'med@fcaoc': 'surcotes',
-                         'oin@ancep': 'surcotes_oi',
-                         'oin@fcaro': 'surcotes_oi', },
-               'mfwam': {'globalcep02': 'mfwamglocep02',
-                         'globalcep01': 'mfwamglocep01',
-                         'reuaro01': 'mfwamreuaro',
-                         'polyaro01': 'mfwampolyaro',
-                         'caledaro01': 'mfwamcaledaro',
-                         'globalarp02': 'mfwamgloarp02',
-                         'globalarpc02': 'mfwamgloarpc02',
-                         'atourxarp01': 'mfwamatourx01arp',
-                         'euratarpc01': 'mfwameurcourt',
-                         'frangparo0025': 'mfwamfrangp0025',
-                         'frangparoifs0025': 'mfwamfrangp0025ifs',
-                         'assmp1': 'mfwamassmp1',
-                         'assmp2': 'mfwamassmp2',
-                         'assms1': 'mfwamassms1',
-                         'assms2': 'mfwamassms2',
-                         'angola0025': 'mfwamangola', },
-               }
+    _keymap = {
+        "arpege": {
+            "4dvarfr": "arpege",
+            "4dvar": "arpege",
+            "pearp": "pearp",
+            "aearp": "aearp",
+            "courtfr": "arpege",
+            "frcourt": "arpege",
+            "court": "arpege",
+        },
+        "mocage": {
+            "camsfcst": "macc",
+            "camsassim": "macc",
+        },
+        "arome": {
+            "3dvarfr": "arome",
+            "france": "arome",
+            "pegase": "pegase",
+        },
+        "aladin": {
+            "antiguy": "antiguy",
+            "caledonie": "caledonie",
+            "nc": "caledonie",
+            "polynesie": "polynesie",
+            "reunion": "reunion",
+        },
+        "hycom": {
+            "atl@anarp": "surcotes",
+            "med@anarp": "surcotes",
+            "atl@fcarp": "surcotes",
+            "med@fcarp": "surcotes",
+            "atl@anaro": "surcotes",
+            "med@anaro": "surcotes",
+            "atl@fcaro": "surcotes",
+            "med@fcaro": "surcotes",
+            "atl@fcaoc": "surcotes",
+            "med@fcaoc": "surcotes",
+            "oin@ancep": "surcotes_oi",
+            "oin@fcaro": "surcotes_oi",
+        },
+        "mfwam": {
+            "globalcep02": "mfwamglocep02",
+            "globalcep01": "mfwamglocep01",
+            "reuaro01": "mfwamreuaro",
+            "polyaro01": "mfwampolyaro",
+            "caledaro01": "mfwamcaledaro",
+            "globalarp02": "mfwamgloarp02",
+            "globalarpc02": "mfwamgloarpc02",
+            "atourxarp01": "mfwamatourx01arp",
+            "euratarpc01": "mfwameurcourt",
+            "frangparo0025": "mfwamfrangp0025",
+            "frangparoifs0025": "mfwamfrangp0025ifs",
+            "assmp1": "mfwamassmp1",
+            "assmp2": "mfwamassmp2",
+            "assms1": "mfwamassms1",
+            "assms2": "mfwamassms2",
+            "angola0025": "mfwamangola",
+        },
+    }
 
 
 class IgakeyFactoryInline(_BaseIgakeyFactory):
@@ -198,52 +211,63 @@ class IgakeyFactoryInline(_BaseIgakeyFactory):
     Given the vapp/vconf, returns a default value for the igakey attribute
     """
 
-    _keymap = {'arpege': {'4dvarfr': 'france',
-                          '4dvar': 'france',
-                          'pearp': 'pearp',
-                          'aearp': 'aearp',
-                          'courtfr': 'frcourt',
-                          'frcourt': 'frcourt',
-                          'court': 'frcourt', },
-               'arome': {'3dvarfr': 'france',
-                         'france': 'france',
-                         'pegase': 'pegase', },
-               'aladin': {'antiguy': 'antiguy',
-                          'caledonie': 'caledonie',
-                          'nc': 'caledonie',
-                          'polynesie': 'polynesie',
-                          'reunion': 'reunion', },
-               'hycom': {'atl@anarp': 'surcotes',
-                         'med@anarp': 'surcotes',
-                         'atl@fcarp': 'surcotes',
-                         'med@fcarp': 'surcotes',
-                         'atl@ancep': 'surcotes',
-                         'med@ancep': 'surcotes',
-                         'atl@fccep': 'surcotes',
-                         'med@fccep': 'surcotes',
-                         'atl@anaro': 'surcotes',
-                         'med@anaro': 'surcotes',
-                         'atl@fcaro': 'surcotes',
-                         'med@fcaro': 'surcotes',
-                         'atl@red': 'surcotes',
-                         'med@red': 'surcotes',
-                         'oin@ancep': 'surcotes_oi',
-                         'oin@fcaro': 'surcotes_oi',
-                         'oin@red': 'surcotes_oi', },
-               'mfwam': {'globalcep02': 'mfwamglocep02',
-                         'globalcep01': 'mfwamglocep01',
-                         'reuaro01': 'mfwamreuaro',
-                         'polyaro01': 'mfwampolyaro',
-                         'caledaro01': 'mfwamcaledaro',
-                         'globalarp02': 'mfwamgloarp02',
-                         'globalarpc02': 'mfwamgloarpc02',
-                         'atourxarp01': 'mfwamatourx01arp',
-                         'euratarpc01': 'mfwameurcourt',
-                         'frangparo0025': 'mfwamfrangp0025',
-                         'frangparoifs0025': 'mfwamfrangp0025ifs',
-                         'assmp1': 'mfwamassmp1',
-                         'assmp2': 'mfwamassmp2',
-                         'assms1': 'mfwamassms1',
-                         'assms2': 'mfwamassms2',
-                         'angola0025': 'mfwamangola', },
-               }
+    _keymap = {
+        "arpege": {
+            "4dvarfr": "france",
+            "4dvar": "france",
+            "pearp": "pearp",
+            "aearp": "aearp",
+            "courtfr": "frcourt",
+            "frcourt": "frcourt",
+            "court": "frcourt",
+        },
+        "arome": {
+            "3dvarfr": "france",
+            "france": "france",
+            "pegase": "pegase",
+        },
+        "aladin": {
+            "antiguy": "antiguy",
+            "caledonie": "caledonie",
+            "nc": "caledonie",
+            "polynesie": "polynesie",
+            "reunion": "reunion",
+        },
+        "hycom": {
+            "atl@anarp": "surcotes",
+            "med@anarp": "surcotes",
+            "atl@fcarp": "surcotes",
+            "med@fcarp": "surcotes",
+            "atl@ancep": "surcotes",
+            "med@ancep": "surcotes",
+            "atl@fccep": "surcotes",
+            "med@fccep": "surcotes",
+            "atl@anaro": "surcotes",
+            "med@anaro": "surcotes",
+            "atl@fcaro": "surcotes",
+            "med@fcaro": "surcotes",
+            "atl@red": "surcotes",
+            "med@red": "surcotes",
+            "oin@ancep": "surcotes_oi",
+            "oin@fcaro": "surcotes_oi",
+            "oin@red": "surcotes_oi",
+        },
+        "mfwam": {
+            "globalcep02": "mfwamglocep02",
+            "globalcep01": "mfwamglocep01",
+            "reuaro01": "mfwamreuaro",
+            "polyaro01": "mfwampolyaro",
+            "caledaro01": "mfwamcaledaro",
+            "globalarp02": "mfwamgloarp02",
+            "globalarpc02": "mfwamgloarpc02",
+            "atourxarp01": "mfwamatourx01arp",
+            "euratarpc01": "mfwameurcourt",
+            "frangparo0025": "mfwamfrangp0025",
+            "frangparoifs0025": "mfwamfrangp0025ifs",
+            "assmp1": "mfwamassmp1",
+            "assmp2": "mfwamassmp2",
+            "assms1": "mfwamassms1",
+            "assms2": "mfwamassms2",
+            "angola0025": "mfwamangola",
+        },
+    }

vortex/nwp/tools/mars.py

@@ -14,16 +14,19 @@ logger = loggers.getLogger(__name__)
 
 class MarsError(Exception):
     """General Mars error."""
+
     pass
 
 
 class MarsConfigurationError(MarsError):
     """Specific Mars configuration error."""
+
     pass
 
 
 class MarsGetError(MarsError):
     """Generic Mars get error."""
+
     pass
 
 
@@ -35,7 +38,9 @@ def findMarsExtractCommand(sh, inifile=None, command=None):
             actual_command = sh.default_target.get("mars:command", None)
         else:
             actual_config = GenericConfigParser(inifile=actual_inifile)
-            if actual_config.has_section('mars') and actual_config.has_option('mars', 'command'):
+            if actual_config.has_section("mars") and actual_config.has_option(
+                "mars", "command"
+            ):
                 actual_command = actual_config.get("mars", "command")
         if actual_command is None:
             raise MarsConfigurationError("Could not find a proper command.")
@@ -53,4 +58,11 @@ def callMarsExtract(sh, query_file, command=None, fatal=True):
     :return: The return code of the Mars extraction.
     """
     command_line = " ".join([command, query_file])
-    return sh.spawn([command_line, ], shell=True, output=False, fatal=fatal)
+    return sh.spawn(
+        [
+            command_line,
+        ],
+        shell=True,
+        output=False,
+        fatal=fatal,
+    )