investigator: Update for secret models

projectatomic · May 25, 2017 · 6c6bfd6 · 6c6bfd6
1 parent d84b652
commit 6c6bfd6
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/commissaire_service/investigator/__init__.py b/src/commissaire_service/investigator/__init__.py
@@ -17,7 +17,7 @@
 
 import commissaire.constants as C
 
-from commissaire.models import Cluster, Host, Network
+from commissaire.models import Cluster, Host, HostCreds, Network
 from commissaire.storage.client import StorageClient
 from commissaire.util.config import ConfigurationError
 from commissaire.util.date import formatted_dt
@@ -116,9 +116,10 @@ def on_investigate(self, message, address, cluster_data={}):
             self.logger.debug('Related cluster: {}'.format(cluster_data))
 
         host = self.storage.get_host(address)
+        host_creds = self.storage.get(HostCreds.new(address=host.address))
         transport = ansibleapi.Transport(host.remote_user)
 
-        key = TemporarySSHKey(host, self.logger)
+        key = TemporarySSHKey(host_creds, self.logger)
         try:
             key.create()
         except Exception as error: