azure/aria/aria-extension-cloudify/src/aria/aria/utils/process.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 """
  17 Process utilities.
  18 """
  19
  20 import os
  21
  22
  23 def append_to_path(*args, **kwargs):
  24     """
  25     Appends one or more paths to the system path of an environment.
  26     The environment will be that of the current process unless another is passed using the
  27     'env' keyword argument.
  28     :param args: paths to append
  29     :param kwargs: 'env' may be used to pass a custom environment to use
  30     """
  31     _append_to_path('PATH', *args, **kwargs)
  32
  33
  34 def append_to_pythonpath(*args, **kwargs):
  35     """
  36     Appends one or more paths to the python path of an environment.
  37     The environment will be that of the current process unless another is passed using the
  38     'env' keyword argument.
  39     :param args: paths to append
  40     :param kwargs: 'env' may be used to pass a custom environment to use
  41     """
  42     _append_to_path('PYTHONPATH', *args, **kwargs)
  43
  44
  45 def _append_to_path(path, *args, **kwargs):
  46     env = kwargs.get('env') or os.environ
  47     env[path] = '{0}{1}{2}'.format(
  48         os.pathsep.join(args),
  49         os.pathsep,
  50         env.get(path, '')
  51     )