1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
import os
import sys
from django.conf import settings
from datetime import datetime


LOCAL_DEV=True
DEBUG= True


if LOCAL_DEV:
    sys.path.insert(0, '/home/aameer/path_to_project')
    sys.path.insert(1, '/home/aameer/path_to_project')

os.environ['DJANGO_SETTINGS_MODULE'] = 'projectname.settings'

import django
django.setup()

#imports from app
from apps.videos.models import Videos
import boto.elastictranscoder

#debug, un comment for boto logs
#boto.set_stream_logger('boto')

#based on server
if settings.DEBUG:
    #for your s3 bucket bucket
    PIPELINE_ID = 'bucket_pipe_line_id'
    #Singapore, check from aws docs
    DEFAULT_REGION_NAME='ap-southeast-1'

print('Pipeline used for the script:')
print(PIPELINE_ID,DEFAULT_REGION_NAME,'\n')

def cleanup():
    if os.path.isfile('video_encoding.lock'):
        os.unlink('video_encoding.lock')

def awstrancoder():
    print('getting elastic trancoder...')
    try:
        elastic_transcoder= boto.connect_elastictranscoder(aws_secret_access_key=settings.AWS_SECRET_ACCESS_KEY,aws_access_key_id=settings.AWS_ACCESS_KEY_ID)
        elastic_transcoder.DefaultRegionName=DEFAULT_REGION_NAME
        print('AWS elastic_trancoder connetion successful...')
        return elastic_transcoder
    except Exception as awsete:
        print('Failed to connect to AWS Transcode')
        print(awsete)
        cleanup()

def get_input_object(project_name,video_id,orig_key,ext):
    #change this funtion according to your s3 key for input video
    print('gettting obeject with key:')
    print('%s/video/%s_%s.mp4'%(project_name,project_name,orig_key))
    input_object = {
        'Key': '%s/video/%s_%s.mp4'%(project_name,project_name,orig_key),
        'Container': ext,
        'AspectRatio': 'auto',
        'FrameRate': 'auto',
        'Resolution': 'auto',
        'Interlaced': 'auto'
    }
    return (input_object)

def get_output_objects(project_name,video_id):
    #here we want 4 different version of encoded vidoes, you can change the s3 key pattern for output videos and even the bucket here
    #you can get PresetId from amazon docs
    output_objects = [
        {
            'Key': '%s/video/%s_1080.mp4'%(project_name,video_id),
            'PresetId': '1351620000001-000001',
            'Rotate': 'auto',
            'ThumbnailPattern': '',
        },
        {
            'Key': '%s/video/%s_720.mp4'%(project_name,video_id),
            'PresetId': '1351620000001-000010',
            'Rotate': 'auto',
            'ThumbnailPattern': '',
        },
        {
            'Key': '%s/video/%s_480.mp4'%(project_name,video_id),
            'PresetId': '1351620000001-000020',
            'Rotate': 'auto',
            'ThumbnailPattern': '',
        },
        {
            'Key': '%s/video/%s_360.mp4'%(project_name,video_id),
            'PresetId': '1351620000001-000040',
            'Rotate': 'auto',
            'ThumbnailPattern': '',
        }
    ]
    return (output_objects)

def create_transcoding_job(location,project_name,video,elastic_transcoder):
    # we are saving in the same bucket here, you can also save in another bucket
    ext = location.split('.')[-1]
    orig_key = location.split('_')[-1].split('.')[0]
    input_object = get_input_object(project_name,video.id,orig_key,ext)
    output_objects=get_output_objects(project_name,video.id)
    response=elastic_transcoder.create_job(pipeline_id=PIPELINE_ID,input_name=input_object,outputs=output_objects)
    aws_job_id =response['Job']['Id']
    print ("new created job id = %s" % aws_job_id)
    print('\n')
    video.awsencoder_state=aws_job_id
    video.save()
	
def main():
    print('='*50)
    print('timestamp[d/m/y-h:m:s]=',datetime.now().strftime("%d/%m/%Y-%H:%M:%S"),'\n')
    print('Video Encoding started')
    if os.path.exists('video_encoding.lock'):
        print('video lock is present, This may be beacuse script is running already or encountered an error when it was run last time')
        sys.exit()
    else:
        open('video_encoding.lock', 'w').write('running')
    elastic_transcoder=awstrancoder()
    #all video which havent been encoded or any other filter will work
    all_videos=Videos.objects.filter(encoded=False)
    print('no of videos identified for trancoding are %s \n' % all_videos.count())
    for video in all_videos:
        project_name='TestProject'
        video_id=video.id
        if not video.awsencoder_state:
            #no aws job id
            if video.s3_url:
                #check if s3_url is present , we can directly mentin any accessible url here
                create_transcoding_job(
                	location=video.s3_url,
                	project_name=project_name,
                	video=video,
                	elastic_transcoder=elastic_transcoder
                )
        else:
            job_id = video.awsencoder_state
            job_details = elastic_transcoder.read_job(id=job_id)['Job']
            job_status = job_details['Status']
            job_outputs = job_details['Outputs']
            if job_status == 'Submitted':
            	print('Job submitted \n')
            elif job_status == 'Progressing':
            	print('Job progressing \n')
            elif job_status == 'Complete':
                print('Job Complete \n')
                video.encoded = True
                video.save()
            elif job_status == 'Canceled':
            	print('Job Canceled \n')
            elif job_status == 'Error':
            	print('Job encountered Error \n')
            else:
            	print('some unkown job status error occured \n')
        print('-' * 25)
    print('-' * 50)
    cleanup()
    print('Video Encoding script Ended \n')

if __name__ == '__main__':
    main()