Fix referrals networks (#35)

stakewise · May 23, 2022 · a9ada93 · a9ada93
1 parent b87d18c
commit a9ada93
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/stakewise_cli/commands/create_referrals_proposal.py b/stakewise_cli/commands/create_referrals_proposal.py
@@ -6,7 +6,14 @@
 from stakewise_cli.coingecko import get_average_range_price
 from stakewise_cli.eth1 import get_block_timestamp, get_referrals, get_web3_client
 from stakewise_cli.ipfs import upload_to_ipfs
-from stakewise_cli.networks import GNOSIS_CHAIN, GOERLI, MAINNET, NETWORKS, PERM_GOERLI
+from stakewise_cli.networks import (
+    GNOSIS_CHAIN,
+    GOERLI,
+    HARBOUR_GOERLI,
+    HARBOUR_MAINNET,
+    MAINNET,
+    NETWORKS,
+)
 from stakewise_cli.proposals import generate_referrals_swise_specification
 from stakewise_cli.queries import get_ethereum_gql_client, get_stakewise_gql_client
 
@@ -22,7 +29,8 @@
     help="The network to generate the referral proposal for",
     prompt="Enter the network name",
     type=click.Choice(
-        [MAINNET, GOERLI, PERM_GOERLI, GNOSIS_CHAIN], case_sensitive=False
+        [MAINNET, GOERLI, HARBOUR_MAINNET, HARBOUR_GOERLI, GNOSIS_CHAIN],
+        case_sensitive=False,
     ),
 )
 @click.option(