added database and mc packet

2017-07-15 23:04:14 +02:00
parent 392af1f44c
commit 40094452ed
3 changed files with 48 additions and 0 deletions
--- a/youtube.py
+++ b/youtube.py
@@ -0,0 +1,48 @@
+import requests
+from datetime import datetime
+import calendar
+
+def search(what, type, api_key):
+    url = "https://www.googleapis.com/youtube/v3/search?part=snippet&q=%s&type=%s&key=%s" % (what, type, api_key)
+    r = requests.get(url)
+    return r.json()
+
+def playlistItems(playlistId, api_key, all=False, nextToken=None):
+        videos = []
+        info_url = "https://www.googleapis.com/youtube/v3/playlistItems?part=snippet&maxResults=50&playlistId=%s&key=%s" % (
+            playlistId,
+            api_key
+        )
+
+        if nextToken is not None:
+            info_url += "&pageToken=%s" % (nextToken)
+
+        request_data = requests.get(info_url)
+
+        request_data_json = request_data.json()
+
+        videos += parseRequestData(request_data_json)
+        if all == True and 'nextPageToken' in request_data_json:
+            videos += playlistItems(playlistId, api_key, request_data_json['nextPageToken'])
+
+        return videos
+            
+            
+
+def parseRequestData(request_data_json):
+        parsed_videos = []
+        for element in request_data_json['items']:
+            video = {}
+            videotime = datetime.strptime(element['snippet']['publishedAt'][:-5], "%Y-%m-%dT%H:%M:%S")
+            timestamp = calendar.timegm(videotime.utctimetuple())
+            video['timestamp'] = int(timestamp)
+            video['videoid'] = element['snippet']['resourceId']['videoId']
+            video['title'] = element['snippet']['title']
+            video['channel_id'] = element['snippet']['channelId']
+            video['nick'] = element['snippet']['channelTitle']
+            parsed_videos.append(video)
+
+        return parsed_videos            
+            
+
+