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
|
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 step_into_sdk_repo() -> str:
return os.chdir(Path(os.getcwd()) / 'azure-sdk-for-python')
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):
step_into_sdk_repo()
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 setup.py bdist_wheel')
check_call('python setup.py sdist --format zip')
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()
|
