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_)