nojhan/weboob-devel
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

[cineoob] new command persons_in_common

Browse source
This commit is contained in:
Julien Veyssier 2013-03-04 16:57:47 +01:00
commit 0036ee21a7
5 changed files with 80 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

39
weboob/applications/cineoob/cineoob.py
View file

@ -159,7 +159,8 @@ class Cineoob(ReplApplication):
'info_person': 'person_info',
'casting': 'person_list',
'filmography': 'movie_list',
'movies_in_common':'movie_list'
'movies_in_common':'movie_list',
'persons_in_common':'person_list'
}
def complete_info(self, text, line, *ignored):
@ -202,6 +203,42 @@ class Cineoob(ReplApplication):
self.cached_format(movie)
self.flush()
def do_persons_in_common(self, line):
"""
persons_in_common movie_ID movie_ID
Get the list of common persons between two movies.
"""
id1, id2 = self.parse_command_args(line, 2, 1)
self.flush()
movie1 = self.get_object(id1, 'get_movie')
if not movie1:
print >>sys.stderr, 'Movie not found: %s' % id1
return 3
movie2 = self.get_object(id2, 'get_movie')
if not movie2:
print >>sys.stderr, 'Movie not found: %s' % id2
return 3
initial_count = self.options.count
self.options.count = None
lid1 = []
for backend, id in self.do('iter_movie_persons_ids', movie1.id):
lid1.append(id)
self.flush()
lid2 = []
for backend, id in self.do('iter_movie_persons_ids', movie2.id):
lid2.append(id)
self.flush()
self.options.count = initial_count
inter = list(set(lid1) & set(lid2))
for common in inter:
person = self.get_object(common, 'get_person')
self.cached_format(person)
self.flush()
def do_info_movie(self, id):
"""
info_movie movie_ID