[v0.8] Updates and Improvements (#8)

[v0.8] Updates and Improvements
sharkykh · Apr 23, 2019 · d6fc71e · d6fc71e
2 parents 5198e28 + 4da3817
commit d6fc71e
Show file tree

Hide file tree

Showing 6 changed files with 582 additions and 455 deletions.
diff --git a/send_test_data.py b/send_test_data.py
@@ -4,25 +4,47 @@
 from __future__ import unicode_literals
 
 import random
+from time import sleep
 
 from tppm import Application
 
 from trakt import Trakt
 
+TEST_DATA = {
+    'movie': [
+        {'tmdb': 404368},  # Ralph Breaks the Internet (2018)
+    ],
+    'episode': [
+        {'tvdb': 5951173},  # Tá no Ar: A TV na TV: 4x04 Episode 4
+        {'tvdb': 5950264},  # Democracy Now!: 2017x32 Tuesday, February 14, 2017
+        {'tvdb': 5966778},  # All In with Chris Hayes: 2017x32 February 14, 2017
+        {'tvdb': 5966780},  # All In with Chris Hayes: 2017x33 February 15, 2017
+        {'tvdb': 5940102},  # Chicago Fire: 5x13 Trading in Scuttlebutt
+        {'tvdb': 6636942},  # Westworld: 2x03 Virtù e Fortuna
+    ]
+}
+
 
 def gen_progress():
     return random.uniform(0.0, 100.0)
 
 
 def send_test_data():
-    Trakt['scrobble'].pause(movie={'ids': {'tmdb': 118340}}, progress=gen_progress())
-    Trakt['scrobble'].pause(episode={'ids': {'tvdb': 5761493}}, progress=gen_progress())
-    Trakt['scrobble'].pause(episode={'ids': {'tvdb': 5951173}}, progress=gen_progress())
-    Trakt['scrobble'].pause(episode={'ids': {'tvdb': 5950264}}, progress=gen_progress())
-    Trakt['scrobble'].pause(episode={'ids': {'tvdb': 5966778}}, progress=gen_progress())
-    Trakt['scrobble'].pause(episode={'ids': {'tvdb': 5966780}}, progress=gen_progress())
-    Trakt['scrobble'].pause(episode={'ids': {'tvdb': 5940102}}, progress=gen_progress())
-    Trakt['scrobble'].pause(episode={'ids': {'tvdb': 6636942}}, progress=gen_progress())
+    index = 1
+    total = len(TEST_DATA['movie']) + len(TEST_DATA['episode'])
+    for category, items in TEST_DATA.items():
+        for item in items:
+            params = {
+                category: {
+                    'ids': item
+                },
+                'progress': gen_progress()
+            }
+            print('Sending [%d/%d]: %s' % (index, total, item))
+            Trakt['scrobble'].pause(**params)
+            index += 1
+
+            sleep(0.1)
 
     print('Test data sent.')