From 8b22a2a7dd2507e164f0780fa38d73ba36912144 Mon Sep 17 00:00:00 2001
From: Cristian <cristian@swapps.com>
Date: Tue, 7 Jul 2020 09:10:36 -0500
Subject: [PATCH] feat: Enable --depth flag (still does nothing)
---
archivebox/cli/archivebox_add.py | 13 +++++++------
tests/test_args.py | 7 +++++++
2 files changed, 14 insertions(+), 6 deletions(-)
diff --git a/archivebox/cli/archivebox_add.py b/archivebox/cli/archivebox_add.py
index 272fe5cf..77a11bd0 100644
--- a/archivebox/cli/archivebox_add.py
+++ b/archivebox/cli/archivebox_add.py
@@ -45,6 +45,13 @@ def main(args: Optional[List[str]]=None, stdin: Optional[IO]=None, pwd: Optional
' ~/Desktop/sites_list.csv\n'
)
)
+ parser.add_argument(
+ "--depth",
+ action="store",
+ default=0,
+ type=int,
+ help="Recursively archive all linked pages up to this many hops away"
+ )
command = parser.parse_args(args or ())
import_str = accept_stdin(stdin)
add(
@@ -63,12 +70,6 @@ if __name__ == '__main__':
# TODO: Implement these
#
# parser.add_argument(
-# '--depth', #'-d',
-# type=int,
-# help='Recursively archive all linked pages up to this many hops away',
-# default=0,
-# )
-# parser.add_argument(
# '--mirror', #'-m',
# action='store_true',
# help='Archive an entire site (finding all linked pages below it on the same domain)',
diff --git a/tests/test_args.py b/tests/test_args.py
index e69de29b..b8df1941 100644
--- a/tests/test_args.py
+++ b/tests/test_args.py
@@ -0,0 +1,7 @@
+import subprocess
+
+from .fixtures import *
+
+def test_depth_flag_is_accepted(tmp_path, process):
+ arg_process = subprocess.run(["archivebox", "add", "https://example.com", "--depth=0"], capture_output=True)
+ assert 'unrecognized arguments: --depth' not in arg_process.stderr.decode('utf-8')
\ No newline at end of file