Add recursive option to Project.recursive_tree()

David Garfinkle · David Garfinkle · commit f66540af366b · 2018-03-07T18:10:50.000-05:00
diff --git a/gitlab/v4/objects.py b/gitlab/v4/objects.py
@@ -2403,7 +2403,7 @@ class Project(SaveMixin, ObjectDeleteMixin, RESTObject):
 
     @cli.register_custom_action('Project', tuple(), ('path', 'ref'))
     @exc.on_http_error(exc.GitlabGetError)
-    def repository_tree(self, path='', ref='', **kwargs):
+    def repository_tree(self, path='', ref='', recursive=False, **kwargs):
         """Return a list of files in the repository.
 
         Args:
@@ -2429,6 +2429,8 @@ def repository_tree(self, path='', ref='', **kwargs):
             query_data['path'] = path
         if ref:
             query_data['ref'] = ref
+        if recursive:
+            query_data['recursive'] = recursive
         return self.manager.gitlab.http_list(gl_path, query_data=query_data,
                                              **kwargs)
 
diff --git a/tools/python_test_v4.py b/tools/python_test_v4.py
@@ -385,6 +385,8 @@
 archive1 = admin_project.repository_archive()
 archive2 = admin_project.repository_archive('master')
 assert(archive1 == archive2)
+recursive_tree = admin_project.repository_tree(recursive=True)
+assert(len(recursive_tree) != 0)
 
 # project file uploads
 filename = "test.txt"
