7fc1ff5e0e
Add file_uploader bot as an example of using ExternalBotHandler's methods for uploading files to Zulip server.
95 lines
2.8 KiB
Python
95 lines
2.8 KiB
Python
import configparser
|
|
import sys
|
|
|
|
from zulip_bots.lib import BotIdentity
|
|
from uuid import uuid4
|
|
|
|
class SimpleStorage:
|
|
def __init__(self):
|
|
self.data = dict()
|
|
|
|
def contains(self, key):
|
|
return (key in self.data)
|
|
|
|
def put(self, key, value):
|
|
self.data[key] = value
|
|
|
|
def get(self, key):
|
|
return self.data[key]
|
|
|
|
class SimpleMessageServer:
|
|
# This class is needed for the incrementor bot, which
|
|
# actually updates messages!
|
|
def __init__(self):
|
|
self.message_id = 0
|
|
self.messages = dict()
|
|
|
|
def send(self, message):
|
|
self.message_id += 1
|
|
message['id'] = self.message_id
|
|
self.messages[self.message_id] = message
|
|
return message
|
|
|
|
def update(self, message):
|
|
self.messages[message['message_id']] = message
|
|
|
|
def upload_file(self, file):
|
|
return dict(result='success', msg='', uri='https://server/user_uploads/{}'.format(uuid4()))
|
|
|
|
class TerminalBotHandler:
|
|
def __init__(self, bot_config_file):
|
|
self.bot_config_file = bot_config_file
|
|
self.storage = SimpleStorage()
|
|
self.message_server = SimpleMessageServer()
|
|
|
|
def identity(self):
|
|
return BotIdentity("bot name", "bot-email@domain")
|
|
|
|
def send_message(self, message):
|
|
if message['type'] == 'stream':
|
|
print('''
|
|
stream: {} topic: {}
|
|
{}
|
|
'''.format(message['to'], message['subject'], message['content']))
|
|
else:
|
|
print('''
|
|
PM response:
|
|
{}
|
|
'''.format(message['content']))
|
|
return self.message_server.send(message)
|
|
|
|
def send_reply(self, message, response):
|
|
print("\nReply from the bot is printed between the dotted lines:\n-------\n{}\n-------".format(response))
|
|
response_message = dict(
|
|
content=response
|
|
)
|
|
return self.message_server.send(response_message)
|
|
|
|
def update_message(self, message):
|
|
self.message_server.update(message)
|
|
print('''
|
|
update to message #{}:
|
|
{}
|
|
'''.format(message['message_id'], message['content']))
|
|
|
|
def upload_file_from_path(self, file_path):
|
|
with open(file_path) as file:
|
|
return self.upload_file(file)
|
|
|
|
def upload_file(self, file):
|
|
return self.message_server.upload_file(file)
|
|
|
|
def get_config_info(self, bot_name, optional=False):
|
|
if self.bot_config_file is None:
|
|
if optional:
|
|
return dict()
|
|
else:
|
|
print('Please supply --bot-config-file argument.')
|
|
sys.exit(1)
|
|
|
|
config = configparser.ConfigParser()
|
|
with open(self.bot_config_file) as conf:
|
|
config.readfp(conf) # type: ignore # readfp->read_file in python 3, so not in stubs
|
|
|
|
return dict(config.items(bot_name))
|