From 16a89521f3a42ec5c973a178c205dd28855bf65a Mon Sep 17 00:00:00 2001 From: han yong Date: Thu, 27 Jun 2024 21:11:32 +0800 Subject: [PATCH] use app.config to fetch configurations --- api/extensions/storage/oci_storage.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/api/extensions/storage/oci_storage.py b/api/extensions/storage/oci_storage.py index d385cd4391370..b0c00077893b2 100644 --- a/api/extensions/storage/oci_storage.py +++ b/api/extensions/storage/oci_storage.py @@ -5,20 +5,19 @@ from botocore.exceptions import ClientError from extensions.storage.base_storage import BaseStorage import boto3 -import os class OCIStorage(BaseStorage): def __init__(self, app: Flask): super().__init__(app) app_config = self.app.config - self.bucket_name = os.getenv('OCI_BUCKET_NAME') + self.bucket_name = app_config.get('OCI_BUCKET_NAME') self.client = boto3.client( 's3', - aws_secret_access_key=os.getenv('OCI_SECRET_KEY'), - aws_access_key_id=os.getenv('OCI_ACCESS_KEY'), - endpoint_url=os.getenv('OCI_ENDPOINT'), - region_name=os.getenv('OCI_REGION') + aws_secret_access_key=app_config.get('OCI_SECRET_KEY'), + aws_access_key_id=app_config.get('OCI_ACCESS_KEY'), + endpoint_url=app_config.get('OCI_ENDPOINT'), + region_name=app_config.get('OCI_REGION') ) def save(self, filename, data):