generic_fn.py

class GenericFn:
    """
    Represents a generic function with its respective relevant attributes.

    Attributes:
        annotation (str): The inline annotation (optional; may be the empty string).
        fn_name (str): The name of the function.
        params (list[str]): A string representing the generic parameters enclosed in angle brackets.
        args (str): A string representing the generic arguments provided when calling the function.
        fn_body (str): The body of the generic function. May contain calls to other generic ~
                       functions. Also contains the return type
    """

    def __init__(
        self,
        annotation: str,
        fn_name: str,
        params: str,
        args: str,
        fn_body: str,
    ):
        self.annotation = annotation
        self.fn_name = fn_name
        self.params = [p.strip() for p in params.split(",")]
        self.args = args
        self.fn_body = fn_body

    def __repr__(self) -> str:
        return (
            f"Annotation: {self.annotation}\n"
            f"Function Name: {self.fn_name}\n"
            f"Parameters: {self.params}\n"
            f"Arguments: {self.args}\n"
            f"Function Body:\n{self.fn_body}\n\n"
        )