Merge pull request Pennyw0rth#658 from nikaiw/fix_host_duplicate

Update database.py

Author: NeffIsBack

nxc/protocols/smb/database.py

@@ -39,7 +39,7 @@ def db_schema(db_conn):
         db_conn.execute(
             """CREATE TABLE "hosts" (
             "id" integer PRIMARY KEY,
-            "ip" text,
+            "ip" text UNIQUE,
             "hostname" text,
             "domain" text,
             "os" text,
@@ -257,7 +257,7 @@ def add_host(
         # TODO: find a way to abstract this away to a single Upsert call
         q = Insert(self.HostsTable)  # .returning(self.HostsTable.c.id)
         update_columns = {col.name: col for col in q.excluded if col.name not in "id"}
-        q = q.on_conflict_do_update(index_elements=self.HostsTable.primary_key, set_=update_columns)
+        q = q.on_conflict_do_update(index_elements=["ip"], set_=update_columns)
 
         self.db_execute(q, hosts)  # .scalar()
         # we only return updated IDs for now - when RETURNING clause is allowed we can return inserted