-
Notifications
You must be signed in to change notification settings - Fork 396
/
pip_build.py
92 lines (73 loc) · 2.73 KB
/
pip_build.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
import argparse
import glob
import os
import shutil
import namex
package = "keras_tuner"
build_directory = "build"
dist_directory = "dist"
to_copy = [
"setup.py",
"setup.cfg",
"README.md",
]
def ignore_files(_, filenames):
return [f for f in filenames if "test" in f]
def build():
if os.path.exists(build_directory):
raise ValueError(f"Directory already exists: {build_directory}")
whl_path = None
try:
# Copy sources (`keras_tuner/` directory and setup files) to build
# directory
os.mkdir(build_directory)
shutil.copytree(
package, os.path.join(build_directory, package), ignore=ignore_files
)
for fname in to_copy:
shutil.copy(fname, os.path.join(f"{build_directory}", fname))
os.chdir(build_directory)
# Restructure the codebase so that source files live in
# `keras_tuner/src`
namex.convert_codebase(package, code_directory="src")
# Generate API __init__.py files in `keras_tuner/`
namex.generate_api_files(package, code_directory="src", verbose=True)
# Make sure to export the __version__ string
from keras_tuner.src import __version__ # noqa: E402
with open(os.path.join(package, "__init__.py")) as f:
init_contents = f.read()
with open(os.path.join(package, "__init__.py"), "w") as f:
f.write(init_contents + "\n\n" + f'__version__ = "{__version__}"\n')
# Build the package
os.system("python3 -m build")
# Save the dist files generated by the build process
os.chdir("..")
if not os.path.exists(dist_directory):
os.mkdir(dist_directory)
for filename in glob.glob(
os.path.join(build_directory, dist_directory, "*.*")
):
shutil.copy(filename, dist_directory)
# Find the .whl file path
for fname in os.listdir(dist_directory):
if __version__ in fname and fname.endswith(".whl"):
whl_path = os.path.abspath(os.path.join(dist_directory, fname))
print(f"Build successful. Wheel file available at {whl_path}")
finally:
# Clean up: remove the build directory (no longer needed)
shutil.rmtree(build_directory)
return whl_path
def install_whl(whl_fpath):
print("Installing wheel file.")
os.system(f"pip3 install {whl_fpath} --force-reinstall --no-dependencies")
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument(
"--install",
action="store_true",
help="Whether to install the generated wheel file.",
)
args = parser.parse_args()
whl_path = build()
if whl_path and args.install:
install_whl(whl_path)