Remove tqdm

pyproject.toml

@@ -8,5 +8,4 @@ dependencies = [
     "docker>=7.1.0",
     "google-cloud-storage>=2.19.0",
     "toml>=0.10.2",
-    "tqdm>=4.67.1",
 ]

storage.py

@@ -1,8 +1,6 @@
-import io
 import os
 import subprocess
 from google.cloud.storage import Client, Bucket
-from tqdm import tqdm
 
 def create_tar(paths: list[str], tar_name: str) -> None:
     """Create an uncompressed tarball without including the full path of the files inside."""
@@ -28,41 +26,24 @@ def extract_tar(tar_path: str, extract_to_path: str) -> None:
     print(f"Extraction complete: {extract_to_path}")
 
 
-def download_from_bucket(storage_client: Client, bucket_name: str, source_blob_name: str, destination_file_name: str) -> None:
-    """Download a file from Google Cloud Storage with progress tracking."""
-    try:
-        bucket = storage_client.get_bucket(bucket_name)
-        blob = bucket.blob(source_blob_name)
-
-        with open(destination_file_name, 'wb') as f:
-            with tqdm.wrapattr(f, "write", total=blob.size, desc=f"Downloading {source_blob_name}") as file_obj:
-                blob.download_to_file(file_obj)
-
-        print(f"File {source_blob_name} downloaded to {destination_file_name}.")
-    except Exception as e:
-        print(f"Error downloading file: {e}")
+def download_from_bucket(storage_client: Client, bucket_name: str, bucket_path: str, local_path: str) -> None:
+    """Download the snapshot tar file from Google Cloud Storage."""
+    bucket: Bucket = storage_client.get_bucket(bucket_name)
+    blob = bucket.blob(bucket_path)
+    print(f"Downloading {bucket_path} from gs://{bucket_name}...")
+
+    # Download the tar file to the local path
+    blob.download_to_filename(local_path)
+    print(f"Downloaded {bucket_path} to {local_path}")
 
 
-def upload_file_to_bucket(
-    storage_client: Client,
-    bucket_name: str,
-    source_file: str,
-    destination_blob_name: str
-) -> None:
-    """Upload a file to the specified Google Cloud Storage bucket with progress tracking."""
+def upload_file_to_bucket(storage_client: Client, bucket_name: str, source_file: str, destination_blob_name: str) -> None:
+    """Upload a file to the specified Google Cloud Storage bucket."""
     try:
-        bucket = storage_client.get_bucket(bucket_name)
+        print(f"Start uploading to {destination_blob_name}.")
+        bucket: Bucket = storage_client.get_bucket(bucket_name)
         blob = bucket.blob(destination_blob_name)
-
-        file_size = os.path.getsize(source_file)
-
-        with tqdm(total=file_size, unit='B', unit_scale=True, desc=f"Uploading {source_file}") as pbar:
-            with open(source_file, "rb") as f:
-                chunk_size = 1024 * 1024 * 32
-                for chunk in iter(lambda: f.read(chunk_size), b""):
-                    blob.upload_from_file(io.BytesIO(chunk), content_type="application/octet-stream")
-                    pbar.update(len(chunk))
-
+        blob.upload_from_filename(source_file)
         print(f"File {source_file} uploaded to {destination_blob_name}.")
     except Exception as e:
         print(f"Error uploading file: {e}")