From 8fc7a7652ab743b9c808939941b031ec419d799c Mon Sep 17 00:00:00 2001 From: jesopo Date: Tue, 13 Nov 2018 11:13:04 +0000 Subject: [PATCH] Return new server_id from Database.servers.add --- src/Database.py | 2 ++ src/utils/cli.py | 5 +++-- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/src/Database.py b/src/Database.py index b54eea0d..2dcd6471 100644 --- a/src/Database.py +++ b/src/Database.py @@ -17,6 +17,8 @@ class Servers(Table): ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)""", [alias, hostname, port, password, tls, ipv4, bindhost, nickname, username, realname]) + return self.database.execute_fetchone( + "SELECT server_id FROM servers ORDER BY server_id DESC LIMIT 1")[0] def get_all(self): return self.database.execute_fetchall( "SELECT server_id, alias FROM servers") diff --git a/src/utils/cli.py b/src/utils/cli.py index 98d6be92..a7ab33c9 100644 --- a/src/utils/cli.py +++ b/src/utils/cli.py @@ -15,5 +15,6 @@ def add_server(database: Database.Database): username = input("username: ") realname = input("realname: ") bindhost = input("bindhost?: ") - database.servers.add(alias, hostname, port, password, ipv4, tls, bindhost, nickname, username, realname) - + admin = input("admin nickname: ") + id = database.servers.add(alias, hostname, port, password, ipv4, tls, + bindhost, nickname, username, realname)