Python script for AWS Elastic Transcoder with boto api

boto-elastic-transcoder-manager-example.py

# -*- coding: utf-8 -*-

import hashlib

import boto3


class ETSManager:
    """
        @todo: manages and provides the ets services
    """
    preset_ids = {
        'hls-2000k': '1351620000001-200010',
        'hls-1500k': '1351620000001-200020',
        'hls-1000k': '1351620000001-200030'
    }

    segment_duration = '4'

    def __init__(self):
        self.client = boto3.client('elastictranscoder', 'ap-northeast-1')

    def transcode_hls(self, pipeline_id, input_key, output_key_prefix):
        """
        Args:
            pipeline_id: id of pipeline for transcoding video for hls
            input_key: uploaded video key (video is stored in S3)
            output_key_prefix: prefix of directory of filename for m3u file
        Returns:
            m3u8 file key of playlist
        """
        output_key = hashlib.sha256(input_key.encode('utf-8')).hexdigest()

        input = {
            'Key': input_key
        }

        hls_2000k = {
            'Key': 'hls2000k/' + output_key,
            'ThumbnailPattern': 'hls2000k/thumbnail/' + output_key + '-{resolution}-{count}',
            'PresetId': self.preset_ids['hls-2000k'],
            'SegmentDuration': self.segment_duration
        }

        hls_1500k = {
            'Key': 'hls1500k/' + output_key,
            'ThumbnailPattern': 'hls1500k/thumbnail/' + output_key + '-{resolution}-{count}',
            'PresetId': self.preset_ids['hls-1500k'],
            'SegmentDuration': self.segment_duration
        }

        hls_1000k = {
            'Key': 'hls1000k/' + output_key,
            'ThumbnailPattern': 'hls1000k/thumbnail/' + output_key + '-{resolution}-{count}',
            'PresetId': self.preset_ids['hls-1000k'],
            'SegmentDuration': self.segment_duration
        }

        outputs = [
            hls_2000k, hls_1500k, hls_1000k
        ]

        playlists = [
            {
                'Name': 'hls_' + output_key,
                'Format': 'HLSv3',
                'OutputKeys': [
                    hls_2000k['Key'], hls_1500k['Key'], hls_1000k['Key']
                ],
                'HlsContentProtection': {
                    'Method': 'aes-128',
                    'KeyStoragePolicy': 'WithVariantPlaylists'
                }
            }
        ]

        job_result = self.client.create_job(PipelineId=pipeline_id,
                                          Input=input,
                                          Outputs=outputs,
                                          OutputKeyPrefix=output_key_prefix + output_key + '/',
                                          Playlists=playlists)

        return '/'.join([output_key, job_result['Job']['Playlists'][0]['Name'] + '.m3u8'])