From 43821cbd1bb6eb76a898e4115b8da15cc5b61f59 Mon Sep 17 00:00:00 2001 From: Luke Murphy Date: Sun, 10 Jan 2021 15:45:12 +0100 Subject: [PATCH] Gather nick and room --- xbotlib.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/xbotlib.py b/xbotlib.py index d4088cc..607d3f5 100644 --- a/xbotlib.py +++ b/xbotlib.py @@ -49,18 +49,21 @@ class Bot(ClientXMPP): def generate_config(self): """Generate bot configuration.""" - jid = ( - input("XMPP address of your bot (e.g. alice@myserver.com): ") - or "alice@myserver.com" - ) + jid = input("XMPP address (e.g. foo@bar.com): ") or "foo@bar.com" password = ( - getpass("Password for the bot account (e.g. my-cool-password): ") - or "my-cool-password" + getpass("Password (e.g. my-cool-password): ") or "my-cool-password" ) + room = input("XMPP room (e.g. foo@bar.com): ") + nick = input("Nickname (e.g. lurkbot): ") config = ConfigParser() config["bot"] = {"jid": jid, "password": password} + if room: + config["bot"]["room"] = room + if nick: + config["bot"]["nick"] = nick + with open("bot.conf", "w") as file_handle: config.write(file_handle)