@@ -454,6 +454,7 @@ def get_payment_addrs(
454454 caching_key : str = "" ,
455455 amount : int | None = None ,
456456 min_amount : int | None = None ,
457+ key_gen_method : clusterlib_utils .KeyGenMethods = clusterlib_utils .KeyGenMethods .DIRECT ,
457458) -> list [clusterlib .AddressRecord ]:
458459 """Create new payment addresses."""
459460 if num < 1 :
@@ -464,6 +465,7 @@ def _create_addrs() -> list[clusterlib.AddressRecord]:
464465 addrs = clusterlib_utils .create_payment_addr_records (
465466 * [f"{ name_template } _fund_addr_{ i } " for i in range (1 , num + 1 )],
466467 cluster_obj = cluster_obj ,
468+ key_gen_method = key_gen_method ,
467469 )
468470 return addrs
469471
@@ -486,6 +488,7 @@ def get_payment_addr(
486488 caching_key : str = "" ,
487489 amount : int | None = None ,
488490 min_amount : int | None = None ,
491+ key_gen_method : clusterlib_utils .KeyGenMethods = clusterlib_utils .KeyGenMethods .DIRECT ,
489492) -> clusterlib .AddressRecord :
490493 """Create a single new payment address."""
491494 return get_payment_addrs (
@@ -496,6 +499,7 @@ def get_payment_addr(
496499 caching_key = caching_key ,
497500 amount = amount ,
498501 min_amount = min_amount ,
502+ key_gen_method = key_gen_method ,
499503 )[0 ]
500504
501505
@@ -508,6 +512,7 @@ def get_pool_users(
508512 caching_key : str = "" ,
509513 amount : int | None = None ,
510514 min_amount : int | None = None ,
515+ payment_key_gen_method : clusterlib_utils .KeyGenMethods = clusterlib_utils .KeyGenMethods .DIRECT ,
511516) -> list [clusterlib .PoolUser ]:
512517 """Create new pool users."""
513518 if num < 1 :
@@ -519,6 +524,7 @@ def _create_pool_users() -> list[clusterlib.PoolUser]:
519524 cluster_obj = cluster_obj ,
520525 name_template = f"{ name_template } _pool_user" ,
521526 no_of_addr = num ,
527+ payment_key_gen_method = payment_key_gen_method ,
522528 )
523529 return users
524530
@@ -541,6 +547,7 @@ def get_pool_user(
541547 caching_key : str = "" ,
542548 amount : int | None = None ,
543549 min_amount : int | None = None ,
550+ payment_key_gen_method : clusterlib_utils .KeyGenMethods = clusterlib_utils .KeyGenMethods .DIRECT ,
544551) -> clusterlib .PoolUser :
545552 """Create a single new pool user."""
546553 return get_pool_users (
@@ -551,6 +558,7 @@ def get_pool_user(
551558 caching_key = caching_key ,
552559 amount = amount ,
553560 min_amount = min_amount ,
561+ payment_key_gen_method = payment_key_gen_method ,
554562 )[0 ]
555563
556564
0 commit comments