Source code for ppc_robot_lib.reporting.transformation.add_client_id

import pandas

from ppc_robot_lib import tasks




def add_client_id(table: pandas.DataFrame, column: str) -> None:
    """
    Add a new column to the table that contains with Client ID from task context.

    **Example:**

        >>> from ppc_robot_lib.steps.transformations import AddClientIdStep
        >>> AddClientIdStep("table", "client_id")

    """
    task_ctx = tasks.get_context()
    table[column] = task_ctx.client_id