.

Author: EdwardJXLi

src/codegen/sdk/core/class_definition.py

@@ -110,19 +110,29 @@ def parent_class_names(self) -> list[Name | ChainedAttribute]:
         return []
 
     @reader
-    def get_parent_class(self, parent_class_name: str) -> Editable | None:
+    def get_parent_class(self, parent_class_name: str, optional: bool = False) -> Editable | None:
         """Returns the parent class node with the specified name.
 
         Retrieves a parent class Name or ChainedAttribute node from this class's list of parent class names that matches
         the specified name.
 
         Args:
             parent_class_name (str): The name of the parent class to find.
+            optional (bool, optional): Whether to return None if the parent class is not found. Defaults to False.
 
         Returns:
             Editable | None: The matching parent class node, or None if no match is found.
         """
-        return next((p for p in self.parent_class_names if p.source == parent_class_name), None)
+        parent_class = [p for p in self.parent_class_names if p.source == parent_class_name]
+        if not parent_class:
+            if not optional:
+                msg = f"Parent class {parent_class_name} not found in class {self.name}. Use optional=True to return None instead."
+                raise ValueError(msg)
+            return None
+        if len(parent_class) > 1:
+            msg = f"Multiple parent classes found with name {parent_class_name} in class {self.name}."
+            raise ValueError(msg)
+        return parent_class[0]
 
     @property
     @reader
@@ -233,30 +243,35 @@ def methods(self, *, max_depth: int | None = 0, private: bool = True, magic: boo
         return list(result.values())
 
     @reader
-    def get_nested_class(self, name: str) -> Self | None:
+    def get_nested_class(self, name: str, optional: bool = False) -> Self | None:
         """Returns a nested class by name from the current class.
 
         Searches through the nested classes defined in the class and returns the first one that matches the given name.
 
         Args:
             name (str): The name of the nested class to find.
+            optional (bool, optional): Whether to return None if the nested class is not found. Defaults to False.
 
         Returns:
             Self | None: The nested class if found, None otherwise.
         """
         for m in self.nested_classes:
             if m.name == name:
                 return m
+        if not optional:
+            msg = f"Nested class {name} not found in class {self.name}. Use optional=True to return None instead."
+            raise ValueError(msg)
         return None
 
     @reader
-    def get_method(self, name: str) -> TFunction | None:
+    def get_method(self, name: str, optional: bool = False) -> TFunction | None:
         """Returns a specific method by name from the class or any of its superclasses.
 
         Searches through the class's methods and its superclasses' methods to find a method with the specified name.
 
         Args:
             name (str): The name of the method to find.
+            optional (bool, optional): Whether to return None if the method is not found. Defaults to False.
 
         Returns:
             TFunction | None: The method if found, None otherwise.
@@ -267,6 +282,9 @@ def get_method(self, name: str) -> TFunction | None:
                 for m in c.methods:
                     if m.name == name:
                         return m
+        if not optional:
+            msg = f"Method {name} not found in class {self.name}. Use optional=True to return None instead."
+            raise ValueError(msg)
         return None
 
     @proxy_property
@@ -293,13 +311,14 @@ def attributes(self, *, max_depth: int | None = 0, private: bool = True) -> list
         return list(result.values())
 
     @reader
-    def get_attribute(self, name: str) -> Attribute | None:
+    def get_attribute(self, name: str, optional: bool = False) -> Attribute | None:
         """Returns a specific attribute by name.
 
         Searches for an attribute with the given name in the current class and its superclasses.
 
         Args:
             name (str): The name of the attribute to search for.
+            optional (bool, optional): Whether to return None if the attribute is not found. Defaults to False.
 
         Returns:
             Attribute | None: The matching attribute if found, None otherwise. If multiple attributes with the same name exist in the inheritance hierarchy, returns the first one found.
@@ -310,6 +329,9 @@ def get_attribute(self, name: str) -> Attribute | None:
                 for m in c.code_block.get_attributes(name):
                     if m.name == name:
                         return m
+        if not optional:
+            msg = f"Attribute {name} not found in class {self.name}. Use optional=True to return None instead."
+            raise ValueError(msg)
         return None
 
     ####################################################################################################################

src/codegen/sdk/typescript/interfaces/has_block.py

@@ -87,21 +87,25 @@ def jsx_elements(self) -> list[JSXElement[Self]]:
         return jsx_elements
 
     @reader
-    def get_component(self, component_name: str) -> JSXElement[Self] | None:
+    def get_component(self, component_name: str, optional: bool = False) -> JSXElement[Self] | None:
         """Returns a specific JSX element from within this symbol's JSX elements.
 
         Searches through all JSX elements in this symbol's code block and returns the first one that matches
         the given component name.
 
         Args:
             component_name (str): The name of the JSX component to find.
+            optional (bool, optional): If True, return None if the component is not found. Defaults to False.
 
         Returns:
             JSXElement[Self] | None: The matching JSX element if found, None otherwise.
         """
         for component in self.jsx_elements:
             if component.name == component_name:
                 return component
+        if not optional:
+            msg = f"Component {component_name} not found in symbol {self.name} of file {self.file_path}. Use optional=True to return None instead."
+            raise ValueError(msg)
         return None
 
     @cached_property