Use chia_rs

pool/absorb_spend.py

@@ -1,6 +1,6 @@
 import logging
 from typing import Any, Dict, List, Optional
-from blspy import AugSchemeMPL, PrivateKey
+from chia_rs import AugSchemeMPL, G2Element, PrivateKey
 
 from chia.consensus.block_rewards import calculate_pool_reward
 from chia.consensus.constants import ConsensusConstants
@@ -22,7 +22,6 @@
     puzzle_for_pk,
 )
 from chia.wallet.util.tx_config import DEFAULT_TX_CONFIG
-from blspy import G2Element
 from chia.wallet.wallet import Wallet
 from chia.wallet.wallet_info import WalletInfo
 

pool/pool.py

@@ -14,7 +14,7 @@
 from decimal import Decimal as D
 from typing import Dict, Optional, Set, List, Tuple
 
-from blspy import AugSchemeMPL, G1Element
+from chia_rs import AugSchemeMPL, G1Element
 from chia.pools.pool_wallet_info import PoolState, PoolSingletonState
 from chia.protocols.pool_protocol import (
     PoolErrorCode,

pool/pool_server.py

@@ -11,7 +11,7 @@
 
 import aiohttp
 import yaml
-from blspy import AugSchemeMPL, G2Element
+from chia_rs import AugSchemeMPL, G2Element
 from aiohttp import web
 from chia.protocols.pool_protocol import (
     PoolErrorCode,

pool/singleton.py

@@ -1,7 +1,7 @@
 from typing import Dict, List, Optional, Tuple
 import logging
 
-from blspy import G2Element
+from chia_rs import G2Element
 
 from chia.consensus.block_record import BlockRecord
 from chia.consensus.coinbase import pool_parent_id

pool/store/pgsql_store.py

@@ -6,7 +6,7 @@
 from typing import Optional, Set, List, Tuple, Dict
 
 import aiopg
-from blspy import G1Element
+from chia_rs import G1Element
 from chia.pools.pool_wallet_info import PoolState
 from chia.protocols.pool_protocol import PostPartialPayload, PostPartialRequest
 from chia.types.blockchain_format.coin import Coin

pool/store/sqlite_store.py

@@ -2,7 +2,7 @@
 from typing import Optional, Set, List, Tuple, Dict
 
 import aiosqlite
-from blspy import G1Element
+from chia_rs import G1Element
 from chia.pools.pool_wallet_info import PoolState
 from chia.types.blockchain_format.sized_bytes import bytes32
 from chia.types.coin_spend import CoinSpend

requirements.txt

@@ -1,6 +1,6 @@
 git+https://github.com/openchia/chia-blockchain.git@stable/2.1#egg=chia-blockchain
 git+https://github.com/olucurious/[email protected]
-blspy~=2.0.2
+chia_rs
 setuptools>=65.5.1
 uvloop~=0.19.0
 aiopg==1.4.0