[QI2-1135] Made default_host configurable

quantuminspire/cli/command_list.py

@@ -335,6 +335,20 @@ def login(
     typer.echo(f"Using member ID {settings.auths[host].team_member_id}")
 
 
+@app.command("set-default-host")
+def set_default_host(
+    host: str = typer.Argument(
+        help="The URL of the platform to which to connect"
+    ),
+) -> None:
+    """Set default_host for interacting with Quantum Inspire.
+    """
+    settings = Settings()
+    settings.default_host = Url(host)
+    settings.write_settings_to_file()
+    typer.echo(f"Default host set to {host}")
+
+
 @app.command("logout")
 def logout(
     host: str = typer.Argument(

quantuminspire/util/configuration.py

@@ -162,11 +162,16 @@ def store_tokens(self, host: Url, tokens: TokenInfo) -> None:
         self.auths[host].tokens = tokens
         member_id = self.get_team_member_id(host=host, access_token=tokens.access_token)
         self.auths[host].team_member_id = member_id
+        self.default_host = host
+        self.write_settings_to_file()
+
+    def write_settings_to_file(self) -> None:
         assert isinstance(self.model_config["json_file"], PathLike)
         Path(self.model_config["json_file"]).write_text(
             self.model_dump_json(indent=2), encoding=self.model_config.get("env_file_encoding")
         )
 
+
     @staticmethod
     async def _fetch_team_member_id(host: str, access_token: str) -> int:
         config = Configuration(host=host, access_token=access_token)

tests/cli/test_command_list.py

@@ -127,3 +127,12 @@ def test_login(mocker: MockerFixture, mocked_config_file: MagicMock) -> None:
     device_session.poll_for_tokens.assert_called_once()
     store_tokens.assert_called_once()
     assert "Login successful!" in result.stdout
+
+
+def test_set_default_host(mocker: MockerFixture, mocked_config_file: MagicMock) -> None:
+    host = "https://example.com"
+    write_settings_to_file = mocker.patch("quantuminspire.cli.command_list.Settings.write_settings_to_file")
+    result = runner.invoke(app, ["set-default-host", host])
+    assert result.exit_code == 0, repr(result.exception)
+    write_settings_to_file.assert_called_once()
+    assert f"Default host set to {host}" in result.stdout