add member functions to SubSlotData

chia-protocol/src/weight_proof.rs

@@ -20,6 +20,20 @@ streamable_struct! (SubSlotData {
     total_iters: Option<u128>,
 });
 
+#[cfg(feature = "py-bindings")]
+use pyo3::prelude::*;
+
+#[cfg_attr(feature = "py-bindings", pymethods)]
+impl SubSlotData {
+    pub fn is_end_of_slot(&self) -> bool {
+        self.cc_slot_end_info.is_some()
+    }
+
+    pub fn is_challenge(&self) -> bool {
+        self.proof_of_space.is_some()
+    }
+}
+
 streamable_struct! (SubEpochChallengeSegment {
     sub_epoch_n: u32,
     sub_slots: Vec<SubSlotData>,

wheel/chia_rs.pyi

@@ -2106,6 +2106,8 @@ class SubSlotData:
     cc_ip_vdf_info: Optional[VDFInfo]
     icc_ip_vdf_info: Optional[VDFInfo]
     total_iters: Optional[int]
+    def is_end_of_slot(self) -> bool: ...
+    def is_challenge(self) -> bool: ...
     def __init__(
         self,
         proof_of_space: Optional[ProofOfSpace],

wheel/generate_type_stubs.py

@@ -167,6 +167,10 @@ def parse_rust_source(filename: str) -> List[Tuple[str, List[str]]]:
     "RewardChainBlock": [
         "def get_unfinished(self) -> RewardChainBlockUnfinished: ...",
     ],
+    "SubSlotData": [
+        "def is_end_of_slot(self) -> bool: ...",
+        "def is_challenge(self) -> bool: ...",
+    ],
 }
 
 classes = []