NPO-Accounting/conservancy_beancount Changeset - 902c313b4dfb

Changeset - 902c313b4dfb

Parent rev.

Child rev.

[Not reviewed]

0 2 0

Brett Smith - 4 years ago 2021-01-09 15:09:08
brettcsmith@brettcsmith.org

cliutil: New function can_run.

2 files changed with 30 insertions and 0 deletions:

conservancy_beancount/cliutil.py

tests/test_cliutil.py

0 comments (0 inline, 0 general)

conservancy_beancount/cliutil.py

➞

Show inline comments

@@ ... / @@ -19,8 +19,9 @@ import operator @@
 import os
 import pkg_resources
 import re
 import signal
 import subprocess
 import sys
 import traceback
 import types
@@ ... / @@ -36,8 +37,9 @@ from typing import ( @@
     cast,
     Any,
     BinaryIO,
     Callable,
     Container,
     IO,
     Iterable,
     NamedTuple,
     NoReturn,
@@ ... @@
         const=f"{progname} version {VERSION}\n{LICENSE}",
         help="Show program version and license information",
+    )
 def can_run(
         cmd: Sequence[str],
         stdout: Optional[int]=subprocess.DEVNULL,
         stderr: Optional[int]=None,
         ok_returncodes: Container[int]=frozenset([0]),
 ) -> bool:
     try:
         with subprocess.Popen(cmd, stdin=subprocess.PIPE, stdout=stdout, stderr=stderr) as proc:
             # Typing says this can be None, but I don't think that's true
             # given that we passed stdin=PIPE.
             proc.stdin.close()  # type:ignore[union-attr]
     except (OSError, subprocess.SubprocessError):
         return False
     else:
         return proc.returncode in ok_returncodes
 def date_arg(arg: str) -> datetime.date:
     return datetime.datetime.strptime(arg, '%Y-%m-%d').date()
 def diff_year(date: datetime.date, diff: int) -> datetime.date:

tests/test_cliutil.py

➞

Show inline comments

@@ ... / @@ -19,8 +19,10 @@ import traceback @@
 import pytest
 from pathlib import Path
 from . import testutil
 from conservancy_beancount import cliutil
 FILE_NAMES = ['-foobar', '-foo.bin']
 STREAM_PATHS = [None, Path('-')]
@@ ... / @@ -226,4 +228,14 @@ def test_version_argument(argparser, capsys, arg): @@
     (datetime.date(2010, 3, 1), -2, datetime.date(2008, 3, 1)),
 ])
 def test_diff_year(date, diff, expected):
     assert cliutil.diff_year(date, diff) == expected
 @pytest.mark.parametrize('cmd,expected', [
     (['true'], True),
     (['true', '--version'], True),
     (['false'], False),
     (['false', '--version'], False),
     ([str(testutil.TESTS_DIR)], False),
 ])
 def test_can_run(cmd, expected):
     assert cliutil.can_run(cmd) == expected

0 comments (0 inline, 0 general)