doubanfm_source.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

# Copyright (C) 2013 liberize <liberize@gmail.com>
#
# 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; either version 2 of the License, or
# (at your option) any later version.
#
# 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, see <http://www.gnu.org/licenses/>.

from gi.repository import GObject, RB, GLib
from login_dialog import LoginDialog
import datetime, _thread

SOURCE_NAME = '豆瓣电台'

class DoubanFMSource(RB.BrowserSource):
	__gtype_name__ = 'DoubanFMSource'
	object = GObject.property(type=GObject.Object)	
	
	def __init__(self):
		RB.BrowserSource.__init__(self, name=SOURCE_NAME)

		self.activated = False
		self.db = None
		self.doubanfm = None
		self.songs_map = {}
		self.songs = []
		self.history = []
		self.login_dialog = None
	
	def do_selected(self):
		if not self.activated:
			self.activated = True
			self.shell = self.props.shell
			self.plugin = self.props.plugin
			self.player = self.shell.props.shell_player
			self.db = self.shell.props.db
			self.entry_type = self.props.entry_type
		if self.doubanfm == None:
			if self.login_dialog == None:
				self.login_dialog = LoginDialog(self.shell)
				self.login_dialog.connect('login-completed', self.on_login_completed)
			self.login_dialog.login()
		self.plugin.show_sidebar(True)
		
	def do_deselected(self):
		self.plugin.show_sidebar(False)
		
	def on_login_completed(self, dialog, doubanfm):
		self.doubanfm = doubanfm
		self.plugin.change_menu_item_state(True)

	def set_channel(self, channel):
		self.doubanfm.channel = channel
		self.new_playlist()

	def get_song_by_title(self, song_title):
		return self.songs_map.get(song_title.decode('utf-8'), None)

	def add_song(self, song):
		entry = RB.RhythmDBEntry.new(self.db, self.entry_type, song.url)
		self.db.entry_set(entry, RB.RhythmDBPropType.TITLE, song.title.encode('utf-8'))
		self.db.entry_set(entry, RB.RhythmDBPropType.ALBUM, song.albumtitle.encode('utf-8'))
		self.db.entry_set(entry, RB.RhythmDBPropType.ARTIST, song.artist.encode('utf-8'))
		self.db.entry_set(entry, RB.RhythmDBPropType.DURATION, song.length)
		if song.rating_avg:
			self.db.entry_set(entry, RB.RhythmDBPropType.RATING, song.rating_avg)
		if song.kbps:
			self.db.entry_set(entry, RB.RhythmDBPropType.BITRATE, int(song.kbps.encode('utf-8')))
		if song.company:
			self.db.entry_set(entry, RB.RhythmDBPropType.GENRE, song.company.encode('utf-8'))
		if song.public_time:
			date = datetime.date(int(song.public_time.encode('utf-8')), 1, 1).toordinal()
			self.db.entry_set(entry, RB.RhythmDBPropType.DATE, date)

	def reset_songs(self, songs):
		for row in self.props.query_model:
			self.db.entry_delete(row[0])
		self.songs = songs
		self.songs_map = {}
		for song in self.songs:
			self.songs_map[song.title] = song
			self.add_song(song)
		self.db.commit()
		GLib.idle_add(self.start_playing)
		
	def start_playing(self):
		if self.player.get_playing_entry() == None:
			self.player.set_playing_source(self)
			self.player.do_next()

	def new_playlist(self):
		_thread.start_new_thread(self.doubanfm.new_playlist, (self.history,
			self.reset_songs))

	def del_song(self, song):
		sids = [each.sid for each in self.songs]
		next = sids.index(song.sid) + 1
		_thread.start_new_thread(self.doubanfm.del_song, (song.sid, song.aid,
			sids[next:], self.reset_songs))

	def fav_song(self, song):
		_thread.start_new_thread(self.doubanfm.fav_song, (song.sid, song.aid))
		song.like = True

	def unfav_song(self, song):
		_thread.start_new_thread(self.doubanfm.unfav_song, (song.sid, song.aid))
		song.like = False

	def skip_song(self, song):
		_thread.start_new_thread(self.doubanfm.skip_song, (song.sid, song.aid,
			self.history, self.reset_songs))
		if (song.sid, 's') not in self.history:
			self.history.insert(0, (song.sid, 's'))
			if len(self.history) > 20:
				self.history.pop()

	def played_song(self, song):
		_thread.start_new_thread(self.doubanfm.played_song, (song.sid, song.aid))
		if (song.sid, 'p') not in self.history:
			self.history.insert(0, (song.sid, 'p'))
			if len(self.history) > 20:
				self.history.pop()
		
	def played_list(self, song):
		_thread.start_new_thread(self.doubanfm.played_list, (song.sid, self.history,
			self.reset_songs))
		
GObject.type_register(DoubanFMSource)