[WIP] Add get_parents method to subtensor

Author: thewhaleking

bittensor/core/async_subtensor.py

@@ -973,6 +973,51 @@ async def get_block_hash(self, block: Optional[int] = None) -> str:
         else:
             return await self.substrate.get_chain_head()
 
+    async def get_parents(
+        self,
+        hotkey: str,
+        netuid: int,
+        block: Optional[int] = None,
+        block_hash: Optional[str] = None,
+        reuse_block: bool = False,
+    ) -> tuple[bool, list[tuple[float, str]], str]:
+        """
+        This method retrieves the parent of a given hotkey and netuid. It queries the SubtensorModule's ParentKeys
+            storage function to get the children and formats them before returning as a tuple.
+
+        Arguments:
+            hotkey (str): The child hotkey SS58.
+            netuid (int): The netuid value.
+            block (Optional[int]): The block number for which the children are to be retrieved.
+            block_hash (Optional[str]): The hash of the block to retrieve the subnet unique identifiers from.
+            reuse_block (bool): Whether to reuse the last-used block hash.
+
+        Returns:
+            A tuple containing a boolean indicating success or failure, a list of formatted
+                parents [(proportion, parent)], and an error message (if applicable)
+        """
+        block_hash = await self.determine_block_hash(block, block_hash, reuse_block)
+        try:
+            parents = await self.substrate.query(
+                module="SubtensorModule",
+                storage_function="ParentKeys",
+                params=[hotkey, netuid],
+                block_hash=block_hash,
+                reuse_block_hash=reuse_block,
+            )
+            if parents:
+                formatted_parents = []
+                for proportion, parent in parents.value:
+                    # Convert U64 to int
+                    formatted_child = decode_account_id(parent[0])
+                    normalized_proportion = u64_normalized_float(proportion)
+                    formatted_parents.append((normalized_proportion, formatted_child))
+                return True, formatted_parents, ""
+            else:
+                return True, [], ""
+        except SubstrateRequestException as e:
+            return False, [], format_error_message(e)
+
     async def get_children(
         self,
         hotkey: str,

bittensor/core/subtensor.py

@@ -751,6 +751,42 @@ def get_hyperparameter(
 
         return getattr(result, "value", result)
 
+    def get_parents(
+        self, hotkey: str, netuid: int, block: Optional[int] = None
+    ) -> tuple[bool, list[tuple[float, str]], str]:
+        """
+        This method retrieves the parent of a given hotkey and netuid. It queries the SubtensorModule's ParentKeys
+            storage function to get the children and formats them before returning as a tuple.
+
+        Arguments:
+            hotkey (str): The child hotkey SS58.
+            netuid (int): The netuid.
+            block (Optional[int]): The block number for which the children are to be retrieved.
+
+        Returns:
+            A tuple containing a boolean indicating success or failure, a list of formatted
+                parents [(proportion, parent)], and an error message (if applicable)
+        """
+        try:
+            parents = self.substrate.query(
+                module="SubtensorModule",
+                storage_function="ParentKeys",
+                params=[hotkey, netuid],
+                block_hash=self.determine_block_hash(block),
+            )
+            if parents:
+                formatted_parents = []
+                for proportion, parent in parents.value:
+                    # Convert U64 to int
+                    formatted_child = decode_account_id(parent[0])
+                    normalized_proportion = u64_normalized_float(proportion)
+                    formatted_parents.append((normalized_proportion, formatted_child))
+                return True, formatted_parents, ""
+            else:
+                return True, [], ""
+        except SubstrateRequestException as e:
+            return False, [], format_error_message(e)
+
     def get_children(
         self, hotkey: str, netuid: int, block: Optional[int] = None
     ) -> tuple[bool, list[tuple[float, str]], str]:

bittensor/core/subtensor_api/subnets.py

@@ -14,6 +14,7 @@ def __init__(self, subtensor: Union["_Subtensor", "_AsyncSubtensor"]):
         self.bonds = subtensor.bonds
         self.difficulty = subtensor.difficulty
         self.get_all_subnets_info = subtensor.get_all_subnets_info
+        self.get_parents = subtensor.get_parents
         self.get_children = subtensor.get_children
         self.get_children_pending = subtensor.get_children_pending
         self.get_current_weight_commit_info = subtensor.get_current_weight_commit_info