oh-my-batch 0.1.0.dev1__py3-none-any.whl → 0.1.0.dev3__py3-none-any.whl

oh_my_batch/cli.py

@@ -1,5 +1,8 @@
+import logging
 import fire
 
+logging.basicConfig(format='%(asctime)s %(name)s: %(message)s', level=logging.INFO)
+
 class JobCli:
 
     def slurm(self):

oh_my_batch/job.py

@@ -12,29 +12,22 @@ from .util import expand_globs, shell_run, parse_csv
 
 logger = logging.getLogger(__name__)
 
+class JobState:
+    NULL = 0
+    PENDING = 1
+    RUNNING = 2
+    CANCELLED = 3
+    COMPLETED = 4
+    FAILED = 5
+    UNKNOWN = 6
 
-class JobState(bytes, Enum):
-    """
-    Job state enumeration
-    """
-    def __new__(cls, value: int, terminal: bool, status_name: str) -> "JobState":
-        obj = bytes.__new__(cls, [value])
-        obj._value_ = value
-        obj.terminal = terminal
-        obj.status_name = status_name
-        return obj
-
-    value: int  # type: ignore
-    terminal: bool
-    status_name: str
-
-    NULL = (0, True, "NULL")
-    PENDING = (1, False, "PENDING")
-    RUNNING = (2, False, "RUNNING")
-    CANCELLED = (3, True, "CANCELLED")
-    COMPLETED = (4, True, "COMPLETED")
-    FAILED = (5, True, "FAILED")
-    UNKNOWN = (6, False, "UNKNOWN")
+    @classmethod
+    def is_terminal(cls, state: int):
+        return state in (JobState.NULL, JobState.COMPLETED, JobState.FAILED, JobState.CANCELLED)
+
+    @classmethod
+    def is_success(cls, state: int):
+        return state == JobState.COMPLETED
 
 
 def new_job(script: str):
@@ -84,7 +77,7 @@ class BaseJobManager:
                 break
 
             # stop if all jobs are terminal and not job to be submitted
-            if (all(j['state'].terminal for j in jobs) and
+            if (all(JobState.is_terminal(j['state']) for j in jobs) and
                     not any(should_submit(j, max_tries) for j in jobs)):
                 break
 
@@ -129,8 +122,8 @@ class Slurm(BaseJobManager):
                         logger.warning('Unknown job %s state: %s',row['JobID'], row['State'])
                     break
             else:
-                job['state'] = JobState.FAILED
-                logger.error('Job %s not found in sacct output', job['id'])
+                if job['id']:
+                    logger.error('Job %s not found in sacct output', job['id'])
 
         # check if there are jobs to be (re)submitted
         for job in jobs:
@@ -170,8 +163,8 @@ class Slurm(BaseJobManager):
 
 
 def should_submit(job: dict, max_tries: int):
-    state: JobState = job['state']
-    if not state.terminal:
+    state: int = job['state']
+    if not JobState.is_terminal(state):
         return False
     if job['tries'] >= max_tries:
         return False

{oh_my_batch-0.1.0.dev1.dist-info → oh_my_batch-0.1.0.dev3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: oh-my-batch
-Version: 0.1.0.dev1
+Version: 0.1.0.dev3
 Summary: 
 License: GPL
 Author: weihong.xu
@@ -17,9 +17,9 @@ Requires-Dist: fire (>=0.7.0,<0.8.0)
 Description-Content-Type: text/markdown
 
 # oh-my-batch
-A simple tool to manipulate batch tasks.
+A simple tool to manipulate batch tasks designed for scientific computing community.
 
-The goal of this to is to provide a handy command tool for
+## Features
 * `omb combo`: generate folders/files from different combinations of parameters
 * `omb batch`: generate batch scripts from multiple working directories
 * `omb job`: track the state of job in job schedular

{oh_my_batch-0.1.0.dev1.dist-info → oh_my_batch-0.1.0.dev3.dist-info}/RECORD

@@ -3,12 +3,12 @@ oh_my_batch/__main__.py,sha256=sWyFZMwWNvhkanwZSJRGfBBDoIevhC028dTSB67i6yI,61
 oh_my_batch/assets/__init__.py,sha256=Exub46UbQaz2V2eXpQeiVfnThQpXaNeuyjlGY6gBSZc,130
 oh_my_batch/assets/functions.sh,sha256=eORxFefV-XrWbG-2I6u-c8uf1XxOQ31LaeVHBumwzJ4,708
 oh_my_batch/batch.py,sha256=e73N-xwxMvgxnWwFMp33PQD1Dy-T-ATjANlwtPRHPQM,3016
-oh_my_batch/cli.py,sha256=G_JxqX0Zbx_EbcDxXbYjJ_4O-EOhmkF1lcMWgQ5ZPqo,375
+oh_my_batch/cli.py,sha256=uelW9ms1N30DipJOcsiuG5K-5VN8O6yu1RNEqex00GY,475
 oh_my_batch/combo.py,sha256=AHFD5CLoczqtjcfl2Rb4A2ucoQU40-cWtDOYjtP-yY4,7680
-oh_my_batch/job.py,sha256=z4iOydxzDNG_L3rgDa4qUVsRAH4L-GOk_RSxCx4GLNA,6035
+oh_my_batch/job.py,sha256=_fETBYpuSd_hNHKnXSwYcSU3OXtU7PO-P2QMfhE-Wfs,5788
 oh_my_batch/util.py,sha256=H8B4zVNH5xRp-NG_uypgvtmz2YSpXy_6LK5ROv6SYrc,2116
-oh_my_batch-0.1.0.dev1.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-oh_my_batch-0.1.0.dev1.dist-info/METADATA,sha256=yTzn4_dXZIJG2iZhJwFGxatYupAPUWUlyN-ciQZfWQw,4456
-oh_my_batch-0.1.0.dev1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-oh_my_batch-0.1.0.dev1.dist-info/entry_points.txt,sha256=ZY2GutSoNjjSyJ4qO2pTeseKUFgoTYdvmgkuZZkwi68,77
-oh_my_batch-0.1.0.dev1.dist-info/RECORD,,
+oh_my_batch-0.1.0.dev3.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+oh_my_batch-0.1.0.dev3.dist-info/METADATA,sha256=sFfJrqAmymhuKZIv4-ydeDX6Y3oby3NWVfiPjQPUAUM,4453
+oh_my_batch-0.1.0.dev3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+oh_my_batch-0.1.0.dev3.dist-info/entry_points.txt,sha256=ZY2GutSoNjjSyJ4qO2pTeseKUFgoTYdvmgkuZZkwi68,77
+oh_my_batch-0.1.0.dev3.dist-info/RECORD,,