import oss2

from app.core.config.settings import AliyunOssConfig


class OssUtils:
    def __init__(self, config: AliyunOssConfig):
        self.bucket = oss2.Bucket(
            oss2.Auth(config.access_key_id, config.access_key_secret),
            config.endpoint,
            config.bucket_name,
        )

    def fetch_oss_file(self, oss_key):
        return self.bucket.get_object(oss_key)

    def save_from_file(self, file_path, oss_key):
        return self.bucket.put_object_from_file(key=oss_key, filename=file_path)