diff --git a/src/nancy/cli/__init__.py b/src/nancy/cli/__init__.py index 3270a06..26e008b 100644 --- a/src/nancy/cli/__init__.py +++ b/src/nancy/cli/__init__.py @@ -3,7 +3,7 @@ import click from ..version import __version__ #from .freeze import freeze, thaw -from .record import record +from . import record # from https://click.palletsprojects.com/en/5.x/advanced/ @@ -36,5 +36,5 @@ def main(): pass #main.add_command(freeze) #main.add_command(thaw) -main.add_command(record) +main.add_command(record.record_cli, name='record') main.add_command(version) diff --git a/src/nancy/cli/record.py b/src/nancy/cli/record.py index 30237b7..570ace4 100644 --- a/src/nancy/cli/record.py +++ b/src/nancy/cli/record.py @@ -5,7 +5,7 @@ from .. import store import os -def do_record(directory): +def record(directory): """Unwrapped record command""" if not os.path.isdir(directory): raise ValueError(f"Cannot record non-existent directory {directory}") @@ -21,7 +21,7 @@ def do_record(directory): @click.command() @click.argument("directory", default='.') -def record(directory): +def record_cli(directory): """ Initialize tracking or record changes to a tracked directory. @@ -30,5 +30,5 @@ def record(directory): directory is part of an existing store, it will be updated and versions of any files changes since the last recording will be incremented. """ - do_record(directory) + record(directory) diff --git a/tests/test_store.py b/tests/test_store.py index 80dbd76..6a0fb48 100644 --- a/tests/test_store.py +++ b/tests/test_store.py @@ -21,8 +21,8 @@ def filled_dir(bare_dir): def test_record_untracked_dir(filled_dir): - from nancy.cli.record import do_record - do_record(filled_dir) + from nancy.cli.record import record + record(filled_dir) @pytest.fixture