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

zulip_bots: Add feature create-plan to Baremetrics bot.

Browse source
This commit is contained in:
Viraat Chandra 2018-01-01 14:58:45 +06:00 committed by showell
parent 68ec3a5ef0
commit 94f82dfe4c
1 changed files with 30 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
zulip_bots/zulip_bots/bots/baremetrics/baremetrics.py
View file

@ -1,6 +1,6 @@
# See readme.md for instructions on running this code. # See readme.md for instructions on running this code.
from typing import Any from typing import Any, List
import requests import requests
import logging import logging
@ -14,12 +14,14 @@ class BaremetricsHandler(object):
} }
self.commands = ['help', 'list-commands', 'account-info', 'list-sources', 'list-plans <source_id>', self.commands = ['help', 'list-commands', 'account-info', 'list-sources', 'list-plans <source_id>',
'list-customers <source_id>', 'list-customers <source_id>', 'list-subscriptions <source_id>', 'create-plan <source_id> '
'list-subscriptions <source_id>'] '<oid> <name> <currency> '
'<amount> <interval> '
'<interval_count>']
self.descriptions = ['Display bot info', 'Display the list of available commands', 'Display the account info', self.descriptions = ['Display bot info', 'Display the list of available commands', 'Display the account info',
'List the sources', 'List the plans for the source', 'List the customers in the source', 'List the sources', 'List the plans for the source', 'List the customers in the source',
'List the subscriptions in the source'] 'List the subscriptions in the source', 'Create a plan in the given source']
self.check_api_key(bot_handler) self.check_api_key(bot_handler)
@ -84,6 +86,12 @@ class BaremetricsHandler(object):
if part_commands[0].lower() == 'list-subscriptions': if part_commands[0].lower() == 'list-subscriptions':
return self.get_subscriptions(part_commands[1]) return self.get_subscriptions(part_commands[1])
if part_commands[0].lower() == 'create-plan':
if len(part_commands) == 8:
return self.create_plan(part_commands[1:])
else:
return 'Invalid number of arguments.'
except IndexError: except IndexError:
return 'Missing Params.' return 'Missing Params.'
except KeyError: except KeyError:
@ -183,4 +191,22 @@ class BaremetricsHandler(object):
return response return response
def create_plan(self, parameters: List[str]) -> str:
data_header = {
'oid': parameters[1],
'name': parameters[2],
'currency': parameters[3],
'amount': int(parameters[4]),
'interval': parameters[5],
'interval_count': int(parameters[6])
}
url = 'https://api.baremetrics.com/v1/{}/plans'.format(parameters[0])
create_plan_response = requests.post(url, data=data_header, headers=self.auth_header)
if 'error' not in create_plan_response.json():
return 'Plan Created.'
else:
return 'Some Error Occurred. Maybe the passed arguments are invalid.'
handler_class = BaremetricsHandler handler_class = BaremetricsHandler