azure/aria/aria-extension-cloudify/src/aria/aria/orchestrator/topology/utils.py

   1 # Licensed to the Apache Software Foundation (ASF) under one or more
   2 # contributor license agreements.  See the NOTICE file distributed with
   3 # this work for additional information regarding copyright ownership.
   4 # The ASF licenses this file to You under the Apache License, Version 2.0
   5 # (the "License"); you may not use this file except in compliance with
   6 # the License.  You may obtain a copy of the License at
   7 #
   8 #     http://www.apache.org/licenses/LICENSE-2.0
   9 #
  10 # Unless required by applicable law or agreed to in writing, software
  11 # distributed under the License is distributed on an "AS IS" BASIS,
  12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 # See the License for the specific language governing permissions and
  14 # limitations under the License.
  15
  16 from copy import deepcopy
  17
  18
  19 def deepcopy_with_locators(value):
  20     """
  21     Like :func:`deepcopy`, but also copies over locators.
  22     """
  23
  24     res = deepcopy(value)
  25     copy_locators(res, value)
  26     return res
  27
  28
  29 def copy_locators(target, source):
  30     """
  31     Copies over ``_locator`` for all elements, recursively.
  32
  33     Assumes that target and source have exactly the same list/dict structure.
  34     """
  35
  36     locator = getattr(source, '_locator', None)
  37     if locator is not None:
  38         try:
  39             setattr(target, '_locator', locator)
  40         except AttributeError:
  41             pass
  42
  43     if isinstance(target, list) and isinstance(source, list):
  44         for i, _ in enumerate(target):
  45             copy_locators(target[i], source[i])
  46     elif isinstance(target, dict) and isinstance(source, dict):
  47         for k, v in target.items():
  48             copy_locators(v, source[k])