Skip to content

context

temporary_class_attr(cls, attr, value)

Context method to temporarily assign a value to a class attribute.

The assigned value will only be held within the context.

Parameters:

Name Type Description Default
cls type[Any]

Class of which the class attribute value is to be assigned.

 required
attr str

Name of class attribute.

 required
value Any

Value to assign to class attribute (must be correct type).

 required
Source code in opensquirrel/utils/context.py 
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
@contextmanager
def temporary_class_attr(cls: type[Any], attr: str, value: Any) -> Generator[None, None, None]:
    """Context method to temporarily assign a value to a class attribute.

    The assigned value will only be held within the context.

    Args:
        cls: Class of which the class attribute value is to be assigned.
        attr: Name of class attribute.
        value: Value to assign to class attribute (must be correct type).
    """
    original_value = getattr(cls, attr)
    setattr(cls, attr, value)
    try:
        yield
    finally:
        setattr(cls, attr, original_value)