Commands return correct exit codes. (#26)

Author: tobiasraabe

.pre-commit-config.yaml

@@ -8,7 +8,7 @@ repos:
     -   id: check-yaml
         exclude: meta.yaml
     -   id: debug-statements
-        exclude: (debugging\.py|build\.py)
+        exclude: (debugging\.py|build\.py|clean\.py|mark/__init__\.py)
     -   id: end-of-file-fixer
 -   repo: https://github.com/asottile/pyupgrade
     rev: v2.7.2

docs/changes.rst

@@ -10,6 +10,7 @@ all releases are available on `Anaconda.org <https://anaconda.org/pytask/pytask>
 ------------------
 
 - :gh:`25` allows to customize the names of the task files.
+- :gh:`26` makes commands return the correct exit codes.
 - :gh:`27` implements the ``pytask_collect_task_teardown`` hook specification to perform
   checks after a task is collected.
 

src/_pytask/build.py

@@ -5,9 +5,9 @@
 
 import click
 from _pytask.config import hookimpl
-from _pytask.database import create_database
 from _pytask.enums import ExitCode
 from _pytask.exceptions import CollectionError
+from _pytask.exceptions import ConfigurationError
 from _pytask.exceptions import ExecutionError
 from _pytask.exceptions import ResolvingDependenciesError
 from _pytask.pluginmanager import get_plugin_manager
@@ -49,16 +49,16 @@ def main(config_from_cli):
 
         config = pm.hook.pytask_configure(pm=pm, config_from_cli=config_from_cli)
 
-        create_database(**config["database"])
-
         session = Session.from_config(config)
-        session.exit_code = ExitCode.OK
 
-    except Exception:
+    except (ConfigurationError, Exception):
         traceback.print_exception(*sys.exc_info())
         session = Session({}, None)
         session.exit_code = ExitCode.CONFIGURATION_FAILED
 
+        if config_from_cli.get("pdb"):
+            pdb.post_mortem()
+
     else:
         try:
             session.hook.pytask_log_session_header(session=session)

src/_pytask/clean.py

@@ -1,5 +1,6 @@
 """Add a command to clean the project from files unknown to pytask."""
 import itertools
+import pdb
 import shutil
 import sys
 import traceback
@@ -71,15 +72,16 @@ def clean(**config_from_cli):
         pm.hook.pytask_add_hooks(pm=pm)
 
         config = pm.hook.pytask_configure(pm=pm, config_from_cli=config_from_cli)
-
         session = Session.from_config(config)
-        session.exit_code = ExitCode.OK
 
     except Exception:
         traceback.print_exception(*sys.exc_info())
         session = Session({}, None)
         session.exit_code = ExitCode.CONFIGURATION_FAILED
 
+        if config_from_cli.get("pdb"):
+            pdb.post_mortem()
+
     else:
         try:
             session.hook.pytask_log_session_header(session=session)
@@ -119,7 +121,7 @@ def clean(**config_from_cli):
             traceback.print_exception(*sys.exc_info())
             session.exit_code = ExitCode.FAILED
 
-    return session
+    sys.exit(session.exit_code)
 
 
 def _collect_all_paths_known_to_pytask(session):

src/_pytask/config.py

@@ -9,6 +9,7 @@
 
 import click
 import pluggy
+from _pytask.shared import convert_truthy_or_falsy_to_bool
 from _pytask.shared import get_first_non_none_value
 from _pytask.shared import parse_paths
 from _pytask.shared import parse_value_or_multiline_option
@@ -112,7 +113,7 @@ def pytask_parse_config(config, config_from_cli, config_from_file):
         config_from_file,
         key="debug_pytask",
         default=False,
-        callback=bool,
+        callback=convert_truthy_or_falsy_to_bool,
     )
     if config["debug_pytask"]:
         config["pm"].trace.root.setwriter(click.echo)

src/_pytask/database.py

@@ -120,3 +120,8 @@ def pytask_parse_config(config, config_from_cli, config_from_file):
         "create_db": config["database_create_db"],
         "create_tables": config["database_create_tables"],
     }
+
+
+@hookimpl
+def pytask_post_parse(config):
+    create_database(**config["database"])

src/_pytask/enums.py

@@ -1,3 +1,4 @@
+"""Enumerations for pytask."""
 import enum
 
 

src/_pytask/exceptions.py

@@ -10,6 +10,10 @@ class NodeNotCollectedError(PytaskError):
     """Exception for nodes which could not be collected."""
 
 
+class ConfigurationError(PytaskError):
+    """Exception during the configuration."""
+
+
 class CollectionError(PytaskError):
     """Exception during collection."""
 

src/_pytask/mark/__init__.py

@@ -1,3 +1,4 @@
+import pdb
 import sys
 import textwrap
 import traceback
@@ -50,9 +51,10 @@ def markers(**config_from_cli):
         pm.hook.pytask_add_hooks(pm=pm)
 
         config = pm.hook.pytask_configure(pm=pm, config_from_cli=config_from_cli)
-
         session = Session.from_config(config)
-        session.exit_code = ExitCode.OK
+
+        if config_from_cli.get("pdb"):
+            pdb.post_mortem()
 
     except Exception:
         traceback.print_exception(*sys.exc_info())
@@ -68,7 +70,7 @@ def markers(**config_from_cli):
             )
             click.echo("")
 
-    return session
+    sys.exit(session.exit_code)
 
 
 @hookimpl

src/_pytask/session.py

@@ -1,4 +1,5 @@
 import attr
+from _pytask.enums import ExitCode
 
 
 @attr.s
@@ -27,6 +28,7 @@ class Session:
     """
     execution_reports = attr.ib(factory=list)
     """Optional[List[pytask.report.ExecutionReport]]: Reports for executed tasks."""
+    exit_code = attr.ib(default=ExitCode.OK)
 
     @classmethod
     def from_config(cls, config):