feat(api): support file format for repository archive

dAnjou · dAnjou · commit 3c38eb693994 · 2021-08-01T12:44:20.000+02:00
diff --git a/docs/gl_objects/projects.rst b/docs/gl_objects/projects.rst
@@ -170,6 +170,14 @@ Get the repository archive::
     # get the archive for a branch/tag/commit
     tgz = project.repository_archive(sha='4567abc')
 
+    # get the archive in a different format
+    zip = project.repository_archive(format='zip')
+
+.. note::
+
+   For the formats available, refer to
+   https://docs.gitlab.com/ce/api/repositories.html#get-file-archive
+
 .. warning::
 
    Archives are entirely stored in memory unless you use the streaming feature.
diff --git a/gitlab/v4/objects/repositories.py b/gitlab/v4/objects/repositories.py
@@ -158,12 +158,18 @@ def repository_contributors(self, **kwargs):
         path = "/projects/%s/repository/contributors" % self.get_id()
         return self.manager.gitlab.http_list(path, **kwargs)
 
-    @cli.register_custom_action("Project", tuple(), ("sha",))
+    @cli.register_custom_action("Project", tuple(), ("sha", "format"))
     @exc.on_http_error(exc.GitlabListError)
     def repository_archive(
-        self, sha=None, streamed=False, action=None, chunk_size=1024, **kwargs
+        self,
+        sha=None,
+        streamed=False,
+        action=None,
+        chunk_size=1024,
+        format=None,
+        **kwargs
     ):
-        """Return a tarball of the repository.
+        """Return an archive of the repository.
 
         Args:
             sha (str): ID of the commit (default branch by default)
@@ -173,6 +179,7 @@ def repository_archive(
             action (callable): Callable responsible of dealing with chunk of
                 data
             chunk_size (int): Size of each chunk
+            format (str): file format (tar.gz by default)
             **kwargs: Extra options to send to the server (e.g. sudo)
 
         Raises:
@@ -183,6 +190,8 @@ def repository_archive(
             str: The binary data of the archive
         """
         path = "/projects/%s/repository/archive" % self.get_id()
+        if format:
+            path += "." + format
         query_data = {}
         if sha:
             query_data["sha"] = sha
