mirror of https://github.com/InternLM/InternLM
				
				
				
			fix(storage): unify the name of ak & sk (#527)
* multipart upload * upload * storage * storage * storage * storage * change ak sk name * change ak sk name * change ak sk name * change ak sk name * storage * storagepull/530/head
							parent
							
								
									809ad9ebc8
								
							
						
					
					
						commit
						828033aed5
					
				| 
						 | 
				
			
			@ -328,8 +328,8 @@ class Boto3Client(StorageClient):
 | 
			
		|||
        """S3 object/file storage management class
 | 
			
		||||
 | 
			
		||||
        Args:
 | 
			
		||||
            s3_access_keys_id (str): S3 access key ID.
 | 
			
		||||
            s3_secret_access_key (str): S3 secret access key.
 | 
			
		||||
            ACCESS_KEY (str): S3 access key ID.
 | 
			
		||||
            SECRET_ACCESS_KEY (str): S3 secret access key.
 | 
			
		||||
            use_threads (bool, optional): Whether to enable multipart. Defaults to True.
 | 
			
		||||
            multipart_chunksize (_type_, optional): Defaults to 8*MB.
 | 
			
		||||
            max_concurrency (int, optional): Defaults to 10.
 | 
			
		||||
| 
						 | 
				
			
			@ -340,11 +340,30 @@ class Boto3Client(StorageClient):
 | 
			
		|||
        super().__init__(boto3)
 | 
			
		||||
        self.botocore = botocore
 | 
			
		||||
        try:
 | 
			
		||||
            s3_access_key_id = os.environ["S3_ACCESS_KEY_ID"]
 | 
			
		||||
            s3_secret_access_key = os.environ["S3_SECRET_ACCESS_KEY_ID"]
 | 
			
		||||
            if os.environ.get("S3_ACCESS_KEY_ID") is not None and os.environ.get("ACCESS_KEY") is not None:
 | 
			
		||||
                s3_access_key_id = os.environ["ACCESS_KEY"]
 | 
			
		||||
                logger.warning("Both 'S3_ACCESS_KEY_ID' and 'ACCESS_KEY' exist, 'ACCESS_KEY' will be used by default")
 | 
			
		||||
            elif os.environ.get("ACCESS_KEY") is None:
 | 
			
		||||
                s3_access_key_id = os.environ["S3_ACCESS_KEY_ID"]
 | 
			
		||||
            else:
 | 
			
		||||
                s3_access_key_id = os.environ["ACCESS_KEY"]
 | 
			
		||||
 | 
			
		||||
            if (
 | 
			
		||||
                os.environ.get("S3_SECRET_ACCESS_KEY_ID") is not None
 | 
			
		||||
                and os.environ.get("SECRET_ACCESS_KEY") is not None
 | 
			
		||||
            ):
 | 
			
		||||
                s3_secret_access_key = os.environ["SECRET_ACCESS_KEY"]
 | 
			
		||||
                logger.warning(
 | 
			
		||||
                    "Both 'S3_SECRET_ACCESS_KEY_ID' and 'SECRET_ACCESS_KEY' exist, "
 | 
			
		||||
                    "'SECRET_ACCESS_KEY' will be used by default"
 | 
			
		||||
                )
 | 
			
		||||
            elif os.environ.get("SECRET_ACCESS_KEY") is None:
 | 
			
		||||
                s3_secret_access_key = os.environ["S3_SECRET_ACCESS_KEY_ID"]
 | 
			
		||||
            else:
 | 
			
		||||
                s3_secret_access_key = os.environ["SECRET_ACCESS_KEY"]
 | 
			
		||||
        except KeyError as exc:
 | 
			
		||||
            raise RuntimeError(
 | 
			
		||||
                "Please set boto3 bucket 'S3_ACCESS_KEY_ID' and 'S3_SECRET_ACCESS_KEY_ID' using environment variable!"
 | 
			
		||||
                "Please set boto3 bucket 'ACCESS_KEY' and 'SECRET_ACCESS_KEY' using environment variable!"
 | 
			
		||||
            ) from exc
 | 
			
		||||
 | 
			
		||||
        self.client = self.handler.client(
 | 
			
		||||
| 
						 | 
				
			
			@ -457,8 +476,8 @@ class VolcClient(StorageClient):
 | 
			
		|||
        """Volc object/file storage management class
 | 
			
		||||
 | 
			
		||||
        Args:
 | 
			
		||||
            access_key (str): Volc access key ID.
 | 
			
		||||
            secret_key (str): Volc secret access key.
 | 
			
		||||
            ACCESS_KEY (str): Volc access key ID.
 | 
			
		||||
            SECRET_ACCESS_KEY (str): Volc secret access key.
 | 
			
		||||
            endpoint (str): Volc tos endpoint.
 | 
			
		||||
            region (str): Volc tos region.
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -466,15 +485,34 @@ class VolcClient(StorageClient):
 | 
			
		|||
        super().__init__(tos)
 | 
			
		||||
 | 
			
		||||
        try:
 | 
			
		||||
            access_key = os.environ["VOLC_ACCESS_KEY_ID"]
 | 
			
		||||
            secret_key = os.environ["VOLC_SECRET_ACCESS_KEY_ID"]
 | 
			
		||||
            if os.environ.get("VOLC_ACCESS_KEY_ID") is not None and os.environ.get("ACCESS_KEY") is not None:
 | 
			
		||||
                access_key = os.environ["ACCESS_KEY"]
 | 
			
		||||
                logger.warning("Both 'VOLC_ACCESS_KEY_ID' and 'ACCESS_KEY' exist, 'ACCESS_KEY' will be used by default")
 | 
			
		||||
            elif os.environ.get("ACCESS_KEY") is None:
 | 
			
		||||
                access_key = os.environ["VOLC_ACCESS_KEY_ID"]
 | 
			
		||||
            else:
 | 
			
		||||
                access_key = os.environ["ACCESS_KEY"]
 | 
			
		||||
 | 
			
		||||
            if (
 | 
			
		||||
                os.environ.get("VOLC_SECRET_ACCESS_KEY_ID") is not None
 | 
			
		||||
                and os.environ.get("SECRET_ACCESS_KEY") is not None
 | 
			
		||||
            ):
 | 
			
		||||
                secret_key = os.environ["SECRET_ACCESS_KEY"]
 | 
			
		||||
                logger.warning(
 | 
			
		||||
                    "Both 'VOLC_SECRET_ACCESS_KEY_ID' and 'SECRET_ACCESS_KEY' exist, "
 | 
			
		||||
                    "'SECRET_ACCESS_KEY' will be used by default"
 | 
			
		||||
                )
 | 
			
		||||
            elif os.environ.get("SECRET_ACCESS_KEY") is None:
 | 
			
		||||
                secret_key = os.environ["VOLC_SECRET_ACCESS_KEY_ID"]
 | 
			
		||||
            else:
 | 
			
		||||
                secret_key = os.environ["SECRET_ACCESS_KEY"]
 | 
			
		||||
        except KeyError as exc:
 | 
			
		||||
            raise RuntimeError(
 | 
			
		||||
                "Please set 'VOLC_ACCESS_KEY_ID' and 'VOLC_SECRET_ACCESS_KEY_ID'",
 | 
			
		||||
                "Please set 'ACCESS_KEY' and 'SECRET_ACCESS_KEY'",
 | 
			
		||||
                "using environment variable!",
 | 
			
		||||
            ) from exc
 | 
			
		||||
 | 
			
		||||
        self.client = self.handler.TosClientV2(access_key, secret_key, endpoint, region)
 | 
			
		||||
        self.client = self.handler.TosClientV2(access_key, secret_key, endpoint, region, enable_crc=False)
 | 
			
		||||
 | 
			
		||||
    @staticmethod
 | 
			
		||||
    def sync_upload_fileobj(handler, bucket_name: str, fp: str, saved_obj=None, **kwargs):
 | 
			
		||||
| 
						 | 
				
			
			@ -623,8 +661,8 @@ class AliClient(StorageClient):
 | 
			
		|||
        """Ali object/file storage management class
 | 
			
		||||
 | 
			
		||||
        Args:
 | 
			
		||||
            access_key (str): Ali access key ID.s
 | 
			
		||||
            secret_key (str): Ali secret access key.
 | 
			
		||||
            ACCESS_KEY (str): Ali access key ID.s
 | 
			
		||||
            SECRET_ACCESS_KEY (str): Ali secret access key.
 | 
			
		||||
            endpoint (str): Ali tos endpoint.
 | 
			
		||||
            bucket_name (str): Ali tos bucket_name.
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -632,16 +670,35 @@ class AliClient(StorageClient):
 | 
			
		|||
        super().__init__(oss2)
 | 
			
		||||
 | 
			
		||||
        try:
 | 
			
		||||
            access_key = os.environ["ALI_ACCESS_KEY_ID"]
 | 
			
		||||
            secret_key = os.environ["ALI_SECRET_ACCESS_KEY_ID"]
 | 
			
		||||
            if os.environ.get("ALI_ACCESS_KEY_ID") is not None and os.environ.get("ACCESS_KEY") is not None:
 | 
			
		||||
                access_key = os.environ["ACCESS_KEY"]
 | 
			
		||||
                logger.warning("Both 'ALI_ACCESS_KEY_ID' and 'ACCESS_KEY' exist, 'ACCESS_KEY' will be used by default")
 | 
			
		||||
            elif os.environ.get("ACCESS_KEY") is None:
 | 
			
		||||
                access_key = os.environ["ALI_ACCESS_KEY_ID"]
 | 
			
		||||
            else:
 | 
			
		||||
                access_key = os.environ["ACCESS_KEY"]
 | 
			
		||||
 | 
			
		||||
            if (
 | 
			
		||||
                os.environ.get("ALI_SECRET_ACCESS_KEY_ID") is not None
 | 
			
		||||
                and os.environ.get("SECRET_ACCESS_KEY") is not None
 | 
			
		||||
            ):
 | 
			
		||||
                secret_key = os.environ["SECRET_ACCESS_KEY"]
 | 
			
		||||
                logger.warning(
 | 
			
		||||
                    "Both 'ALI_SECRET_ACCESS_KEY_ID' and 'SECRET_ACCESS_KEY' exist, "
 | 
			
		||||
                    "'SECRET_ACCESS_KEY' will be used by default"
 | 
			
		||||
                )
 | 
			
		||||
            elif os.environ.get("SECRET_ACCESS_KEY") is None:
 | 
			
		||||
                secret_key = os.environ["ALI_SECRET_ACCESS_KEY_ID"]
 | 
			
		||||
            else:
 | 
			
		||||
                secret_key = os.environ["SECRET_ACCESS_KEY"]
 | 
			
		||||
        except KeyError as exc:
 | 
			
		||||
            raise RuntimeError(
 | 
			
		||||
                "Please set 'ALI_ACCESS_KEY_ID' and 'ALI_SECRET_ACCESS_KEY_ID'",
 | 
			
		||||
                "Please set 'ACCESS_KEY' and 'SECRET_ACCESS_KEY'",
 | 
			
		||||
                "using environment variable!",
 | 
			
		||||
            ) from exc
 | 
			
		||||
 | 
			
		||||
        self.auth = self.handler.Auth(access_key, secret_key)
 | 
			
		||||
        self.client = self.handler.Bucket(self.auth, endpoint, bucket_name)
 | 
			
		||||
        self.client = self.handler.Bucket(self.auth, endpoint, bucket_name, enable_crc=False)
 | 
			
		||||
 | 
			
		||||
    @staticmethod
 | 
			
		||||
    def sync_upload_fileobj(handler, fp: str, saved_obj=None, **kwargs):
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
		Reference in New Issue