1 #!/usr/bin/env python 2 # -*- coding: utf-8 -*- 3 # Copyright 2011 Google Inc. All Rights Reserved. 4 # 5 # Licensed under the Apache License, Version 2.0 (the "License"); 6 # you may not use this file except in compliance with the License. 7 # You may obtain a copy of the License at 8 # 9 # http://www.apache.org/licenses/LICENSE-2.0 10 # 11 # Unless required by applicable law or agreed to in writing, software 12 # distributed under the License is distributed on an "AS IS" BASIS, 13 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14 # See the License for the specific language governing permissions and 15 # limitations under the License. 16 17 """Setup installation module for gsutil.""" 18 19 import os 20 21 from setuptools import find_packages 22 from setuptools import setup 23 from setuptools.command import build_py 24 from setuptools.command import sdist 25 26 long_desc = """ 27 gsutil is a Python application that lets you access Google Cloud Storage from 28 the command line. You can use gsutil to do a wide range of bucket and object 29 management tasks, including: 30 * Creating and deleting buckets. 31 * Uploading, downloading, and deleting objects. 32 * Listing buckets and objects. 33 * Moving, copying, and renaming objects. 34 * Editing object and bucket ACLs. 35 """ 36 37 requires = [ 38 'boto==2.38.0', 39 'crcmod>=1.7', 40 'gcs-oauth2-boto-plugin>=1.9', 41 'google-apitools==0.4.10', 42 'httplib2>=0.8', 43 'oauth2client>=1.4.11', 44 'protorpc>=0.10.0', 45 'pyOpenSSL>=0.13', 46 'python-gflags>=2.0', 47 'retry_decorator>=1.0.0', 48 'six>=1.9.0', 49 # Not using 1.02 because of: 50 # https://code.google.com/p/socksipy-branch/issues/detail?id=3 51 'SocksiPy-branch==1.01', 52 ] 53 54 dependency_links = [ 55 # Note: this commit ID should be kept in sync with the 'third_party/boto' 56 # entry in 'git submodule status'. 57 # pylint: disable=line-too-long 58 'https://github.com/boto/boto/archive/cb8aeec987ddcd5fecd206e38777b9a15cb0bcab.tar.gz#egg=boto-2.38.0', 59 ] 60 61 CURDIR = os.path.abspath(os.path.dirname(__file__)) 62 BOTO_DIR = os.path.join(CURDIR, 'third_party', 'boto') 63 64 with open(os.path.join(CURDIR, 'VERSION'), 'r') as f: 65 VERSION = f.read().strip() 66 67 with open(os.path.join(CURDIR, 'CHECKSUM'), 'r') as f: 68 CHECKSUM = f.read() 69 70 71 def PlaceNeededFiles(self, target_dir): 72 """Populates necessary files into the gslib module and unit test modules.""" 73 target_dir = os.path.join(target_dir, 'gslib') 74 self.mkpath(target_dir) 75 76 # Copy the gsutil root VERSION file into gslib module. 77 with open(os.path.join(target_dir, 'VERSION'), 'w') as fp: 78 fp.write(VERSION) 79 80 # Copy the gsutil root CHECKSUM file into gslib module. 81 with open(os.path.join(target_dir, 'CHECKSUM'), 'w') as fp: 82 fp.write(CHECKSUM) 83 84 # Copy the Boto test module required by gsutil unit tests. 85 tests_dir = os.path.join(target_dir, 'tests') 86 self.mkpath(tests_dir) 87 mock_storage_dst = os.path.join(tests_dir, 'mock_storage_service.py') 88 mock_storage_src1 = os.path.join( 89 BOTO_DIR, 'tests', 'integration', 's3', 'mock_storage_service.py') 90 mock_storage_src2 = os.path.join( 91 CURDIR, 'gslib', 'tests', 'mock_storage_service.py') 92 mock_storage_src = ( 93 mock_storage_src1 94 if os.path.isfile(mock_storage_src1) else mock_storage_src2) 95 if not os.path.isfile(mock_storage_src): 96 raise Exception('Unable to find required boto test source file at %s or %s.' 97 % (mock_storage_src1, mock_storage_src2)) 98 with open(mock_storage_src, 'r') as fp: 99 mock_storage_contents = fp.read() 100 with open(mock_storage_dst, 'w') as fp: 101 fp.write('#\n' 102 '# This file was copied during gsutil package generation from\n' 103 '# the Boto test suite, originally located at:\n' 104 '# tests/integration/s3/mock_storage_service.py\n' 105 '# DO NOT MODIFY\n' 106 '#\n\n') 107 fp.write(mock_storage_contents) 108 109 110 class CustomBuildPy(build_py.build_py): 111 """Excludes update command from package-installed versions of gsutil.""" 112 113 def byte_compile(self, files): 114 for filename in files: 115 # Note: we exclude the update command here because binary distributions 116 # (built via setup.py bdist command) don't abide by the MANIFEST file. 117 # For source distributions (built via setup.py sdist), the update command 118 # will be excluded by the MANIFEST file. 119 if 'gslib/commands/update.py' in filename: 120 os.unlink(filename) 121 build_py.build_py.byte_compile(self, files) 122 123 def run(self): 124 if not self.dry_run: 125 PlaceNeededFiles(self, self.build_lib) 126 build_py.build_py.run(self) 127 128 129 class CustomSDist(sdist.sdist): 130 131 def make_release_tree(self, base_dir, files): 132 sdist.sdist.make_release_tree(self, base_dir, files) 133 PlaceNeededFiles(self, base_dir) 134 135 136 setup( 137 name='gsutil', 138 version=VERSION, 139 url='https://developers.google.com/storage/docs/gsutil', 140 download_url='https://developers.google.com/storage/docs/gsutil_install', 141 license='Apache 2.0', 142 author='Google Inc.', 143 author_email='gs-team (at] google.com', 144 description=('A command line tool for interacting with cloud storage ' 145 'services.'), 146 long_description=long_desc, 147 zip_safe=True, 148 classifiers=[ 149 'Development Status :: 5 - Production/Stable', 150 'Environment :: Console', 151 'Intended Audience :: Developers', 152 'Intended Audience :: System Administrators', 153 'License :: OSI Approved :: Apache Software License', 154 'Natural Language :: English', 155 'Topic :: System :: Filesystems', 156 'Topic :: Utilities', 157 ], 158 platforms='any', 159 packages=find_packages(exclude=['third_party']), 160 include_package_data=True, 161 entry_points={ 162 'console_scripts': [ 163 'gsutil = gslib.__main__:main', 164 ], 165 }, 166 install_requires=requires, 167 dependency_links=dependency_links, 168 cmdclass={ 169 'build_py': CustomBuildPy, 170 'sdist': CustomSDist, 171 } 172 ) 173