From dc102f258d75493b11735fdd69e25f92be186c1f Mon Sep 17 00:00:00 2001 From: jesopo Date: Wed, 24 Apr 2019 12:07:30 +0100 Subject: Add Database.servers.get_by_alias, move IRCBot.get_server to IRCBot.get_server_by_id, add IRCBot.get_server_by_alias and change !connect/!disconnect to take aliases instead of IDs --- modules/admin.py | 28 ++++++++++++---------------- 1 file changed, 12 insertions(+), 16 deletions(-) (limited to 'modules/admin.py') diff --git a/modules/admin.py b/modules/admin.py index c9096e54..8d72a12b 100644 --- a/modules/admin.py +++ b/modules/admin.py @@ -52,15 +52,12 @@ class Module(ModuleManager.BaseModule): :usage: :permission: connect """ - id = event["args_split"][0] - if not id.isdigit(): - raise utils.EventError("Please provide a numeric server ID") + alias = event["args"] + id = self.bot.database.servers.get_by_alias(alias) + if id == None: + raise utils.EventError("Unknown server alias") - id = int(id) - if not self.bot.database.servers.get(id): - raise utils.EventError("Unknown server ID") - - existing_server = self.bot.get_server(id) + existing_server = self.bot.get_server_by_id(id) if existing_server: raise utils.EventError("Already connected to %s" % str( existing_server)) @@ -77,14 +74,13 @@ class Module(ModuleManager.BaseModule): """ id = event["server"].id if event["args"]: - id = event["args_split"][0] - if not id.isdigit(): - raise utils.EventError("Please provide a numeric server ID") - - id = int(id) - if not self.bot.database.servers.get(id): - raise utils.EventError("Unknown server ID") - server = self.bot.get_server(id) + print("alias") + alias = event["args"] + id = self.bot.database.servers.get_by_alias(alias) + if id == None: + raise utils.EventError("Unknown server alias") + print(id) + server = self.bot.get_server_by_id(id) server.disconnect() self.bot.disconnect(server) -- cgit v1.3.1-10-gc9f91