bots: Update foursquare bot to use get_config_info().

This commit is contained in:
Robert Hönig 2017-06-18 17:06:07 +02:00 committed by showell
parent c4a85b2f74
commit a10ee59853
2 changed files with 2 additions and 26 deletions

View file

@ -4,26 +4,11 @@ from __future__ import absolute_import
import datetime as dt import datetime as dt
import re import re
import requests import requests
from os.path import expanduser
from six.moves import configparser as cp
from six.moves import range from six.moves import range
home = expanduser('~')
CONFIG_PATH = home + '/zulip/api/bots/foursquare/foursquare.config'
def get_api_key():
# foursquare.config must have been moved from
# ~/zulip/api/bots/foursquare/foursquare.config into
# ~/foursquare.config for program to work
# see readme.md for more information
with open(CONFIG_PATH) as settings:
config = cp.ConfigParser()
config.readfp(settings)
return config.get('Foursquare', 'api_key')
class FoursquareHandler(object): class FoursquareHandler(object):
def __init__(self): def initialize(self, bot_handler):
self.api_key = get_api_key() self.api_key = bot_handler.get_config_info('foursquare', 'Foursquare')['api_key']
def usage(self): def usage(self):
return ''' return '''
@ -112,12 +97,3 @@ Example Inputs:
return return
handler_class = FoursquareHandler handler_class = FoursquareHandler
def test_get_api_key():
# must change to your own api key for test to work
result = get_api_key()
assert result == 'abcdefghijksm'
if __name__ == '__main__':
test_get_api_key()
print('Success')