Generate Identity from account

Author: tupui

src/soroban/models.py

@@ -1,8 +1,10 @@
+import pathlib
+
 from pydantic import model_validator, HttpUrl, BaseModel
 from pydantic_settings import BaseSettings, SettingsConfigDict
 from stellar_sdk import Keypair, Network
 
-__all__ = ["Identity"]
+__all__ = ["Identity", "NetworkConfig"]
 
 
 class Identity(BaseSettings):
@@ -23,6 +25,23 @@ def load_keys(self) -> "Identity":
         self.public_key = self.keypair.public_key
         return self
 
+    @classmethod
+    def from_source_account(
+        cls, account: Keypair | str | pathlib.Path | None = None
+    ) -> "Identity":
+        if account is None:
+            identity = Identity()
+        elif isinstance(account, (str, pathlib.Path)):
+            fname = (
+                account
+                if pathlib.Path(account).is_file()
+                else pathlib.Path(".soroban/identity") / account / ".toml"
+            )
+            identity = Identity(_env_file=fname)
+        else:
+            identity = Identity(keypair=account)
+        return identity
+
 
 class NetworkConfig(BaseSettings):
     soroban_rpc_url: HttpUrl = HttpUrl("https://soroban-testnet.stellar.org:443")
@@ -33,4 +52,4 @@ class NetworkConfig(BaseSettings):
 
 class SorobanConfig(BaseModel):
     network: NetworkConfig = NetworkConfig()
-    keys: list[Identity] = [Identity()]
+    keys: list[Identity] = None

tests/test_models.py

@@ -8,14 +8,21 @@
 class TestIdentity:
 
     def test_file(self):
-        alice_fname = pathlib.Path(__file__).parent
-        soroban.Identity(_env_file=alice_fname / "alice.toml")
+        alice_fname = pathlib.Path(__file__).parent / "alice.toml"
+        soroban.Identity(_env_file=alice_fname)
 
     def test_from_pk(self):
         keypair = Keypair.random()
         soroban.Identity(secret_key=keypair.secret)
         soroban.Identity(keypair=keypair)
 
+    def test_from_source_account(self):
+        alice_fname = pathlib.Path(__file__).parent / "alice.toml"
+        soroban.Identity.from_source_account(account=alice_fname)
+
+        keypair = Keypair.random()
+        soroban.Identity.from_source_account(account=keypair)
+
     def test_raises(self):
         with pytest.raises(ValueError, match="provide a secret key or a Keypair"):
             soroban.Identity()