89 lines
3.1 KiB
Python
89 lines
3.1 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
|
|
from weboob.tools.application.results import BaseItem, FieldException, ObjectItem
|
|
|
|
|
|
class VideoItem(ObjectItem):
|
|
def format(self, select=[]):
|
|
if select:
|
|
return u'\t'.join(self.get(field) for field in select)
|
|
else:
|
|
lines = [
|
|
u'ID: %s' % self.obj.id,
|
|
u'title: %s'% self.obj.title,
|
|
u'duration: %s'% self.obj.formatted_duration,
|
|
u'URL: %s'% self.obj.url,
|
|
u'author: %s'% self.obj.author,
|
|
u'date: %s'% self.obj.date,
|
|
u'rating_max: %s'% self.obj.rating,
|
|
u'rating: %s'% self.obj.rating_max,
|
|
]
|
|
return u'\n'.join(lines)
|
|
|
|
|
|
class SearchResultItem(BaseItem):
|
|
def __init__(self, _id, title, duration):
|
|
self.id = _id
|
|
self.title = title
|
|
self.duration = duration
|
|
|
|
def get(self, name):
|
|
try:
|
|
return getattr(self, name)
|
|
except AttributeError:
|
|
raise FieldException(name)
|
|
|
|
def format(self, select=[]):
|
|
if select:
|
|
return u'\t'.join(unicode(self.get(field)) for field in select)
|
|
else:
|
|
return u'%s %s (%s)' % (self.id, self.title, self.duration)
|
|
|
|
|
|
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):
|
|
for backend, video in self.weboob.do('get_video', id):
|
|
if video is None:
|
|
continue
|
|
print self.format(VideoItem(video))
|
|
|
|
@ConsoleApplication.command('Search videos')
|
|
def command_search(self, pattern=None):
|
|
print u'Search pattern: %s' % pattern if pattern else u'Last videos'
|
|
for backend, video in self.weboob.do(
|
|
'iter_search_results', pattern=pattern, nsfw=self.options.nsfw):
|
|
print self.format(SearchResultItem(video.id, title=video.title, duration=video.formatted_duration))
|