weboob-devel/weboob/frontends/videoob/application.py
2010-04-26 19:47:01 +02:00

76 lines
2.9 KiB
Python

# -*- coding: utf-8 -*-
"""
Copyright(C) 2010 Christophe Benz, Romain Bignon
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, version 3 of the License.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
"""
from weboob.capabilities.video import ICapVideoProvider
from weboob.tools.application import ConsoleApplication
class Videoob(ConsoleApplication):
APPNAME = 'videoob'
VERSION = '1.0'
COPYRIGHT = 'Copyright(C) 2010 Christophe Benz, Romain Bignon'
CONFIG = {}
def __init__(self):
ConsoleApplication.__init__(self)
self._parser.add_option('--nsfw', action='store_true', help='enable non-suitable for work videos')
def main(self, argv):
self.load_modules(ICapVideoProvider)
return self.process_command(*argv[1:])
@ConsoleApplication.command('Get video information (accept ID or URL)')
def command_info(self, id):
results = {}
for backend in self.weboob.iter_backends():
video = backend.get_video(id)
if video is None:
continue
rows = []
rows.append(('ID', video.id))
if video.title:
rows.append(('Video title', video.title))
if video.duration:
rows.append(('Duration', '%d:%02d:%02d' % (
video.duration / 3600, (video.duration % 3600) / 60, video.duration % 60)))
if video.url:
rows.append(('URL', video.url))
if video.author:
rows.append(('Author', video.author))
if video.date:
rows.append(('Date', video.date))
if video.rating_max:
rows.append(('Rating', '%s / %s' % (video.rating, video.rating_max)))
elif video.rating:
rows.append(('Rating', video.rating))
results[backend.name] = rows
return results
@ConsoleApplication.command('Search videos')
def command_search(self, pattern=None):
results = {}
if pattern:
results['BEFORE'] = u'Search pattern: %s' % pattern
else:
results['BEFORE'] = u'Last videos'
results['HEADER'] = ('ID', 'Title', 'Duration')
for backend in self.weboob.iter_backends():
results[backend.name] = [(video.id, video.title, video.formatted_duration) for video in
backend.iter_search_results(pattern=pattern, nsfw=self.options.nsfw)]
return results