1 # GNU MediaGoblin -- federated, autonomous media hosting
2 # Copyright (C) 2011, 2012 MediaGoblin contributors. See AUTHORS.
4 # This program is free software: you can redistribute it and/or modify
5 # it under the terms of the GNU Affero General Public License as published by
6 # the Free Software Foundation, either version 3 of the License, or
7 # (at your option) any later version.
9 # This program is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 # GNU Affero General Public License for more details.
14 # You should have received a copy of the GNU Affero General Public License
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
21 from mediagoblin
import mg_globals
as mgg
22 from mediagoblin
.processing
import create_pub_filepath
23 from . import transcoders
27 _log
= logging
.getLogger(__name__
)
28 _log
.setLevel(logging
.DEBUG
)
31 def sniff_handler(media_file
, **kw
):
32 transcoder
= transcoders
.VideoTranscoder()
33 data
= transcoder
.discover(media_file
.name
)
35 _log
.debug('Discovered: {0}'.format(data
))
38 _log
.error('Could not discover {0}'.format(
42 if data
['is_video'] == True:
48 def process_video(entry
):
50 Process a video entry, transcode the queued media files (originals) and
51 create a thumbnail for the entry.
53 video_config
= mgg
.global_config
['media_type:mediagoblin.media_types.video']
55 workbench
= mgg
.workbench_manager
.create_workbench()
57 queued_filepath
= entry
.queued_media_file
58 queued_filename
= workbench
.localized_file(
59 mgg
.queue_store
, queued_filepath
,
62 medium_filepath
= create_pub_filepath(
64 '{original}-640p.webm'.format(
65 original
=os
.path
.splitext(
66 queued_filepath
[-1])[0] # Select the file name without .ext
69 thumbnail_filepath
= create_pub_filepath(
70 entry
, 'thumbnail.jpg')
72 # Create a temporary file for the video destination
73 tmp_dst
= tempfile
.NamedTemporaryFile()
76 # Transcode queued file to a VP8/vorbis file that fits in a 640x640 square
77 transcoder
= transcoders
.VideoTranscoder()
78 transcoder
.transcode(queued_filename
, tmp_dst
.name
)
80 # Push transcoded video to public storage
81 _log
.debug('Saving medium...')
82 mgg
.public_store
.get_file(medium_filepath
, 'wb').write(
84 _log
.debug('Saved medium')
86 entry
.media_files
['webm_640'] = medium_filepath
88 # Save the width and height of the transcoded video
89 entry
.media_data
['video'] = {
90 u
'width': transcoder
.dst_data
.videowidth
,
91 u
'height': transcoder
.dst_data
.videoheight
}
93 # Create a temporary file for the video thumbnail
94 tmp_thumb
= tempfile
.NamedTemporaryFile()
97 # Create a thumbnail.jpg that fits in a 180x180 square
98 transcoders
.VideoThumbnailer(queued_filename
, tmp_thumb
.name
)
100 # Push the thumbnail to public storage
101 _log
.debug('Saving thumbnail...')
102 mgg
.public_store
.get_file(thumbnail_filepath
, 'wb').write(
104 _log
.debug('Saved thumbnail')
106 entry
.media_files
['thumb'] = thumbnail_filepath
108 if video_config
['keep_original']:
109 # Push original file to public storage
110 queued_file
= file(queued_filename
, 'rb')
113 original_filepath
= create_pub_filepath(
117 with mgg
.public_store
.get_file(original_filepath
, 'wb') as \
119 _log
.debug('Saving original...')
120 original_file
.write(queued_file
.read())
121 _log
.debug('Saved original')
123 entry
.media_files
['original'] = original_filepath
125 mgg
.queue_store
.delete_file(queued_filepath
)
127 # Save the MediaEntry