def loadActresses(self, count=100):
"""function to read actresses data and create a relationship between actress and movie,"""
# read actresses from file
actresses = self.readArtists("actress", count)
self.filePointerActresses = self.fileHandleActresses.tell()
objActress = Actress("", 0)
objMovie = Movie("", "", 0)
for actress in actresses:
name, title, year = actress
actress = objActress.getArtistByName(name, self.actressesList)
# check if actress already not exist then add it
if actress == None:
self.artistCounter += 1
actressId = self.artistCounter
newActress = Actress(actressId, name)
self.actressesList.append(newActress)
else:
actressId = actress.getArtistId()
# check if movie already exists. only if the movie doesn't exist, it will create an ID and
if not objMovie.doesMovieExist(title, self.moviesList) and \
objMovie.getMovieByName(title, self.moviesList) == None:
self.movieCounter += 1
movieId = self.movieCounter
movie = Movie(movieId, title, year)
self.moviesList.append(movie)
# create relation between actor and movie
relation = ArtistMovieRelation(actressId, movieId)
self.relationsList.append(relation)
def loadActors(self, count=100):
"""function to read actors data and create a relationship between actors and movie."""
actors = self.readArtists("actor", count)
self.filePointerActors = self.fileHandleActors.tell()
objActor = Actor("", 0)
objMovie = Movie("", "", 0)
for actor in actors:
name, title, year = actor
# check if actor does not exist else create ID
if objActor.getArtistByName(name, self.actorsList) == None:
self.artistCounter += 1
actorId = self.artistCounter
newActor = Actor(actorId, name)
self.actorsList.append(newActor)
# searches the movie list to check if movies have not been already visualized
if not objMovie.doesMovieExist(title, self.moviesList) and \
objMovie.getMovieByName(title, self.moviesList) == None:
self.movieCounter += 1
movieId = self.movieCounter
movie = Movie(movieId, title, year)
self.moviesList.append(movie)
# create relation between actor and movie
relation = ArtistMovieRelation(actorId, movieId)
self.relationsList.append(relation)