mirror of
https://github.com/ivellioscolin/pykd.git
synced 2025-04-20 11:43:23 +08:00
c342479aed
Changes to setup.py: Allow specifying below dirs for clean setuptools build. Be default these dirs will be set automatically based on solution global configuration. - "--build-base" for build - "--dist-dir" for bdist_wheel - "--egg-base" for egg_info "--python-tag" is set based on current python major.minor version. bdist_zip will package symbol along with binary as they should always match. Package version is dynamically set: - Read 1st line of VERSION file as package version, if the file exists. Useful for release branch. - If no VERSION file, parse from pykdver.h, plus +g append commit SHA. "--pplat-name" accepts both win32 and x86, x64 win-amd64 and x64 due to VS naming convention. Changes to buildall.cmd Simplify batch command by put all python version and configs in array. Signed-off-by: Colin Xu <colin.xu@gmail.com>
189 lines
7.2 KiB
Python
189 lines
7.2 KiB
Python
import os
|
|
import shutil
|
|
import zipfile
|
|
import sys
|
|
import argparse
|
|
import re
|
|
import setuptools
|
|
import git
|
|
|
|
if sys.version_info.major == 2:
|
|
import pathlib2
|
|
|
|
_name = "pykd"
|
|
_desc = "python windbg extension"
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--plat-name', choices = ['win32', 'win-amd64', 'x86', 'x64'], required = True, type = str.lower)
|
|
parser.add_argument('--build-base', type = str, required = False)
|
|
parser.add_argument('--dist-dir', type = str, required = False)
|
|
parser.add_argument('--egg-base', type = str, required = False)
|
|
args, unkonwn_args = parser.parse_known_args()
|
|
|
|
platform = ''
|
|
platform_alt = ''
|
|
configuration = 'Release'
|
|
if (args.plat_name == 'win32') or (args.plat_name == 'x86'):
|
|
args.plat_name = 'win32'
|
|
platform = 'win32'
|
|
platform_alt = 'x86'
|
|
elif (args.plat_name == 'win-amd64') or (args.plat_name == 'x64'):
|
|
args.plat_name = 'win-amd64'
|
|
platform = 'x64'
|
|
platform_alt = 'x64'
|
|
else:
|
|
print("Unsupported platform %s" %(args.plat_name))
|
|
exit
|
|
|
|
python_tag = "cp%d%d" %(sys.version_info.major, sys.version_info.minor)
|
|
|
|
dir_solution = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
|
|
|
_version = ''
|
|
|
|
# Using VERSION file on releaase branch
|
|
if os.path.isfile(os.path.join(os.path.dirname(__file__), 'VERSION')):
|
|
with open(os.path.join(os.path.dirname(__file__), 'VERSION'), 'r') as verf:
|
|
_version = verf.readline()
|
|
|
|
# Using versions from pykdver.h and branch HEAD
|
|
if _version == '':
|
|
versions = ['', '', '', '']
|
|
if os.path.isfile(os.path.join(dir_solution, 'pykd', 'pykdver.h')):
|
|
with open(os.path.join(os.path.join(dir_solution, 'pykd', 'pykdver.h')), 'r') as verf:
|
|
lines = verf.readlines()
|
|
for line in lines:
|
|
ver_raw_match = re.findall(r"^#define\s*PYKD_VERSION_MAJOR\s*(.*)$", line)
|
|
if (len(ver_raw_match)):
|
|
versions[0] = ver_raw_match[0]
|
|
continue
|
|
ver_raw_match = re.findall(r"^#define\s*PYKD_VERSION_MINOR\s*(.*)$", line)
|
|
if (len(ver_raw_match)):
|
|
versions[1] = ver_raw_match[0]
|
|
continue
|
|
ver_raw_match = re.findall(r"^#define\s*PYKD_VERSION_SUBVERSION\s*(.*)$", line)
|
|
if (len(ver_raw_match)):
|
|
versions[2] = ver_raw_match[0]
|
|
continue
|
|
ver_raw_match = re.findall(r"^#define\s*PYKD_VERSION_BUILDNO\s*(.*)$", line)
|
|
if (len(ver_raw_match)):
|
|
versions[3] = ver_raw_match[0]
|
|
continue
|
|
|
|
_version = "%s.%s.%s.%s+g" % (versions[0], versions[1], versions[2], versions[3])
|
|
|
|
try:
|
|
repo = git.Repo(dir_solution)
|
|
_version = "%s%s" % (_version, repo.git.rev_parse(repo.head, short = True))
|
|
except:
|
|
print("Skip using HEAD SHA as non git repo")
|
|
|
|
dir_dbgsdk_bin = os.path.join(dir_solution, 'kdlibcpp', 'bin', platform_alt)
|
|
if not os.path.exists(dir_dbgsdk_bin):
|
|
print("DbgSDK bin path %s doesn't exist" %(dir_dbgsdk_bin))
|
|
sys.exit()
|
|
|
|
dir_pykd_bin = os.path.join(dir_solution, 'Out', platform, "%s_%d.%d" %(configuration, sys.version_info.major, sys.version_info.minor))
|
|
if not os.path.exists(os.path.join(dir_pykd_bin, 'pykd.pyd')):
|
|
print("pykd.pyd doesn't exist in %s" %(dir_pykd_bin))
|
|
sys.exit()
|
|
|
|
if args.build_base is not None:
|
|
dir_build_base = args.build_base
|
|
else:
|
|
dir_build_base = os.path.join(dir_solution, 'Obj', 'Wheel', platform, "%s_%d.%d" %(configuration, sys.version_info.major, sys.version_info.minor), 'setuptools')
|
|
|
|
if args.dist_dir is not None:
|
|
dir_dist_wheel = args.dist_dir
|
|
else:
|
|
dir_dist_wheel = dir_pykd_bin
|
|
|
|
if args.egg_base is not None:
|
|
dir_egg_base = args.egg_base
|
|
else:
|
|
dir_egg_base = dir_build_base
|
|
|
|
dir_wheel_package = os.path.join(dir_build_base, _name)
|
|
|
|
if os.path.exists(dir_build_base):
|
|
shutil.rmtree(dir_build_base)
|
|
if sys.version_info.major == 2:
|
|
pathlib2.Path(dir_build_base).mkdir(exist_ok = True, parents = True)
|
|
else:
|
|
os.makedirs(dir_build_base, exist_ok = True)
|
|
|
|
if os.path.exists(dir_wheel_package):
|
|
shutil.rmtree(dir_wheel_package)
|
|
if sys.version_info.major == 2:
|
|
pathlib2.Path(dir_wheel_package).mkdir(exist_ok = True, parents = True)
|
|
else:
|
|
os.makedirs(dir_wheel_package, exist_ok = True)
|
|
|
|
print("Using DbgSDK from %s" %(dir_dbgsdk_bin))
|
|
print("Using pykd binary from %s" %(dir_pykd_bin))
|
|
print("Using build directory as %s" %(dir_build_base))
|
|
print("Using egg info directory as %s" %(dir_egg_base))
|
|
print("Using Wheel output directory as %s" %(dir_dist_wheel))
|
|
|
|
shutil.copy2(os.path.join(dir_solution, 'setup', '__init__.py'), dir_wheel_package)
|
|
shutil.copy2(os.path.join(dir_pykd_bin, 'pykd.pyd'), dir_wheel_package)
|
|
for dbgSdkBin in [ f for f in os.listdir(dir_dbgsdk_bin) if not os.path.isdir(f) ]:
|
|
shutil.copy2(os.path.join(dir_dbgsdk_bin, dbgSdkBin), dir_wheel_package)
|
|
|
|
if 'bdist_wheel' in unkonwn_args:
|
|
print("Building bdist_wheel ...")
|
|
sys.argv = ['setup.py']
|
|
sys.argv.append("build")
|
|
sys.argv.append("--build-base=%s" %(dir_build_base))
|
|
sys.argv.append("--plat-name=%s" %(args.plat_name))
|
|
sys.argv.append("bdist_wheel")
|
|
sys.argv.append("--dist-dir=%s" %(dir_dist_wheel))
|
|
sys.argv.append("--plat-name=%s" %(args.plat_name))
|
|
sys.argv.append("--python-tag=%s" %(python_tag))
|
|
sys.argv.append("egg_info")
|
|
sys.argv.append("--egg-base=%s" %(dir_egg_base))
|
|
|
|
setuptools.setup(
|
|
name = _name,
|
|
version = _version,
|
|
description = _desc,
|
|
packages = ['pykd'],
|
|
package_dir = {'pykd': os.path.relpath(dir_wheel_package)},
|
|
package_data = { 'pykd' :["*.pyd", "*.dll"]},
|
|
include_package_data=True,
|
|
zip_safe = False
|
|
)
|
|
|
|
if 'bdist_zip' in unkonwn_args:
|
|
zip_name = "pykd-%s-%s-%s.zip" %(_version, python_tag, args.plat_name)
|
|
print("Building bdist_zip %s ..." %(zip_name))
|
|
with zipfile.ZipFile(os.path.join(dir_dist_wheel, zip_name), mode='w') as archive:
|
|
for dbgSdkBin in [ f for f in os.listdir(dir_dbgsdk_bin) if not os.path.isdir(f) ]:
|
|
print("zipped %s" %(dbgSdkBin))
|
|
archive.write(os.path.join(dir_dbgsdk_bin, dbgSdkBin), arcname = dbgSdkBin, compress_type = zipfile.ZIP_DEFLATED)
|
|
print("zipped pykd.pyd")
|
|
archive.write(os.path.join(dir_pykd_bin, 'pykd.pyd'), arcname = 'pykd.pyd', compress_type = zipfile.ZIP_DEFLATED)
|
|
print("zipped pykd.pdb")
|
|
archive.write(os.path.join(dir_pykd_bin, 'pykd.pdb'), arcname = 'pykd.pdb', compress_type = zipfile.ZIP_DEFLATED)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|