Skip to content

Module pyastrobee.utils.python_utils

Utility functions broadly related to python as a whole

View Source
"""Utility functions broadly related to python as a whole"""

from typing import Any

def print_red(message: Any):

    """Helper function for printing in red text

    Args:

        message (Any): The message to print out in red

    """

    print(f"\033[31m{message}\033[0m")

def print_green(message: Any):

    """Helper function for printing in green text

    Args:

        message (Any): The message to print out in green

    """

    print(f"\033[32m{message}\033[0m")

def flatten(l: list[list]) -> list:

    """Flatten a list of lists into a single list

    Args:

        l (list[list]): List of lists to flaten

    Returns:

        list: Flattened list

    """

    return [item for sublist in l for item in sublist]

Functions

flatten

def flatten(
    l: list[list]
) -> list

Flatten a list of lists into a single list

Parameters:

Name Type Description Default
l list[list] List of lists to flaten None

Returns:

Type Description
list Flattened list
View Source
def flatten(l: list[list]) -> list:

    """Flatten a list of lists into a single list

    Args:

        l (list[list]): List of lists to flaten

    Returns:

        list: Flattened list

    """

    return [item for sublist in l for item in sublist]
def print_green(
    message: Any
)

Helper function for printing in green text

Parameters:

Name Type Description Default
message Any The message to print out in green None
View Source
def print_green(message: Any):

    """Helper function for printing in green text

    Args:

        message (Any): The message to print out in green

    """

    print(f"\033[32m{message}\033[0m")
def print_red(
    message: Any
)

Helper function for printing in red text

Parameters:

Name Type Description Default
message Any The message to print out in red None
View Source
def print_red(message: Any):

    """Helper function for printing in red text

    Args:

        message (Any): The message to print out in red

    """

    print(f"\033[31m{message}\033[0m")