class MemoryRepository(AbstractRepository):
def __init__(self):
self._movies = WatchList()
self._users = list()
self._actors = list()
self._directors = list()
self._genres = list()
self._reviews = list()
self._i = iter(self._movies)
def add_user(self, user: User):
self._users.append(user)
def get_user(self, username) -> User:
user = None
for u in self._users:
if u.user_name == username:
user = u
return user
def add_movie(self, movie: Movie):
self._movies.add_movie(movie)
def get_movie(self, index: int) -> Movie:
return self._movies.select_movie_to_watch(index)
def get_movie_by_page(self, page: int):
return self._movies.select_movie_by_page(page)
def get_page_of_previous_movie(self, movie: Movie):
pre_page = movie.page - 1
if pre_page < 0:
return 0
else:
return pre_page
def get_page_of_next_movie(self, movie: Movie):
return (movie.page + 1)
def get_number_of_movies(self):
return self._movies.size()
def get_first_movie(self) -> Movie:
return self._movies.first_movie_in_watchlist()
def get_last_movie(self) -> Movie:
return self._movies.last_movie_in_watchlist()
def get_next_movie(self):
return next(self._i)
def add_actor(self, actor: Actor):
self._actors.append(actor)
def get_actor(self) -> List[Actor]:
return self._actors
def add_director(self, director: Director):
self._directors.append(director)
def get_director(self) -> List[Director]:
return self._directors
def add_genre(self, genre: Genre):
self._genres.append(genre)
def get_genre(self) -> List[Genre]:
return self._genres
def test_first_select():
watchlist = WatchList()
watchlist.add_movie(Movie("Moana", 2016))
assert watchlist.first_movie_in_watchlist() == Movie("Moana", 2016)