→ rest-client-yts/

rest-client-yts/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))