Added rating and views to video information.

YtManagerApp/utils/iterutils.py

@@ -1,3 +1,7 @@
+import itertools
+from typing import Iterable
+
+
 def first_true(*args, default=False, pred=None):
     """Returns the first true value in the iterable.
 
@@ -10,3 +14,19 @@ def first_true(*args, default=False, pred=None):
     # first_true([a,b,c], x) --> a or b or c or x
     # first_true([a,b], x, f) --> a if f(a) else b if f(b) else x
     return next(filter(pred, args), default)
+
+
+def as_chunks(iterable: Iterable, chunk_size: int):
+    """
+    Iterates an iterable in chunks of chunk_size elements.
+    :param iterable: An iterable containing items to iterate.
+    :param chunk_size: Chunk size
+    :return: Returns a generator which will yield chunks of size chunk_size
+    """
+
+    it = iter(iterable)
+    while True:
+        chunk = tuple(itertools.islice(it, chunk_size))
+        if not chunk:
+            return
+        yield chunk

YtManagerApp/utils/youtube.py

@@ -3,10 +3,13 @@ from googleapiclient.errors import Error as APIError
 from google_auth_oauthlib.flow import InstalledAppFlow
 from django.conf import settings
 import re
+from YtManagerApp.utils.iterutils import as_chunks
 
 API_SERVICE_NAME = 'youtube'
 API_VERSION = 'v3'
 
+YOUTUBE_LIST_LIMIT = 50
+
 
 class YoutubeException(Exception):
     pass
@@ -28,6 +31,10 @@ class YoutubePlaylistNotFoundException(YoutubeException):
     pass
 
 
+class YoutubeVideoNotFoundException(YoutubeException):
+    pass
+
+
 class YoutubeChannelInfo(object):
     def __init__(self, result_dict):
         self.__id = result_dict['id']
@@ -130,6 +137,27 @@ class YoutubePlaylistItem(object):
         return self.__snippet['position']
 
 
+class YoutubeVideoStatistics(object):
+    def __init__(self, result_dict):
+        self.id = result_dict['id']
+        self.stats = result_dict['statistics']
+
+    def get_view_count(self):
+        return int(self.stats['viewCount'])
+
+    def get_like_count(self):
+        return int(self.stats['likeCount'])
+
+    def get_dislike_count(self):
+        return int(self.stats['dislikeCount'])
+
+    def get_favorite_count(self):
+        return int(self.stats['favoriteCount'])
+
+    def get_comment_count(self):
+        return int(self.stats['commentCount'])
+
+
 class YoutubeAPI(object):
     def __init__(self, service):
         self.service = service
@@ -230,6 +258,26 @@ class YoutubeAPI(object):
             else:
                 last_page = True
 
+    def get_single_video_stats(self, video_id) -> YoutubeVideoStatistics:
+        result = list(self.get_video_stats([video_id]))
+        if len(result) < 1:
+            raise YoutubeVideoNotFoundException('Could not find video with id ' + video_id + '!')
+        return result[0]
+
+    def get_video_stats(self, video_id_list):
+        for chunk in as_chunks(video_id_list, YOUTUBE_LIST_LIMIT):
+            kwargs = {
+                "part": "statistics",
+                "maxResults": YOUTUBE_LIST_LIMIT,
+                "id": ','.join(chunk)
+            }
+            result = self.service.videos()\
+                .list(**kwargs)\
+                .execute()
+
+            for item in result['items']:
+                yield YoutubeVideoStatistics(item)
+
     # @staticmethod
     # def build_oauth() -> 'YoutubeAPI':
     #     flow =