diff --git a/tools/generate_html/run.py b/tools/generate_html/run.py index c6349b18becb35c6579d5991d701161989ae4e9b..cdcf602f0cdf1c4cacff1fac2e900289a9ca5b63 100644 --- a/tools/generate_html/run.py +++ b/tools/generate_html/run.py @@ -21,11 +21,11 @@ from lxml import etree from replace_html_menu import replace_html_menu, modify_menu_num # 下载仓库 -def git_clone(repo_url, repo_dir): +def git_clone(repo_url, repo_dir, repo_branch): if not os.path.exists(repo_dir): print("Cloning repo.....") os.makedirs(repo_dir, exist_ok=True) - Repo.clone_from(repo_url, repo_dir, branch="master", depth=1) + Repo.clone_from(repo_url, repo_dir, branch=repo_branch, depth=1) print("Cloning Repo Done.") # 更新仓库 @@ -173,7 +173,7 @@ def main(version, user, pd, WGETDIR, release_url, generate_list): status_code = requests.get(repo_url, headers=headers, timeout=30).status_code if status_code == 200: if not os.path.exists(repo_path): - git_clone(repo_url, repo_path) + git_clone(repo_url, repo_path, branch_) if data[i]['environ'] == "MSC_PATH": if data[i]['name'] == "mindscience": git_update(repo_path, branch_)