From 0d79392d93ddaa87aaa751ecfbe76ab3e51c930e Mon Sep 17 00:00:00 2001
From: Marshall Hallenbeck <marshall.hallenbeck@gmail.com>
Date: Mon, 6 Mar 2023 19:19:13 -0500
Subject: [PATCH] update variable name

---
 cme/protocols/smb/database.py | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/cme/protocols/smb/database.py b/cme/protocols/smb/database.py
index 12e25519..869c1266 100755
--- a/cme/protocols/smb/database.py
+++ b/cme/protocols/smb/database.py
@@ -209,7 +209,7 @@ class database:
         Check if this host has already been added to the database, if not add it in.
         """
         domain = domain.split('.')[0].upper()
-        update_hosts = []
+        hosts = []
 
         q = select(self.ComputersTable).filter(
             self.ComputersTable.c.ip == ip
@@ -232,7 +232,7 @@ class database:
 
         # create new computer
         if not results:
-            update_hosts = [host]
+            hosts = [host]
         # update existing hosts data
         else:
             for host in results:
@@ -258,13 +258,13 @@ class database:
                     computer_data["petitpotam"] = petitpotam
                 if dc is not None:
                     computer_data["dc"] = dc
-                update_hosts.append(computer_data)
+                hosts.append(computer_data)
 
-        logging.debug(f"Update Hosts: {update_hosts}")
+        logging.debug(f"Update Hosts: {hosts}")
         asyncio.run(
             self.conn.execute(
                 sqlite_upsert(self.ComputersTable),
-                update_hosts
+                hosts
             )
         )
         # asyncio.run(self.conn.close())
@@ -295,6 +295,7 @@ class database:
             credential_data["password"] = password
         if group_id is not None:
             credential_data["groupid"] = group_id
+            credential_data["groupid"] = group_id
         if pillaged_from is not None:
             credential_data["pillaged_from"] = pillaged_from
 
@@ -729,6 +730,9 @@ class database:
             q = select(self.GroupsTable).filter(
                 self.GroupsTable.c.id == filter_term
             )
+            results = asyncio.run(self.conn.execute(q)).first()
+            # all() returns a list, so we keep the return format the same so consumers don't have to guess
+            return [results]
         elif group_name and group_domain:
             q = select(self.GroupsTable).filter(
                 func.lower(self.GroupsTable.c.username) == func.lower(group_name),