feat(api): support file format for repository archive

dAnjou · web-flow · commit ba4e0da9ae8e · 2021-07-31T22:26:26.000+02:00
diff --git a/gitlab/v4/objects/repositories.py b/gitlab/v4/objects/repositories.py
@@ -158,12 +158,12 @@ 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 +173,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 +184,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
