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
|
import os
from glob import glob
from subprocess import check_call, call
from pathlib import Path
def print_check(cmd: str):
check_call(cmd, shell=True)
def print_exec(cmd: str):
call(cmd, shell=True)
def git_clean():
print_check("git checkout .")
print_check("git clean -fd")
print_check("git reset --hard HEAD")
class AutoPrivatePackage:
"""
This class can generate private SDK package
"""
def __init__(self):
self.usr = ""
self.target_branch = ""
self.package_name = ""
def get_input(self):
print("Please commit your code before execute this script!!!")
branch = input("Please input your target branch(e.g. azclibot:t2-compute-2022-01-21-22956):")
info = branch.split(":")
self.usr = info[0]
self.target_branch = info[1]
self.package_name = info[1].split("-")[1]
def checkout_target_branch(self):
git_clean()
print_exec(f"git remote add {self.usr} https://github.com/{self.usr}/azure-sdk-for-python.git")
print_check(f"git fetch {self.usr} {self.target_branch}")
print_check(f"git checkout {self.usr}/{self.target_branch}")
def step_into_package_folder(self):
root_path = os.getcwd()
result = glob(f"{root_path}/sdk/*/azure-mgmt-{self.package_name}")
if len(result) == 0:
raise Exception(f"do not find azure-mgmt-{self.package_name}")
elif len(result) > 1:
raise Exception(f"find multi target package: {str(result)}")
os.chdir(str(Path(result[0])))
def generate_private_package(self):
self.step_into_package_folder()
check_call("python -m build")
print(f'\n package in : {str(Path(os.getcwd()) / "dist")}')
os.system("pause")
def run(self):
self.get_input()
self.checkout_target_branch()
self.generate_private_package()
if __name__ == "__main__":
instance = AutoPrivatePackage()
instance.run()
|
