Merge pull request #32 from ydb-platform/compression

support compression parameter

Author: gridnevvvit

ydb/export.py

@@ -123,6 +123,7 @@ def __init__(self):
         self.secret_key = None
         self.number_of_retries = 0
         self.storage_class = None
+        self.compression = None
 
     def with_scheme(self, scheme):
         self.scheme = scheme
@@ -132,6 +133,10 @@ def with_storage_class(self, storage_class):
         self.storage_class = storage_class
         return self
 
+    def with_compression(self, compression):
+        self.compression = compression
+        return self
+
     def with_bucket(self, bucket):
         self.bucket = bucket
         return self
@@ -212,6 +217,9 @@ def _export_to_s3_request_factory(settings):
     if settings.number_of_retries > 0:
         request.settings.number_of_retries = settings.number_of_retries
 
+    if settings.compression is not None:
+        request.settings.compression = settings.compression
+
     for source_path, destination_prefix in settings.items:
         request.settings.items.add(
             source_path=source_path,