return poster object

This commit is contained in:
j 2010-09-06 22:45:11 +02:00
parent 32375a9ff9
commit 1330646d52

View file

@ -197,6 +197,16 @@ class Movie(models.Model):
'poster_width': 'posterWidth', 'poster_width': 'posterWidth',
'poster_height': 'posterHeight' 'poster_height': 'posterHeight'
} }
def poster_json(self):
poster = {}
poster['width'] = self.poster_width
poster['height'] = self.poster_height
if self.poster:
poster['url'] = self.poster.url
else:
poster['url'] = self.poster_url
return poster
def get_json(self, fields=None): def get_json(self, fields=None):
movie = {} movie = {}
for key in self._public_fields: for key in self._public_fields:
@ -213,6 +223,7 @@ class Movie(models.Model):
if not fields: if not fields:
movie['poster'] = self.get_poster() movie['poster'] = self.get_poster()
movie['stream'] = self.get_stream() movie['stream'] = self.get_stream()
movie['poster'] = self.poster_json()
if fields: if fields:
for f in fields: for f in fields:
if f.endswith('.length') and f[:-7] in ('cast', 'genre', 'trivia'): if f.endswith('.length') and f[:-7] in ('cast', 'genre', 'trivia'):