Browse Source

chore: (#9089 followup) fix storage factory constructor (#9609)

Bowen Liang 6 months ago
parent
commit
d170d78530

+ 2 - 3
api/extensions/storage/aws_s3_storage.py

@@ -4,7 +4,6 @@ from collections.abc import Generator
 import boto3
 from botocore.client import Config
 from botocore.exceptions import ClientError
-from flask import Flask
 
 from extensions.storage.base_storage import BaseStorage
 
@@ -14,8 +13,8 @@ logger = logging.getLogger(__name__)
 class AwsS3Storage(BaseStorage):
     """Implementation for Amazon Web Services S3 storage."""
 
-    def __init__(self, app: Flask):
-        super().__init__(app)
+    def __init__(self):
+        super().__init__()
         app_config = self.app.config
         self.bucket_name = app_config.get("S3_BUCKET_NAME")
         if app_config.get("S3_USE_AWS_MANAGED_IAM"):

+ 2 - 3
api/extensions/storage/supabase_storage.py

@@ -2,7 +2,6 @@ import io
 from collections.abc import Generator
 from pathlib import Path
 
-from flask import Flask
 from supabase import Client
 
 from extensions.storage.base_storage import BaseStorage
@@ -11,8 +10,8 @@ from extensions.storage.base_storage import BaseStorage
 class SupabaseStorage(BaseStorage):
     """Implementation for supabase obs storage."""
 
-    def __init__(self, app: Flask):
-        super().__init__(app)
+    def __init__(self):
+        super().__init__()
         app_config = self.app.config
         self.bucket_name = app_config.get("SUPABASE_BUCKET_NAME")
         self.client = Client(