diff options
| author | 2019-11-25 18:17:30 +0000 | |
|---|---|---|
| committer | 2019-11-25 18:17:30 +0000 | |
| commit | 6a6e789ec9e3c6dd365922ac630a405b830e8c1a (patch) | |
| tree | 7e2e1675d7d1fab7d4240da83c0179076f89c938 /modules/youtube.py | |
| parent | include <title> with !google/!g output (diff) | |
add `cookies` and `.json()` to utils.http.Response objects
Diffstat (limited to 'modules/youtube.py')
| -rw-r--r-- | modules/youtube.py | 25 |
1 files changed, 11 insertions, 14 deletions
diff --git a/modules/youtube.py b/modules/youtube.py index 56866681..24a662bf 100644 --- a/modules/youtube.py +++ b/modules/youtube.py @@ -30,8 +30,7 @@ class Module(ModuleManager.BaseModule): def get_video_page(self, video_id): return utils.http.request(URL_YOUTUBEVIDEO, get_params={ "part": "contentDetails,snippet,statistics", - "id": video_id, "key": self.bot.config["google-api-key"]}, - json=True) + "id": video_id, "key": self.bot.config["google-api-key"]}).json() def _number(self, n): if n: @@ -39,8 +38,8 @@ class Module(ModuleManager.BaseModule): def video_details(self, video_id): page = self.get_video_page(video_id) - if page.data["items"]: - item = page.data["items"][0] + if page["items"]: + item = page["items"][0] snippet = item["snippet"] statistics = item["statistics"] content = item["contentDetails"] @@ -91,8 +90,8 @@ class Module(ModuleManager.BaseModule): "key": self.bot.config["google-api-key"]}, json=True) def playlist_details(self, playlist_id): page = self.get_playlist_page(playlist_id) - if page.data["items"]: - item = page.data["items"][0] + if page["items"]: + item = page["items"][0] snippet = item["snippet"] content = item["contentDetails"] @@ -121,12 +120,11 @@ class Module(ModuleManager.BaseModule): search_page = utils.http.request(URL_YOUTUBESEARCH, get_params={"q": query, "part": "snippet", "maxResults": "1", "type": "video", - "key": self.bot.config["google-api-key"]}, - json=True) + "key": self.bot.config["google-api-key"]}).json() if search_page: - if search_page.data["pageInfo"]["totalResults"] > 0: - video_id = search_page.data["items"][0]["id"]["videoId"] + if search_page["pageInfo"]["totalResults"] > 0: + video_id = search_page["items"][0]["id"]["videoId"] return "https://youtu.be/%s" % video_id @utils.hook("received.command.yt", alias_of="youtube") @@ -157,11 +155,10 @@ class Module(ModuleManager.BaseModule): search_page = utils.http.request(URL_YOUTUBESEARCH, get_params={"q": search, "part": "snippet", "maxResults": "1", "type": "video", "key": self.bot.config["google-api-key"], - "safeSearch": safe}, json=True) + "safeSearch": safe}).json() if search_page: - if search_page.data["pageInfo"]["totalResults"] > 0: - url = URL_VIDEO % search_page.data[ - "items"][0]["id"]["videoId"] + if search_page["pageInfo"]["totalResults"] > 0: + url = URL_VIDEO % search_page["items"][0]["id"]["videoId"] else: raise utils.EventError("No videos found") else: |
