def test_movie_with_when(self):
movie = Movie()
movie.name = u'Titanic'
movie.dvd_release = datetime.strptime('1989-08-17', '%Y-%m-%d')
movie.theater_release = datetime.strptime('1989-02-17', '%Y-%m-%d')
movie.rating = 9
movie.create()
queue = TorrentQueue.load_pending_queue(async=False)
self.assertEquals(1, len(queue))
movie = Movie.load(id=1)
self.assertEquals(1, len(movie))
def _create_directory(self, torrent_queue_id):
""" Create a directory for content of files downloaded """
movie = Movie.find(
Movie.torrent_queue_id == torrent_queue_id, async=False
).one()
if movie is not None:
return self.__mkdir(movie.name)
tv_show = TVShow.find(
TVShow.torrent_queue_id == torrent_queue_id, async=False
).one()
if tv_show is not None:
return self.__mkdir(tv_show.name)
return None
def get_movies(self, **kwargs):
""" Loads all entries for movies
:param kwargs: Pass in argument for querying
:return: defered return
"""
movies = yield Movie.find(**kwargs)
movies.order_by(Desc(Movie.id))
data = list()
for movie in movies:
if movie.torrent_queue.status == 'DELETED':
continue
data.append(movie.dict(json=True))
defer.returnValue(data)
