remote/client: Move initial-state code into a function

Put this code into a function to reduce the size of the _get_target()
function.

Signed-off-by: Simon Glass <sjg@chromium.org>

Author: sjg20

labgrid/remote/client.py

@@ -875,6 +875,15 @@ def _prepare_manager(self):
         manager.session = self
         manager.loop = self.loop
 
+    def set_initial_state(self, target):
+        if self.args.state:
+            strategy = target.get_driver("Strategy")
+            if self.args.initial_state:
+                print(f"Setting initial state to {self.args.initial_state}")
+                strategy.force(self.args.initial_state)
+            logging.info("Transitioning into state %s", self.args.state)
+            strategy.transition(self.args.state)
+
     def _get_target(self, place):
         self._prepare_manager()
         target = None
@@ -885,13 +894,7 @@ def _get_target(self, place):
                     print(f"Selected role {self.role} from configuration file")
             target = self.env.get_target(self.role)
         if target:
-            if self.args.state:
-                strategy = target.get_driver("Strategy")
-                if self.args.initial_state:
-                    print(f"Setting initial state to {self.args.initial_state}")
-                    strategy.force(self.args.initial_state)
-                logging.info("Transitioning into state %s", self.args.state)
-                strategy.transition(self.args.state)
+            self.set_initial_state(target)
         else:
             target = Target(place.name, env=self.env)
             RemotePlace(target, name=place.name)