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