Update boto3 config for cron service (#611)

duolicious · Jan 18, 2025 · 315b2d7 · 315b2d7
1 parent d6e8a24
commit 315b2d7
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/service/cron/checkphotos/__init__.py b/service/cron/checkphotos/__init__.py
@@ -8,6 +8,7 @@
 )
 import asyncio
 import boto3
+from botocore.config import Config
 import os
 import random
 import io
@@ -42,6 +43,10 @@
     endpoint_url=BOTO_ENDPOINT_URL,
     aws_access_key_id=R2_ACCESS_KEY_ID,
     aws_secret_access_key=R2_ACCESS_KEY_SECRET,
+    config=Config(
+        request_checksum_calculation='WHEN_REQUIRED',
+        response_checksum_validation='WHEN_REQUIRED',
+    ),
 )
 
 async def update_blurhashes(uuids: list[str]):

diff --git a/service/cron/util/__init__.py b/service/cron/util/__init__.py
@@ -4,6 +4,7 @@
 from service.cron.util.sql import *
 import asyncio
 import boto3
+from botocore.config import Config
 import io
 import os
 import traceback
@@ -53,6 +54,10 @@ async def delete_images_from_object_store(
         endpoint_url=BOTO_ENDPOINT_URL,
         aws_access_key_id=R2_ACCESS_KEY_ID,
         aws_secret_access_key=R2_ACCESS_KEY_SECRET,
+        config=Config(
+            request_checksum_calculation='WHEN_REQUIRED',
+            response_checksum_validation='WHEN_REQUIRED',
+        ),
     )
 
     for chunk in chunks: