vcs.maemo.org Git - maevies/blob - ui/maeviesui/util/moviemanager.py

   1 # -*- coding: utf-8 -*-
   2
   3 ###########################################################################
   4 #    Maevies
   5 #    Copyright (C) 2010 Simón Pena <spenap@gmail.com>
   6 #
   7 #    This program is free software: you can redistribute it and/or modify
   8 #    it under the terms of the GNU General Public License as published by
   9 #    the Free Software Foundation, either version 3 of the License, or
  10 #    (at your option) any later version.
  11 #
  12 #    This program is distributed in the hope that it will be useful,
  13 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 #    GNU General Public License for more details.
  16 #
  17 #    You should have received a copy of the GNU General Public License
  18 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 ###########################################################################
  20
  21 import dbus
  22 import gobject
  23 from dbus import glib
  24 import gtk
  25 from dbus.exceptions import DBusException
  26
  27 gobject.threads_init()
  28 glib.init_threads()
  29
  30 MINFO_PROVIDER_INTERFACE = 'com.simonpena.maevies.minfoprovider'
  31 MINFO_PROVIDER_BUS_NAME = 'com.simonpena.maevies.minfoprovider'
  32 TMDB_MOVIE_INTERFACE = 'com.simonpena.maevies.tmdbmovie'
  33 TMDB_MOVIE_BUS_NAME = 'com.simonpena.maevies.tmdbmovie'
  34 TMDB = 0
  35 WATC = 1
  36
  37 class MovieManager:
  38
  39     def __init__(self, response_received_cb=None):
  40         self.response_received_cb = response_received_cb
  41         self._bus = None
  42         self._provider = None
  43
  44         self._bus = dbus.SessionBus()
  45
  46     def _get_provider(self):
  47         if not self._provider:
  48             self._provider = self._create_minfo_provider()
  49         return self._provider
  50
  51     def _create_minfo_provider(self):
  52         provider_proxy = self._bus.get_object(MINFO_PROVIDER_BUS_NAME,
  53                                              '/MInfoProvider')
  54         provider_interface = dbus.Interface(provider_proxy,
  55                                             dbus_interface=MINFO_PROVIDER_INTERFACE)
  56         provider_interface.connect_to_signal('ResponseReceived',
  57                                              self._on_response_received)
  58         return provider_interface
  59
  60     def query(self, movie_name='', query_type=TMDB):
  61         try:
  62             self._get_provider().Query(query_type, movie_name)
  63         except DBusException:
  64             self._on_response_received([])
  65
  66     def _on_response_received(self, interface, object_paths):
  67         movies = []
  68         for path in object_paths:
  69             movies.append(MovieProxy(self._bus, path))
  70
  71         if self.response_received_cb:
  72             self.response_received_cb(movies)
  73
  74 class MovieImage:
  75     def __init__(self, image_struct):
  76         self._type, self._url, self._size, self._id = image_struct
  77
  78     def get_url(self):
  79         return self._url
  80
  81     def get_id(self):
  82         return self._id
  83
  84     def get_size(self):
  85         return self._size
  86
  87     def get_type(self):
  88         return self._type
  89
  90     def __str__(self):
  91         str = ''
  92         for field in [self._id, self._url, self._type, self._size]:
  93             str += '\t%s\n' % field
  94         return str
  95
  96 class MovieProxy:
  97
  98     def __init__(self, bus, object_path):
  99         self._bus = bus
 100         self.interface = self._create_movie_interface(object_path)
 101         self.fields = ['Title', 'Release date', 'Rating', 'Popularity']
 102         self._images = self._retrieve_images()
 103
 104     def _create_movie_interface(self, object_path):
 105         proxy = self._bus.get_object(TMDB_MOVIE_BUS_NAME,
 106                                     object_path)
 107         interface = dbus.Interface(proxy,
 108                                    dbus_interface=TMDB_MOVIE_INTERFACE)
 109         return interface
 110
 111     def _retrieve_images(self):
 112         images = { }
 113         dbus_images = self.interface.GetImages()
 114         for dbus_image in dbus_images:
 115             image = MovieImage(dbus_image)
 116             if not images.has_key(image.get_type()):
 117                 images[image.get_type()] = {}
 118             images[image.get_type()][image.get_size()] = image
 119         return images
 120
 121     def get_value(self, field):
 122         if field == 'Title':
 123             return self.get_title()
 124         elif field == 'Release date':
 125             return self.get_released()
 126         elif field == 'Rating':
 127             return self.get_rating()
 128         elif field == 'Popularity':
 129             return self.get_popularity()
 130         else:
 131             return ''
 132
 133     def get_title(self):
 134         return self.interface.GetTitle()
 135
 136     def get_popularity(self):
 137         return self.interface.GetPopularity()
 138
 139     def get_rating(self):
 140         return self.interface.GetRating()
 141
 142     def get_image(self, type, size):
 143         return self._images[type][size]
 144
 145     def get_released(self):
 146         return self.interface.GetReleased()
 147
 148     def get_info(self):
 149         return ('<b>%(title)s</b>\n<small><i>Popularity: </i>%(popularity)s'
 150                 '    <i>Rating: </i>%(rating)s    <i>Released: </i>'
 151                 '%(released)s</small>' %
 152                 {'title' : self.get_title(),
 153                  'popularity' : self.get_popularity(),
 154                  'rating' : self.get_rating(),
 155                  'released' : self.get_released()})
 156
 157     def get_overview(self):
 158         return self.interface.GetOverview()
 159
 160     def get_placeholder_image(self):
 161         return self._get_placeholder_pixbuf()
 162
 163     def _get_placeholder_pixbuf(self):
 164         pixbuf = gtk.IconTheme().load_icon('general_video_file', 48, 0)
 165         return pixbuf