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

[cinema] a person may have several roles in a movie, cap and apps adapted

Browse source
This commit is contained in:
Julien Veyssier 2014-11-26 13:33:35 +01:00 • committed by Romain Bignon
commit 6ec12f70db
3 changed files with 28 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

20
weboob/capabilities/cinema.py
View file

@ -46,17 +46,19 @@ class Movie(BaseObject):
BaseObject.__init__(self, id)
self.original_title = original_title
def get_role_by_person_name(self,name):
def get_roles_by_person_name(self,name):
for role in self.roles.keys():
if name.lower() in [person[1].lower() for person in self.roles[role]]:
return role
return None
def get_role_by_person_id(self,id):
def get_roles_by_person_id(self,id):
result = []
for role in self.roles.keys():
if id in [person[0] for person in self.roles[role]]:
return role
return None
result.append(role)
return result
class Person(BaseObject):
@ -80,7 +82,7 @@ class Person(BaseObject):
BaseObject.__init__(self, id)
self.name = name
def get_role_by_movie_title(self,title):
def get_roles_by_movie_title(self,title):
for role in self.roles.keys():
for mt in [movie[1] for movie in self.roles[role]]:
# title we have is included ?
@ -88,11 +90,13 @@ class Person(BaseObject):
return role
return None
def get_role_by_movie_id(self,id):
def get_roles_by_movie_id(self,id):
result = []
for role in self.roles.keys():
if id in [movie[0] for movie in self.roles[role]]:
return role
return None
result.append(role)
return result
class CapCinema(Capability):