| # 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) |