add onefuzz containers files download_dir (#598)

fixes #571
This commit is contained in:
bmc-msft
2021-02-26 10:27:51 -05:00
committed by GitHub
parent 4cd2de0e93
commit bc6c8408c4
5 changed files with 43 additions and 13 deletions

View File

@ -215,7 +215,9 @@ class Files(Endpoint):
client = self._get_client(container)
client.upload_file(file_path, blob_name)
def upload_dir(self, container: primitives.Container, dir_path: str) -> None:
def upload_dir(
self, container: primitives.Container, dir_path: primitives.Directory
) -> None:
""" uploads a directory to a container """
self.logger.debug("uploading directory to container %s:%s", container, dir_path)
@ -223,6 +225,18 @@ class Files(Endpoint):
client = self._get_client(container)
client.upload_dir(dir_path)
def download_dir(
self, container: primitives.Container, dir_path: primitives.Directory
) -> None:
""" downloads a container to a directory """
self.logger.debug(
"downloading container to directory %s:%s", container, dir_path
)
client = self._get_client(container)
client.download_dir(dir_path)
class Versions(Endpoint):
""" Onefuzz Instance """