From 1f07bbcf97d5caee221cf888cb402661aae11230 Mon Sep 17 00:00:00 2001 From: Marshall Hallenbeck Date: Sat, 4 Mar 2023 22:13:11 -0500 Subject: [PATCH] feat(mssqldb): update remove_credentials to use sqlalchemy --- cme/protocols/mssql/database.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/cme/protocols/mssql/database.py b/cme/protocols/mssql/database.py index a54b9a75..3d3b67c0 100755 --- a/cme/protocols/mssql/database.py +++ b/cme/protocols/mssql/database.py @@ -185,14 +185,16 @@ class database: return user_rowid - def remove_credentials(self, credIDs): + def remove_credentials(self, creds_id): """ Removes a credential ID from the database """ - for credID in credIDs: - cur = self.conn.cursor() - cur.execute("DELETE FROM users WHERE id=?", [credID]) - cur.close() + for cred_id in creds_id: + self.conn.query(self.users_table).filter( + self.users_table.c.id == cred_id + ).delete() + self.conn.commit() + self.conn.close() def add_admin_user(self, credtype, domain, username, password, host): cur = self.conn.cursor()