NPO-Accounting/conservancy_beancount Changeset - 3fbd05d55303

Changeset - 3fbd05d55303

Parent rev.

Child rev.

[Not reviewed]

0 2 0

Brett Smith - 4 years ago 2020-05-30 02:05:26
brettcsmith@brettcsmith.org

cliutil: Add is_main_script function.

2 files changed with 9 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

@@ ... / @@ -15,12 +15,13 @@ GNU Affero General Public License for more details. @@
 You should have received a copy of the GNU Affero General Public License
 along with this program.  If not, see <https://www.gnu.org/licenses/>."""
 import argparse
 import enum
 import inspect
 import logging
 import operator
 import os
 import pkg_resources
 import signal
 import sys
@@ ... @@
         action=InfoAction,
         nargs=0,
         const=f"{progname} version {VERSION}\n{LICENSE}",
         help="Show program version and license information",
+    )
 def is_main_script() -> bool:
     """Return true if the caller is the "main" program."""
     stack = inspect.stack(context=False)
     return len(stack) <= 3 and stack[-1].function.startswith('<')
 def setup_logger(logger: Union[str, logging.Logger]='',
                  loglevel: int=logging.INFO,
                  stream: TextIO=sys.stderr,
                  fmt: str='%(name)s: %(levelname)s: %(message)s',
 ) -> logging.Logger:
     if isinstance(logger, str):

tests/test_cliutil.py

➞

Show inline comments

@@ ... / @@ -88,12 +88,15 @@ def test_excepthook_traceback(caplog): @@
     caplog.set_level(logging.DEBUG)
     with pytest.raises(SystemExit) as exc_check:
         cliutil.ExceptHook()(*args)
     assert caplog.records
     assert caplog.records[-1].message == ''.join(traceback.format_exception(*args))
 def test_is_main_script():
     assert not cliutil.is_main_script()
 @pytest.mark.parametrize('arg,expected', [
     ('debug', logging.DEBUG),
     ('info', logging.INFO),
     ('warning', logging.WARNING),
     ('warn', logging.WARNING),
     ('error', logging.ERROR),

0 comments (0 inline, 0 general)