src/python/grpcio_status/status_commands.py

   1 # Copyright 2018 The gRPC Authors
   2 #
   3 # Licensed under the Apache License, Version 2.0 (the "License");
   4 # you may not use this file except in compliance with the License.
   5 # You may obtain a copy of the License at
   6 #
   7 #     http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 # Unless required by applicable law or agreed to in writing, software
  10 # distributed under the License is distributed on an "AS IS" BASIS,
  11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 # See the License for the specific language governing permissions and
  13 # limitations under the License.
  14 """Provides distutils command classes for the GRPC Python setup process."""
  15
  16 import os
  17 import shutil
  18
  19 import setuptools
  20
  21 ROOT_DIR = os.path.abspath(os.path.dirname(os.path.abspath(__file__)))
  22 STATUS_PROTO = os.path.join(
  23     ROOT_DIR, '../../../third_party/googleapis/google/rpc/status.proto')
  24 PACKAGE_STATUS_PROTO_PATH = 'grpc_status/google/rpc'
  25 LICENSE = os.path.join(ROOT_DIR, '../../../LICENSE')
  26
  27
  28 class Preprocess(setuptools.Command):
  29     """Command to copy LICENSE from root directory."""
  30
  31     description = ''
  32     user_options = []
  33
  34     def initialize_options(self):
  35         pass
  36
  37     def finalize_options(self):
  38         pass
  39
  40     def run(self):
  41         if os.path.isfile(STATUS_PROTO):
  42             if not os.path.isdir(PACKAGE_STATUS_PROTO_PATH):
  43                 os.makedirs(PACKAGE_STATUS_PROTO_PATH)
  44             shutil.copyfile(
  45                 STATUS_PROTO,
  46                 os.path.join(ROOT_DIR, PACKAGE_STATUS_PROTO_PATH,
  47                              'status.proto'))
  48         if os.path.isfile(LICENSE):
  49             shutil.copyfile(LICENSE, os.path.join(ROOT_DIR, 'LICENSE'))