diff --git a/rest-client-yts.py b/rest-client-yts.py new file mode 100755 index 0000000..5e965ac --- /dev/null +++ b/rest-client-yts.py @@ -0,0 +1,34 @@ +#!/usr/bin/env python3 +# List latest good movies on YTS + +from simple_rest_client.api import API +from simple_rest_client.resource import Resource +import progressbar + + +class MovieResource(Resource): + actions = { + 'list': {'method': 'GET', 'url': 'list_movies.json'}, + } + + +def good(movie): + return movie['rating'] >= 7.5 + + +api = API(api_root_url='https://yts.am/api/v2') +api.add_resource(resource_name='movies', resource_class=MovieResource) + +max_movies = 20 +with progressbar.ProgressBar(max_value=max_movies) as bar: + page = 0 + movies = [] + while len(movies) < max_movies and page < 20: + page += 1 + movies_new = api.movies.list(params={'limit': 50, 'page': page}).body['data']['movies'] + movies_new = [movie for movie in movies_new if good(movie)] + movies += movies_new + bar.update(min(max_movies, len(movies))) + +for movie in sorted(movies, key=lambda m: m['rating'], reverse=True): + print('{rating:3.1f} {title}\n {url}'.format(**movie))