remote/client: ssh command: first look for SSHDriver

This makes it possible to use bindings on SSHDriver from the
environment file.

Signed-off-by: Nicolas Labriet <[email protected]>
(cherry picked from commit 438211f)

Author: nlabriet

labgrid/remote/client.py

@@ -1024,17 +1024,21 @@ def _get_ssh(self):
         place = self.get_acquired_place()
         target = self._get_target(place)
 
-        from ..resource import NetworkService
         try:
-            resource = target.get_resource(NetworkService, name=self.args.name)
-        except NoResourceFoundError:
-            ip = self._get_ip(place)
-            if not ip:
-                return
-            resource = NetworkService(target, address=str(ip), username='root')
-
-        drv = self._get_driver_or_new(target, "SSHDriver", name=resource.name)
-        return drv
+            drv = target.get_driver("SSHDriver", name=self.args.name)
+            return drv
+        except NoDriverFoundError:
+            from ..resource import NetworkService
+            try:
+                resource = target.get_resource(NetworkService, name=self.args.name)
+            except NoResourceFoundError:
+                ip = self._get_ip(place)
+                if not ip:
+                    return
+                resource = NetworkService(target, address=str(ip), username='root')
+
+            drv = self._get_driver_or_new(target, "SSHDriver", name=resource.name)
+            return drv
 
     def ssh(self):
         drv = self._get_ssh()