# -*- coding: utf-8 -*-
"""
@Author ： Rocky
@Time ： 2022/10/17 16:57
@File ：AmazonKVSUtil.py
"""
import boto3
from Object.AWS.AmazonKinesisVideoUtil import AmazonKinesisVideoObject


class AmazonKVAMObject:
    def __init__(self, aws_access_key_id, secret_access_key, region_name, stream_name, api_name):
        self.access_id = aws_access_key_id
        self.access_secret = secret_access_key
        self.region_name = region_name
        self.kv_client_conn = AmazonKinesisVideoObject(aws_access_key_id, secret_access_key, region_name)
        self.endpoint = self.kv_client_conn.get_data_endpoint(stream_name, api_name)
        self.kvam_client_coon = boto3.client(
            'kinesis-video-archived-media',
            endpoint_url=self.endpoint,
            aws_access_key_id=aws_access_key_id,
            aws_secret_access_key=secret_access_key,
            region_name=region_name
        )

    def get_hls_streaming_session_url(self, stream_name):
        """
        获取视频流数据保留时间
        @param stream_name: 视频流名称
        @return HLSStreamingSessionURL: 媒体播放器可用于检索HLS主播放列表的URL
        """
        return self.kvam_client_coon.get_hls_streaming_session_url(StreamName=stream_name)['HLSStreamingSessionURL']