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
|
#!/usr/bin/env python
# --------------------------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for license information.
from subprocess import check_call
import argparse
import os
import logging
from tox_helper_tasks import find_whl, get_package_details
logging.getLogger().setLevel(logging.INFO)
root_dir = os.path.abspath(os.path.join(os.path.abspath(__file__), "..", "..", ".."))
def get_package_wheel_path(pkg_root):
# parse setup.py to get package name and version
pkg_name, _, version = get_package_details(os.path.join(pkg_root, "setup.py"))
# Check if wheel is already built and available for current package
prebuilt_dir = os.getenv("PREBUILT_WHEEL_DIR")
if prebuilt_dir:
prebuilt_package_path = find_whl(prebuilt_dir, pkg_name, version)
else:
return None
if __name__ == "__main__":
parser = argparse.ArgumentParser(
description="Run apistubgen against target folder. "
)
parser.add_argument(
"-t",
"--target",
dest="target_package",
help="The target package directory on disk.",
required=True,
)
parser.add_argument(
"-w",
"--work_dir",
dest="work_dir",
help="Working directory to run apistubgen",
required=True,
)
args = parser.parse_args()
# Check if a wheel is already built for current package and install from wheel when available
# If wheel is not available then install package from source
pkg_path = get_package_wheel_path(args.target_package)
if not pkg_path:
pkg_path = args.target_package
check_call(["apistubgen", "--pkg-path", pkg_path,], cwd=args.work_dir)
|
