| Index: third_party/boto/services/sonofmmm.py
|
| diff --git a/third_party/boto/services/sonofmmm.py b/third_party/boto/services/sonofmmm.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..acb7e610673ee0001f1dcb1dda58fc4b4c97684b
|
| --- /dev/null
|
| +++ b/third_party/boto/services/sonofmmm.py
|
| @@ -0,0 +1,81 @@
|
| +# Copyright (c) 2006,2007 Mitch Garnaat http://garnaat.org/
|
| +#
|
| +# Permission is hereby granted, free of charge, to any person obtaining a
|
| +# copy of this software and associated documentation files (the
|
| +# "Software"), to deal in the Software without restriction, including
|
| +# without limitation the rights to use, copy, modify, merge, publish, dis-
|
| +# tribute, sublicense, and/or sell copies of the Software, and to permit
|
| +# persons to whom the Software is furnished to do so, subject to the fol-
|
| +# lowing conditions:
|
| +#
|
| +# The above copyright notice and this permission notice shall be included
|
| +# in all copies or substantial portions of the Software.
|
| +#
|
| +# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
| +# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABIL-
|
| +# ITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT
|
| +# SHALL THE AUTHOR BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
|
| +# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
| +# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
|
| +# IN THE SOFTWARE.
|
| +
|
| +import boto
|
| +from boto.services.service import Service
|
| +from boto.services.message import ServiceMessage
|
| +import os
|
| +import mimetypes
|
| +
|
| +class SonOfMMM(Service):
|
| +
|
| + def __init__(self, config_file=None):
|
| + Service.__init__(self, config_file)
|
| + self.log_file = '%s.log' % self.instance_id
|
| + self.log_path = os.path.join(self.working_dir, self.log_file)
|
| + boto.set_file_logger(self.name, self.log_path)
|
| + if self.sd.has_option('ffmpeg_args'):
|
| + self.command = '/usr/local/bin/ffmpeg ' + self.sd.get('ffmpeg_args')
|
| + else:
|
| + self.command = '/usr/local/bin/ffmpeg -y -i %s %s'
|
| + self.output_mimetype = self.sd.get('output_mimetype')
|
| + if self.sd.has_option('output_ext'):
|
| + self.output_ext = self.sd.get('output_ext')
|
| + else:
|
| + self.output_ext = mimetypes.guess_extension(self.output_mimetype)
|
| + self.output_bucket = self.sd.get_obj('output_bucket')
|
| + self.input_bucket = self.sd.get_obj('input_bucket')
|
| + # check to see if there are any messages queue
|
| + # if not, create messages for all files in input_bucket
|
| + m = self.input_queue.read(1)
|
| + if not m:
|
| + self.queue_files()
|
| +
|
| + def queue_files(self):
|
| + boto.log.info('Queueing files from %s' % self.input_bucket.name)
|
| + for key in self.input_bucket:
|
| + boto.log.info('Queueing %s' % key.name)
|
| + m = ServiceMessage()
|
| + if self.output_bucket:
|
| + d = {'OutputBucket' : self.output_bucket.name}
|
| + else:
|
| + d = None
|
| + m.for_key(key, d)
|
| + self.input_queue.write(m)
|
| +
|
| + def process_file(self, in_file_name, msg):
|
| + base, ext = os.path.splitext(in_file_name)
|
| + out_file_name = os.path.join(self.working_dir,
|
| + base+self.output_ext)
|
| + command = self.command % (in_file_name, out_file_name)
|
| + boto.log.info('running:\n%s' % command)
|
| + status = self.run(command)
|
| + if status == 0:
|
| + return [(out_file_name, self.output_mimetype)]
|
| + else:
|
| + return []
|
| +
|
| + def shutdown(self):
|
| + if os.path.isfile(self.log_path):
|
| + if self.output_bucket:
|
| + key = self.output_bucket.new_key(self.log_file)
|
| + key.set_contents_from_filename(self.log_path)
|
| + Service.shutdown(self)
|
|
|