Computer >> บทช่วยสอนคอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

วิธีใช้ Boto3 ดึง Secret แบบ Plain Text จากรูปแบบ Binary/Encrypted ใน AWS Secrets Manager

ข้อกำหนดปัญหา: ใช้ไลบรารี boto3 ใน Python เพื่อดึงค่า secret แบบ plain text จากรูปแบบ binary/encrypted ที่เก็บใน AWS Secrets Manager

ขั้นตอนการแก้ปัญหา

  • ขั้นตอน 1: นำเข้า boto3, botocore.exceptions และ base64

  • ขั้นตอน 2: secret_stored_location เป็นพารามิเตอร์ที่จำเป็น

  • ขั้นตอน 3: สร้าง AWS session ด้วย boto3 พร้อมระบุ region_name

  • ขั้นตอน 4: สร้าง AWS client สำหรับ secretsmanager

  • ขั้นตอน 5: เรียก get_secret_value โดยส่ง SecretId

  • ขั้นตอน 6: ตรวจสอบว่าค่าที่ได้เป็น plain text (SecretString) หรือ encrypted (SecretBinary)

  • ขั้นตอน 7: หากเป็น encrypted ให้ถอดรหัสด้วย base64.b64decode

  • ขั้นตอน 8: คืนค่า secret ในรูปแบบ plain text

  • ขั้นตอน 9: จัดการข้อยกเว้นทั่วไป

โค้ดตัวอย่าง

import boto3
import base64
from botocore.exceptions import ClientError

def get_decrypted_secret_details(secret_stored_location):
    session = boto3.session.Session()
    s3_client = session.client('secretsmanager')
    try:
        response = s3_client.get_secret_value(SecretId=secret_stored_location)
        if 'SecretString' not in response:
            decoded_secret_values = base64.b64decode(response['SecretBinary'])
            return decoded_secret_values
        return response['SecretString']
    except ClientError as e:
        raise Exception("boto3 client error in get_decrypted_secret_details: " + e.__str__())
    except Exception as e:
        raise Exception("Unexpected error in get_decrypted_secret_details: " + e.__str__())

a = get_decrypted_secret_details('/secrets/aws')
print(a)

ผลลัพธ์

{"user":"SERVICE_USER","accesskey":"I**************"}

ฟังก์ชันนี้จัดการทั้ง secret แบบ string และ binary อัตโนมัติ เหมาะสำหรับการดึงข้อมูลที่อาจถูกเก็บในรูปแบบต่างกัน