meerschaum 2.3.0.dev1__py3-none-any.whl → 2.3.0rc1__py3-none-any.whl

meerschaum/actions/delete.py

@@ -10,9 +10,9 @@ from __future__ import annotations
 from meerschaum.utils.typing import Any, SuccessTuple, Union, Optional, List
 
 def delete(
-        action: Optional[List[str]] = None,
-        **kw: Any
-    ) -> SuccessTuple:
+    action: Optional[List[str]] = None,
+    **kw: Any
+) -> SuccessTuple:
     """
     Delete an element.
 
@@ -35,23 +35,21 @@ def delete(
 
 
 def _complete_delete(
-        action: Optional[List[str]] = None,
-        **kw: Any
-    ) -> List[str]:
+    action: Optional[List[str]] = None,
+    **kw: Any
+) -> List[str]:
     """
     Override the default Meerschaum `complete_` function.
-
     """
-    from meerschaum.actions.start import _complete_start_jobs
     from meerschaum.actions.edit import _complete_edit_config
     if action is None:
         action = []
     options = {
         'connector': _complete_delete_connectors,
         'connectors': _complete_delete_connectors,
-        'config' : _complete_edit_config,
-        'job' : _complete_start_jobs,
-        'jobs' : _complete_start_jobs,
+        'config': _complete_edit_config,
+        'job': _complete_delete_jobs,
+        'jobs': _complete_delete_jobs,
     }
 
     if (
@@ -390,22 +388,22 @@ def _complete_delete_connectors(
 
 
 def _delete_jobs(
-        action: Optional[List[str]] = None,
-        executor_keys: Optional[str] = None,
-        noask: bool = False,
-        nopretty: bool = False,
-        force: bool = False,
-        yes: bool = False,
-        debug: bool = False,
-        **kw
-    ) -> SuccessTuple:
+    action: Optional[List[str]] = None,
+    executor_keys: Optional[str] = None,
+    noask: bool = False,
+    nopretty: bool = False,
+    force: bool = False,
+    yes: bool = False,
+    debug: bool = False,
+    **kw
+) -> SuccessTuple:
     """
     Remove a job's log files and delete the job's ID.
     
     If the job is running, ask to kill the job first.
 
     """
-    from meerschaum.utils.jobs import (
+    from meerschaum.jobs import (
         Job,
         get_running_jobs,
         get_stopped_jobs,
@@ -496,6 +494,51 @@ def _delete_jobs(
     )
 
 
+def _complete_delete_jobs(
+    action: Optional[List[str]] = None,
+    executor_keys: Optional[str] = None,
+    line: str = '',
+    _get_job_method: Optional[str, List[str]] = None,
+    **kw
+) -> List[str]:
+    from meerschaum._internal.shell.Shell import shell_attrs
+    from meerschaum.jobs import (
+        get_jobs,
+        get_filtered_jobs,
+        get_restart_jobs,
+        get_stopped_jobs,
+        get_paused_jobs,
+        get_running_jobs,
+    )
+    from meerschaum.utils.misc import remove_ansi
+
+    executor_keys = executor_keys or remove_ansi(shell_attrs.get('executor_keys', 'local'))
+
+    jobs = get_jobs(executor_keys)
+    if _get_job_method:
+        method_keys = [_get_job_method] if isinstance(_get_job_method, str) else _get_job_method
+        method_jobs = {}
+        for method_key in method_keys:
+            method_func = locals()[f'get_{method_key}_jobs']
+            method_jobs.update(method_func(jobs=jobs))
+        jobs = method_jobs
+
+    if not action:
+        return list(jobs)
+
+    possibilities = []
+    _line_end = line.split(' ')[-1]
+    for name in jobs:
+        if name in action:
+            continue
+        if _line_end == '':
+            possibilities.append(name)
+            continue
+        if name.startswith(action[-1]):
+            possibilities.append(name)
+    return possibilities
+
+
 def _delete_venvs(
         action: Optional[List[str]] = None,
         yes: bool = False,

meerschaum/actions/edit.py

@@ -11,9 +11,9 @@ import meerschaum as mrsm
 from meerschaum.utils.typing import List, Any, SuccessTuple, Optional, Dict
 
 def edit(
-        action: Optional[List[str]] = None,
-        **kw: Any
-    ) -> SuccessTuple:
+    action: Optional[List[str]] = None,
+    **kw: Any
+) -> SuccessTuple:
     """
     Edit an existing element.
     """

meerschaum/actions/install.py

@@ -10,9 +10,9 @@ from __future__ import annotations
 from meerschaum.utils.typing import List, Any, SuccessTuple, Optional, Union
 
 def install(
-        action: Optional[List[str]] = None,
-        **kw: Any
-    ) -> SuccessTuple:
+    action: Optional[List[str]] = None,
+    **kw: Any
+) -> SuccessTuple:
     """
     Install Meerschaum plugins or Python packages.
     """
@@ -26,9 +26,9 @@ def install(
 
 
 def _complete_install(
-        action: Optional[List[str]] = None,
-        **kw: Any
-    ) -> List[str]:
+    action: Optional[List[str]] = None,
+    **kw: Any
+) -> List[str]:
     """
     Override the default Meerschaum `complete_` function.
     """
@@ -52,13 +52,13 @@ def _complete_install(
 
 
 def _install_plugins(
-        action: Optional[List[str]] = None,
-        repository: Optional[str] = None,
-        skip_deps: bool = False,
-        force: bool = False,
-        debug: bool = False,
-        **kw: Any
-    ) -> SuccessTuple:
+    action: Optional[List[str]] = None,
+    repository: Optional[str] = None,
+    skip_deps: bool = False,
+    force: bool = False,
+    debug: bool = False,
+    **kw: Any
+) -> SuccessTuple:
     """
     Install a plugin.
     
@@ -103,10 +103,10 @@ def _install_plugins(
 
 
 def _complete_install_plugins(
-        action: Optional[List[str]] = None,
-        repository: Optional[str] = None,
-        **kw: Any
-    ) -> List[str]:
+    action: Optional[List[str]] = None,
+    repository: Optional[str] = None,
+    **kw: Any
+) -> List[str]:
     """
     Search for plugins to autocomplete command line text.
     NOTE: Disabled for the time being so we don't interrupt the user typing.
@@ -143,12 +143,12 @@ class NoVenv:
     pass
 
 def _install_packages(
-        action: Optional[List[str]] = None,
-        sub_args: Optional[List[str]] = None,
-        venv: Union[str, NoVenv] = NoVenv,
-        debug: bool = False,
-        **kw: Any
-    ) -> SuccessTuple:
+    action: Optional[List[str]] = None,
+    sub_args: Optional[List[str]] = None,
+    venv: Union[str, NoVenv] = NoVenv,
+    debug: bool = False,
+    **kw: Any
+) -> SuccessTuple:
     """
     Install PyPI packages into the Meerschaum virtual environment.
     
@@ -179,19 +179,17 @@ def _install_packages(
     )
 
 
-def _complete_install_packages(
-        **kw : Any
-    ) -> List[str]:
+def _complete_install_packages(**kw : Any) -> List[str]:
     return []
 
 
 def _install_required(
-        action: Optional[List[str]] = None,
-        repository: Optional[str] = None,
-        force: bool = False,
-        debug: bool = False,
-        **kw
-    ) -> SuccessTuple:
+    action: Optional[List[str]] = None,
+    repository: Optional[str] = None,
+    force: bool = False,
+    debug: bool = False,
+    **kw
+) -> SuccessTuple:
     """
     Install the required packages for Meerschaum plugins.
     Each plugin's packages will be installed into its virtual environment.
@@ -237,6 +235,16 @@ def _complete_install_required(*args, **kw) -> List[str]:
     return _complete_uninstall_plugins(*args, **kw)
 
 
+def _install_systemd(
+    action: Optional[List[str]] = None,
+    **kwargs: Any
+) -> SuccessTuple:
+    """
+    Install the Meerschaum job monitor as a systemd service.
+    """
+    import sys
+
+
 ### NOTE: This must be the final statement of the module.
 ###       Any subactions added below these lines will not
 ###       be added to the `help` docstring.

meerschaum/actions/pause.py

@@ -27,13 +27,20 @@ def _complete_pause(
     """
     Override the default Meerschaum `complete_` function.
     """
-    from meerschaum.actions.start import _complete_start_jobs
+    from meerschaum.actions.delete import _complete_delete_jobs
+    from functools import partial
+
     if action is None:
         action = []
 
+    _complete_pause_jobs = partial(
+        _complete_delete_jobs,
+        _get_job_method='running',
+    )
+
     options = {
-        'job' : _complete_start_jobs,
-        'jobs' : _complete_start_jobs,
+        'job': _complete_pause_jobs,
+        'jobs': _complete_pause_jobs,
     }
 
     if (
@@ -49,43 +56,53 @@ def _complete_pause(
 
 
 def _pause_jobs(
-        action: Optional[List[str]] = None,
-        noask: bool = False,
-        force: bool = False,
-        yes: bool = False,
-        nopretty: bool = False,
-        **kw
-    ) -> SuccessTuple:
+    action: Optional[List[str]] = None,
+    executor_keys: Optional[str] = None,
+    noask: bool = False,
+    force: bool = False,
+    yes: bool = False,
+    nopretty: bool = False,
+    debug: bool = False,
+    **kw
+) -> SuccessTuple:
     """
-    Pause running jobs that were started with `-d` or `start job`.
-    
-    To see running jobs, run `show jobs`.
+    Pause (suspend) running jobs.
     """
     from meerschaum.utils.formatting._jobs import pprint_jobs
     from meerschaum.utils.daemon import (
         get_filtered_daemons, get_running_daemons, get_stopped_daemons, get_paused_daemons,
     )
+    from meerschaum.jobs import (
+        get_filtered_jobs,
+        get_running_jobs,
+        get_stopped_jobs,
+        get_paused_jobs,
+    )
     from meerschaum.utils.warnings import warn
     from meerschaum.utils.prompt import yes_no
     from meerschaum.utils.misc import items_str
-    daemons = get_filtered_daemons(action, warn=(not nopretty))
-    _running_daemons = get_running_daemons(daemons)
-    _paused_daemons = get_paused_daemons(daemons)
-    _stopped_daemons = get_stopped_daemons(daemons)
-    if action and _stopped_daemons and not nopretty:
+
+    jobs = get_filtered_jobs(executor_keys, action, debug=debug, warn=(not nopretty))
+    running_jobs = get_running_jobs(executor_keys, jobs, debug=debug)
+    paused_jobs = get_paused_jobs(executor_keys, jobs, debug=debug)
+    stopped_jobs = get_stopped_jobs(executor_keys, jobs, debug=debug)
+
+    if action and stopped_jobs and not nopretty:
         warn(
-            f"Skipping stopped job" + ("s" if len(_stopped_daemons) > 1 else '') + " '" +
-                ("', '".join(d.daemon_id for d in _stopped_daemons)) + "'.",
-            stack = False
+            f"Skipping stopped job" + ("s" if len(stopped_jobs) > 1 else '')
+            + " '"
+            + ("', '".join(stopped_jobs.keys()))
+            + "'.",
+            stack=False,
         )
 
-    daemons_to_pause = _running_daemons + _paused_daemons
-    if not daemons_to_pause:
+    jobs_to_pause = {**running_jobs, **paused_jobs}
+    if not jobs_to_pause:
         return False, "No running jobs to pause. You can start jobs with `-d` or `start jobs`."
 
     if not action:
         if not force:
-            pprint_jobs(_running_daemons)
+            pprint_jobs(running_jobs)
             if not yes_no(
                 "Pause all running jobs?",
                 noask=noask, yes=yes, default='n'
@@ -93,9 +110,9 @@ def _pause_jobs(
                 return False, "No jobs were paused."
 
     successes, fails = [], []
-    for d in daemons_to_pause:
-        pause_success, pause_msg = d.pause()
-        (successes if pause_success else fails).append(d.daemon_id)
+    for name, job in jobs_to_pause.items():
+        pause_success, pause_msg = job.pause()
+        (successes if pause_success else fails).append(name)
 
     msg = (
         (

meerschaum/actions/restart.py

@@ -0,0 +1,107 @@
+#! /usr/bin/env python3
+# vim:fenc=utf-8
+
+"""
+Restart stopped jobs which have not been manually stopped.
+"""
+
+from meerschaum.utils.typing import SuccessTuple, Optional, List, Any
+
+def restart(
+    action: Optional[List[str]] = None,
+    executor_keys: Optional[str] = None,
+    **kwargs: Any
+) -> SuccessTuple:
+    """
+    Restart stopped jobs which have not been manually stopped.
+    """
+    from meerschaum.actions import choose_subaction
+    attach_options = {
+        'jobs': _restart_jobs,
+    }
+    return choose_subaction(action, attach_options, executor_keys=executor_keys, **kwargs)
+
+
+def _complete_restart(
+    action: Optional[List[str]] = None,
+    **kw: Any
+) -> List[str]:
+    """
+    Override the default Meerschaum `complete_` function.
+    """
+    from functools import partial
+    from meerschaum.actions.delete import _complete_delete_jobs
+
+    if action is None:
+        action = []
+
+    _complete_restart_jobs = partial(
+        _complete_delete_jobs,
+        _get_job_method='restart',
+    )
+
+    options = {
+        'job': _complete_restart_jobs,
+        'jobs': _complete_restart_jobs,
+    }
+
+    if (
+        len(action) > 0 and action[0] in options
+            and kw.get('line', '').split(' ')[-1] != action[0]
+    ):
+        sub = action[0]
+        del action[0]
+        return options[sub](action=action, **kw)
+
+    from meerschaum._internal.shell import default_action_completer
+    return default_action_completer(action=(['restart'] + action), **kw)
+
+
+def _restart_jobs(
+    action: Optional[List[str]] = None,
+    executor_keys: Optional[str] = None,
+    loop: bool = False,
+    min_seconds: Optional[float] = 1.0,
+    debug: bool = False,
+    **kwargs: Any
+) -> SuccessTuple:
+    """
+    Restart stopped jobs which have not been manually stopped.
+    """
+    import time
+    from meerschaum.jobs import (
+        get_restart_jobs,
+        get_filtered_jobs,
+        check_restart_jobs,
+    )
+    from meerschaum.utils.misc import items_str
+    from meerschaum.utils.warnings import info
+    action = action or []
+
+    while True:
+        jobs = get_filtered_jobs(executor_keys, action, include_hidden=True, debug=debug)
+        restart_jobs = get_restart_jobs(executor_keys, jobs, debug=debug) if not action else jobs
+        if not restart_jobs and not loop:
+            return True, "No jobs need to be restarted."
+
+        info(
+            "Checking job"
+            + ('s' if len(restart_jobs) != 1 else '')
+            + ' '
+            + items_str(list(restart_jobs.keys()))
+            + '...'
+        )
+
+        check_success, check_msg = check_restart_jobs(
+            executor_keys,
+            restart_jobs,
+            debug=debug,
+        )
+        if not loop:
+            break
+
+        if min_seconds is not None and min_seconds != 0.0:
+            info(f"Sleeping for {min_seconds} seconds...")
+            time.sleep(min_seconds)
+
+    return check_success, check_msg

meerschaum/actions/show.py

@@ -54,7 +54,7 @@ def _complete_show(
     """
     Override the default Meerschaum `complete_` function.
     """
-    from meerschaum.actions.start import _complete_start_jobs
+    from meerschaum.actions.delete import _complete_delete_jobs
 
     if action is None:
         action = []
@@ -65,10 +65,10 @@ def _complete_show(
         'config'    : _complete_show_config,
         'package'   : _complete_show_packages,
         'packages'  : _complete_show_packages,
-        'job'       : _complete_start_jobs,
-        'jobs'      : _complete_start_jobs,
-        'log'       : _complete_start_jobs,
-        'logs'      : _complete_start_jobs,
+        'job'       : _complete_delete_jobs,
+        'jobs'      : _complete_delete_jobs,
+        'log'       : _complete_delete_jobs,
+        'logs'      : _complete_delete_jobs,
     }
 
     if (
@@ -560,7 +560,7 @@ def _show_jobs(
     """
     Show the currently running and stopped jobs.
     """
-    from meerschaum.utils.jobs import get_filtered_jobs
+    from meerschaum.jobs import get_filtered_jobs
     from meerschaum.utils.formatting._jobs import pprint_jobs
 
     jobs = get_filtered_jobs(executor_keys, action, debug=debug)
@@ -575,7 +575,7 @@ def _show_jobs(
                 "      - start api -d\n" +
                 "      - start job sync pipes --loop"
             )
-        return False, "No jobs to show."
+        return True, "No jobs to show."
 
     pprint_jobs(jobs, nopretty=nopretty)
     return True, "Success"
@@ -602,7 +602,7 @@ def _show_logs(
     from functools import partial
     from datetime import datetime, timezone
     from meerschaum.utils.packages import attempt_import, import_rich
-    from meerschaum.utils.jobs import get_filtered_jobs, Job
+    from meerschaum.jobs import get_filtered_jobs, Job
     from meerschaum.utils.warnings import warn, info
     from meerschaum.utils.formatting import get_console, ANSI, UNICODE
     from meerschaum.utils.misc import tail

meerschaum/actions/start.py

@@ -10,9 +10,9 @@ from __future__ import annotations
 from meerschaum.utils.typing import SuccessTuple, Optional, List, Any
 
 def start(
-        action: Optional[List[str]] = None,
-        **kw: Any,
-    ) -> SuccessTuple:
+    action: Optional[List[str]] = None,
+    **kw: Any,
+) -> SuccessTuple:
     """
     Start subsystems (API server, background job, etc.).
     """
@@ -28,16 +28,23 @@ def start(
 
 
 def _complete_start(
-        action: Optional[List[str]] = None,
-        **kw: Any
-    ) -> List[str]:
+    action: Optional[List[str]] = None,
+    **kw: Any
+) -> List[str]:
     """
     Override the default Meerschaum `complete_` function.
     """
+    from meerschaum.actions.delete import _complete_delete_jobs
+    from functools import partial
 
     if action is None:
         action = []
 
+    _complete_start_jobs = partial(
+        _complete_delete_jobs,
+        _get_job_method=['stopped', 'paused'],
+    )
+
     options = {
         'job': _complete_start_jobs,
         'jobs': _complete_start_jobs,
@@ -76,13 +83,13 @@ def _start_api(action: Optional[List[str]] = None, **kw):
     return actions['api'](action=['start'], **kw)
 
 def _start_jobs(
-        action: Optional[List[str]] = None,
-        name: Optional[str] = None,
-        sysargs: Optional[List[str]] = None,
-        executor_keys: Optional[str] = None,
-        debug: bool = False,
-        **kw
-    ) -> SuccessTuple:
+    action: Optional[List[str]] = None,
+    name: Optional[str] = None,
+    sysargs: Optional[List[str]] = None,
+    executor_keys: Optional[str] = None,
+    debug: bool = False,
+    **kw
+) -> SuccessTuple:
     """
     Run a Meerschaum action as a background job.
     
@@ -115,7 +122,7 @@ def _start_jobs(
     import textwrap
     from meerschaum.utils.warnings import warn, info
     from meerschaum.utils.daemon._names import get_new_daemon_name
-    from meerschaum.utils.jobs import (
+    from meerschaum.jobs import (
         Job,
         get_jobs,
         get_filtered_jobs,
@@ -123,6 +130,7 @@ def _start_jobs(
         get_running_jobs,
         get_paused_jobs,
         get_restart_jobs,
+        _install_healthcheck_job,
     )
     from meerschaum._internal.arguments._parse_arguments import parse_arguments
     from meerschaum.actions import actions
@@ -133,7 +141,7 @@ def _start_jobs(
     from meerschaum.utils.misc import items_str
 
     names = []
-    jobs = get_filtered_jobs(executor_keys, debug=debug)
+    jobs = get_filtered_jobs(executor_keys, action, debug=debug)
 
     new_job = len(list(action)) > 0
     _potential_jobs = {'known': [], 'unknown': []}
@@ -166,7 +174,7 @@ def _start_jobs(
                         + items_str(_potential_jobs['unknown'])
                         + " will be ignored."
                     ),
-                    stack = False
+                    stack=False
                 )
 
         ### Determine the `names` list.
@@ -232,7 +240,7 @@ def _start_jobs(
                 stop_success_tuple = actions['stop'](
                     action=['jobs'] + [_name for _name in _filtered_running_jobs],
                     force=True,
-                    mrsm_instance=mrsm_instance,
+                    executor_keys=executor_keys,
                     debug=debug,
                 )
                 if not stop_success_tuple[0]:
@@ -294,33 +302,10 @@ def _start_jobs(
         + ("Failed to start job" + ("s" if len(_failures) != 1 else '')
             + f" {items_str(_failures)}." if _failures else '')
     )
+    _install_healthcheck_job()
     return len(_failures) == 0, msg
 
 
-def _complete_start_jobs(
-    action: Optional[List[str]] = None,
-    executor_keys: Optional[str] = None,
-    line: str = '',
-    **kw
-) -> List[str]:
-    from meerschaum.utils.jobs import get_filtered_jobs
-    jobs = get_filtered_jobs(executor_keys, action)
-    if not action:
-        return list(jobs)
-
-    possibilities = []
-    _line_end = line.split(' ')[-1]
-    for name in jobs:
-        if name in action:
-            continue
-        if _line_end == '':
-            possibilities.append(name)
-            continue
-        if name.startswith(action[-1]):
-            possibilities.append(name)
-    return possibilities
-
-
 def _start_gui(
         action: Optional[List[str]] = None,
         mrsm_instance: Optional[str] = None,