o }oi @sUddlmZddlZddlZddlmZddlmZddlm Z er/ddlm Z ddl m Z eZ ddlZddlmZWdn1sGwYGd d d ZdddZerjddlmZedZded<dSdS)) annotationsN) TYPE_CHECKING) try_import)ArtifactNotFound)BinaryIO)S3Client) ClientErrorc@sBeZdZdZ ddddd dZdddZdddZdddZdS)Boto3ArtifactStoreaAn artifact backend for Boto3. Args: bucket_name: The name of the bucket to store artifacts. client: A Boto3 client to use for storage operations. If not specified, a new client will be created. avoid_buf_copy: If True, skip procedure to copy the content of the source file object to a buffer before uploading it to S3 ins. This is default to False because using ``upload_fileobj()`` method of Boto3 client might close the source file object. Example: .. code-block:: python import optuna from optuna.artifacts import upload_artifact from optuna.artifacts import Boto3ArtifactStore artifact_store = Boto3ArtifactStore("my-bucket") def objective(trial: optuna.Trial) -> float: ... = trial.suggest_float("x", -10, 10) file_path = generate_example(...) upload_artifact( artifact_store=artifact_store, file_path=file_path, study_or_trial=trial, ) return ... NF)avoid_buf_copy bucket_namestrclientS3Client | Noner boolreturnNonecCs(t||_|p td|_||_dS)Ns3)_importscheckbucketboto3r _avoid_buf_copy)selfr r r rK/home/ubuntu/.local/lib/python3.10/site-packages/optuna/artifacts/_boto3.py__init__;s zBoto3ArtifactStore.__init__ artifact_idrc Csnz |jj|j|d}Wnty)}zt|r$td|jd|d|d}~ww|d}|dus5J|S)NBucketKeyzArtifact storage with bucket: z, artifact_id: z was not foundBody)r get_objectrr_is_not_found_errorrget)rrobjebodyrrr open_readerEs  zBoto3ArtifactStore.open_reader content_bodycCsB|}|jst}t|||d|}|j||j|dS)Nr) rioBytesIOshutil copyfileobjseekr upload_fileobjr)rrr(fsrcbufrrrwriteSs  zBoto3ArtifactStore.writecCs|jj|j|ddS)Nr)r delete_objectr)rrrrrremove\szBoto3ArtifactStore.remove)N)r r r rr rrr)rr rr)rr r(rrr)rr rr)__name__ __module__ __qualname____doc__rr'r1r3rrrrr s&  r r%rrrcCs8|jdid}|jdid}|dkp|dkS)NErrorCodeResponseMetadataHTTPStatusCode NoSuchKeyi)responser#)r% error_codehttp_status_coderrrr"`sr") ArtifactStorer@_)r%rrr) __future__rr)r+typingroptuna._importsroptuna.artifacts.exceptionsrr mypy_boto3_s3rrrbotocore.exceptionsrr r"optuna.artifacts._protocolr@rB__annotations__rrrrs&      K