forked from mirrors/gecko-dev
e4a6fe2194
Original Revision: https://phabricator.services.mozilla.com/D213425 Differential Revision: https://phabricator.services.mozilla.com/D215657
47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
from typing import Any
|
|
|
|
from taskgraph.task import Task
|
|
from taskgraph.util.readonlydict import ReadOnlyDict
|
|
|
|
immutable_types = {int, float, bool, str, type(None), ReadOnlyDict}
|
|
|
|
|
|
def deepcopy(obj: Any) -> Any:
|
|
"""Perform a deep copy of an object with a tree like structure.
|
|
|
|
This is a re-implementation of Python's `copy.deepcopy` function with a few key differences:
|
|
|
|
1. Unlike the stdlib, this does *not* support copying graph-like structure,
|
|
which allows it to be more efficient than deepcopy on tree-like structures
|
|
(such as Tasks).
|
|
2. This special cases support for `taskgraph.task.Task` objects.
|
|
|
|
Args:
|
|
obj: The object to deep copy.
|
|
|
|
Returns:
|
|
A deep copy of the object.
|
|
"""
|
|
ty = type(obj)
|
|
if ty in immutable_types:
|
|
return obj
|
|
if ty is dict:
|
|
return {k: deepcopy(v) for k, v in obj.items()}
|
|
if ty is list:
|
|
return [deepcopy(elt) for elt in obj]
|
|
if ty is Task:
|
|
task = Task(
|
|
kind=deepcopy(obj.kind),
|
|
label=deepcopy(obj.label),
|
|
attributes=deepcopy(obj.attributes),
|
|
task=deepcopy(obj.task),
|
|
description=deepcopy(obj.description),
|
|
optimization=deepcopy(obj.optimization),
|
|
dependencies=deepcopy(obj.dependencies),
|
|
soft_dependencies=deepcopy(obj.soft_dependencies),
|
|
if_dependencies=deepcopy(obj.if_dependencies),
|
|
)
|
|
if obj.task_id:
|
|
task.task_id = obj.task_id
|
|
return task
|
|
raise NotImplementedError(f"copying '{ty}' from '{obj}'")
|