xenofem/python-zulip-api
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

zulip_bots: Check API key on initialization for link_shortener bot.

Browse source
This commit is contained in:
Viraat Chandra 2017-12-27 20:26:25 +05:30 • committed by Robert Hönig
parent 79b32f6c05
commit db0e88a1b9
1 changed files with 15 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

15
zulip_bots/zulip_bots/bots/link_shortener/link_shortener.py
View file

@ -1,5 +1,6 @@
import re import re
import requests import requests
import logging
from typing import Any, Dict from typing import Any, Dict
@ -16,6 +17,20 @@ class LinkShortenerHandler(object):
def initialize(self, bot_handler: Any) -> None: def initialize(self, bot_handler: Any) -> None:
self.config_info = bot_handler.get_config_info('link_shortener') self.config_info = bot_handler.get_config_info('link_shortener')
self.check_api_key()
def check_api_key(self) -> None:
test_request = requests.post(
'https://www.googleapis.com/urlshortener/v1/url',
json={'longUrl': 'www.youtube.com/watch'},
params={'key': self.config_info['key']}
) # type: Any
test_request_data = test_request.json()
try:
if test_request_data['error']['errors'][0]['reason'] == 'keyInvalid':
logging.error('Invalid key. Follow the instructions in doc.md for setting API key.')
except KeyError:
pass
def handle_message(self, message: Dict[str, str], bot_handler: Any) -> None: def handle_message(self, message: Dict[str, str], bot_handler: Any) -> None:
REGEX_STR = ( REGEX_STR = (