Merge pull request #173 from saratomaz/update_get_stake_vkey_hash

update get_stake_vkey_hash to accept stake vkey as input
input-output-hk · Apr 4, 2023 · ed33f5f · ed33f5f
2 parents 4f93fea + 8bcf8f7
commit ed33f5f
Showing 1 changed file with 12 additions and 5 deletions.
diff --git a/cardano_clusterlib/stake_address_group.py b/cardano_clusterlib/stake_address_group.py
@@ -261,20 +261,27 @@ def gen_stake_addr_and_keys(
 
     def get_stake_vkey_hash(
         self,
-        stake_vkey_file: FileType,
+        stake_vkey_file: Optional[FileType] = None,
+        stake_vkey: Optional[str] = None,
     ) -> str:
         """Return the hash of a stake address key.
 
         Args:
-            stake_vkey_file: A path to stake vkey file.
+            stake_vkey_file: A path to stake vkey file (optional).
+            stake_vkey: A stake vkey (Bech32, optional).
 
         Returns:
             str: A generated hash.
         """
+        if stake_vkey:
+            cli_args = ["--stake-verification-key", stake_vkey]
+        elif stake_vkey_file:
+            cli_args = ["--stake-verification-key-file", str(stake_vkey_file)]
+        else:
+            raise AssertionError("Either `stake_vkey` or `stake_vkey_file` is needed.")
+
         return (
-            self._clusterlib_obj.cli(
-                ["stake-address", "key-hash", "--stake-verification-key-file", str(stake_vkey_file)]
-            )
+            self._clusterlib_obj.cli(["stake-address", "key-hash", *cli_args])
             .stdout.rstrip()
             .decode("ascii")
         )