weboob-devel/weboob/applications/booblyrics/booblyrics.py
2013-03-13 01:16:38 +01:00

108 lines
3.3 KiB
Python

# -*- coding: utf-8 -*-
# Copyright(C) 2013 Julien Veyssier
#
# This file is part of weboob.
#
# weboob is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# weboob 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 Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with weboob. If not, see <http://www.gnu.org/licenses/>.
from __future__ import with_statement
import sys
from weboob.capabilities.lyrics import ICapLyrics
from weboob.tools.application.repl import ReplApplication
from weboob.capabilities.base import NotAvailable,NotLoaded
from weboob.tools.application.formatters.iformatter import IFormatter, PrettyFormatter
from weboob.core import CallErrors
__all__ = ['Booblyrics']
class LyricsInfoFormatter(IFormatter):
MANDATORY_FIELDS = ('id', 'title', 'artist', 'content')
def format_obj(self, obj, alias):
result = u'%s%s%s\n' % (self.BOLD, obj.title, self.NC)
result += 'ID: %s\n' % obj.fullid
result += 'Title: %s\n' % obj.title
result += 'Artist: %s\n' % obj.artist
result += '\n%sContent%s\n' % (self.BOLD, self.NC)
result += obj.content
return result
class LyricsListFormatter(PrettyFormatter):
MANDATORY_FIELDS = ('id', 'title', 'artist')
def get_title(self, obj):
return obj.title
def get_description(self, obj):
artist = u''
if obj.artist != NotAvailable and obj.artist != NotLoaded:
artist = obj.artist
return '%s' % artist
class Booblyrics(ReplApplication):
APPNAME = 'booblyrics'
VERSION = '0.f'
COPYRIGHT = 'Copyright(C) 2013 Julien Veyssier'
DESCRIPTION = "Console application allowing to search for song lyrics on various websites."
SHORT_DESCRIPTION = "search and display song lyrics"
CAPS = ICapTorrent
EXTRA_FORMATTERS = {'lyrics_list': LyricsListFormatter,
'lyrics_info': LyricsInfoFormatter,
}
COMMANDS_FORMATTERS = {'search': 'lyrics_list',
'info': 'lyrics_info',
}
def complete_info(self, text, line, *ignored):
args = line.split(' ')
if len(args) == 2:
return self._complete_object()
def do_info(self, id):
"""
info ID
Get information about song lyrics.
"""
songlyrics = self.get_object(id, 'get_lyrics')
if not lyrics:
print >>sys.stderr, 'Song lyrics not found: %s' % id
return 3
self.start_format()
self.format(torrent)
self.flush()
def do_search(self, pattern):
"""
search [PATTERN]
Search lyrics.
"""
self.change_path([u'search'])
if not pattern:
pattern = None
self.start_format(pattern=pattern)
for backend, songlyrics in self.do('iter_lyrics', pattern=pattern):
self.cached_format(songlyrics)
self.flush()