examples/api_test/aws/get_token.py

#!/usr/bin/python3
# -*- coding: utf-8 -*-
import argparse
from datetime import datetime
import json
import os
from pathlib import Path
import sys
import time
from zoneinfo import ZoneInfo  # Python 3.9+ 自带，无需安装

pwd = os.path.abspath(os.path.dirname(__file__))
sys.path.append(os.path.join(pwd, "../"))

import boto3

from project_settings import environment, project_path


def get_args():
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--service",
        default="aws_us_east",
        type=str
    )
    args = parser.parse_args()
    return args


def main():
    args = get_args()

    service = environment.get(key=args.service, dtype=json.loads)
    print(service)
    aws_access_key_id = service["AWS_ACCESS_KEY_ID"]
    aws_secret_access_key = service["AWS_SECRET_ACCESS_KEY"]
    aws_default_region = service["AWS_DEFAULT_REGION"]

    os.environ["AWS_ACCESS_KEY_ID"] = aws_access_key_id
    os.environ["AWS_SECRET_ACCESS_KEY"] = aws_secret_access_key
    os.environ["AWS_DEFAULT_REGION"] = aws_default_region

    sts_client = boto3.client('sts')

    response = sts_client.get_session_token(
        DurationSeconds=3600
    )

    return


if __name__ == "__main__":
    main()