From 89c12e845a89f84cd2b32d17b78f3798a36cef6d Mon Sep 17 00:00:00 2001
From: jesopo
Date: Fri, 12 Jul 2019 09:30:22 +0100
Subject: add --remove-server/-R cli arg

---
 src/Database.py | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'src')

diff --git a/src/Database.py b/src/Database.py
index 995fc643..d734eaf7 100644
--- a/src/Database.py
+++ b/src/Database.py
@@ -21,6 +21,10 @@ class Servers(Table):
             realname])
         return self.database.execute_fetchone(
             "SELECT server_id FROM servers ORDER BY server_id DESC LIMIT 1")[0]
+    def by_alias(self, alias: str) -> typing.Optional[int]:
+        ids = self.database.execute_fetchone(
+            "SELECT server_id FROM servers WHERE alias=?", [alias])
+        return ids[0] if ids else None
     def get_all(self):
         return self.database.execute_fetchall(
             "SELECT server_id, alias FROM servers")
@@ -45,6 +49,8 @@ class Servers(Table):
             raise ValueError("Unknown column on servers table '%s'" % column)
         self.database.execute(
             "UPDATE servers SET %s=? WHERE server_id=?" % column, [value, id])
+    def delete(self, id: int):
+        self.database.execute("DELETE FROM servers WHERE server_id=?", [id])
 
 class Channels(Table):
     def add(self, server_id: int, name: str):
-- 
cgit v1.3.1-10-gc9f91