Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

Add youtube-safesearch channel setting, to allow channels to opt-in to turning

Browse source 
off any safe search filters (youtube.py)
This commit is contained in:
jesopo 2019-01-11 11:30:29 +00:00
parent e62b3b58f4
commit 59094568e0
1 changed files with 9 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
modules/youtube.py
View file

@ -19,6 +19,9 @@ ARROW_DOWN = "↓"
@utils.export("channelset", {"setting": "auto-youtube", @utils.export("channelset", {"setting": "auto-youtube",
"help": "Disable/Enable automatically getting info from youtube URLs", "help": "Disable/Enable automatically getting info from youtube URLs",
"validate": utils.bool_or_none}) "validate": utils.bool_or_none})
@utils.export("channelset", {"setting": "youtube-safesearch",
"help": "Turn safe search off/on",
"validate": utils.bool_or_none})
class Module(ModuleManager.BaseModule): class Module(ModuleManager.BaseModule):
def get_video_page(self, video_id, part): def get_video_page(self, video_id, part):
return utils.http.request(URL_YOUTUBEVIDEO, get_params={"part": part, return utils.http.request(URL_YOUTUBEVIDEO, get_params={"part": part,
@ -92,12 +95,16 @@ class Module(ModuleManager.BaseModule):
video_id = url_match.group(1) video_id = url_match.group(1)
if search or video_id: if search or video_id:
safe_setting = event["target"].get_setting("youtube-safesearch",
True)
safe = "moderate" if safe_setting else "none"
if not video_id: if not video_id:
search_page = utils.http.request(URL_YOUTUBESEARCH, search_page = utils.http.request(URL_YOUTUBESEARCH,
get_params={"q": search, "part": "snippet", get_params={"q": search, "part": "snippet",
"maxResults": "1", "type": "video", "maxResults": "1", "type": "video",
"key": self.bot.config["google-api-key"]}, "key": self.bot.config["google-api-key"],
json=True) "safeSearch": safe}, json=True)
if search_page: if search_page:
if search_page.data["pageInfo"]["totalResults"] > 0: if search_page.data["pageInfo"]["totalResults"] > 0:
video_id = search_page.data["items"][0]["id"]["videoId"] video_id = search_page.data["items"][0]["id"]["videoId"]