bots: Update foursquare bot to use get_config_info().
This commit is contained in:
parent
c4a85b2f74
commit
a10ee59853
|
@ -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')
|
|
||||||
|
|
Loading…
Reference in a new issue