Merge pull request #62 from bugout-dev/mint-badge

Mint badge

scripts/mint_badge.py

@@ -0,0 +1,77 @@
+"""
+Mint badges to addresses.
+
+Caller specifies:
+1. Blockchain
+2. Terminus contract address
+3. Badge pool id
+4. Transaction parameters (gas, confirmations, etc.)
+5. Addresses to mint badges to
+
+The script checks if the addresses already have badges. It filters *out* the addresses that already
+have the badge in question. It mints a single badge to all addresses that do *not* have the badge.
+"""
+
+import argparse
+
+from brownie import network
+from dao import TerminusFacet
+
+parser = argparse.ArgumentParser(description="Mint Terminus badges to addresses")
+TerminusFacet.add_default_arguments(parser, True)
+parser.add_argument(
+    "--pool-id",
+    required=True,
+    type=int,
+    help="ID of Terminus pool representing the badge",
+)
+parser.add_argument(
+    "--recipients", nargs="+", help="Addresses that badge should be minted to"
+)
+parser.add_argument(
+    "-y",
+    "--yes",
+    action="store_true",
+    help="Set this flag to signal y on all confirmation prompts",
+)
+
+args = parser.parse_args()
+
+if args.address is None:
+    raise ValueError(
+        "Please specify the address of a Terminus contract using the --address argument."
+    )
+
+if len(args.recipients) > 200:
+    raise ValueError("This script can process at most 200 recipients at a time.")
+
+network.connect(args.network)
+
+recipients = list(set(args.recipients))
+
+terminus = TerminusFacet.TerminusFacet(args.address)
+pool_uri = terminus.uri(args.pool_id)
+balances = zip(
+    recipients,
+    terminus.balance_of_batch(recipients, [args.pool_id for _ in recipients]),
+)
+
+valid_recipients = [recipient for recipient, balance in balances if balance == 0]
+
+print(
+    f"Badge information -- Terminus address: {args.address}, pool ID: {args.pool_id}, pool URI: {pool_uri}"
+)
+print("\n- ".join(["Intended recipients:"] + valid_recipients))
+
+if not args.yes:
+    permission_check = input("Proceed? (y/N)")
+    if permission_check.strip().lower() != "y":
+        raise Exception("You did not wish to proceed")
+
+amounts = [1 for _ in valid_recipients]
+transaction_config = TerminusFacet.get_transaction_config(args)
+transaction_info = terminus.pool_mint_batch(
+    args.pool_id, valid_recipients, amounts, transaction_config
+)
+
+print(transaction_info)