效果展示
说明
-
本来是想写的完善一些，但由于是自用，所以写出来后发现已经解决了自己的问题，所有 2和3功能没有写。
-
执行的话，需要
cmd之后 直接Python BatchGitPull.py运行下面代码即可。 -
里面同时涉及到其他Pyhon知识点(写给自己的，因为自己老记不住)：
1. 输入任意键退出控制台
2. 控制台输出不同颜色
3. 获取命令执行后的返回结果
4. 根据不同的输入执行不同的代码
5. 获取用户管理员权限，从而刷新系统DNS
6. … -
注意：
is_admin这个访问是来判断是否有管理员权限的，如果担心代码有问题，可以去掉管理员权限的相关代码。我是因为个人情况，不加的话，无法进行系统DNS的刷新，导致有时候梯子换了频道后，还是Pull不下来，所以出此下策。
具体代码
#!/usr/bin/python
# -*- coding:utf-8 -*-
import os
import time
import ctypes
import sys
import subprocess
import colorama
from colorama import Fore, Styledef is_admin():try:return ctypes.windll.shell32.IsUserAnAdmin()except:return Falsedir_root = 'N:/Program Files/sd-webui-aki-v4.4/ComfyUI/ComfyUI/custom_nodes'def PrintError(log):print(Fore.RED + log + Style.RESET_ALL) # 红色def PrintSuccess(log):print(Fore.GREEN + log + Style.RESET_ALL) # 绿色def BatchPull():dir_route = dir_rootif os.path.exists(dir_route) == False:PrintError("没有这个文件夹 " + dir_root)return# 切换到test目录os.chdir(dir_route)current_directory = os.getcwd()dirs = os.listdir()# 定义git命令command = 'git pull' # origin master'netCommand = 'ipconfig/flushdns'index = 0error_list = {}for code_dir in dirs:index += 1# 拼接路径(当前目录+代码目录)# os.system(netCommand)# 使用 subprocess.runresult = subprocess.run(netCommand, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, shell=True)# 获取标准输出# 获取标准错误输出（如果有）if result.stderr:PrintError("网络错误输出：" + result.stderr)else:print("刷新网络成功!") # + result.stdout)full_path = os.path.join(current_directory, code_dir)if os.path.isdir(full_path) == False:print("跳过：不是文件夹 " + full_path)continueos.chdir(full_path)print(str(index) + " : 进行 Pull " + code_dir)# os.system(command)result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, shell=True)waitTimer = 2# 获取标准错误输出（如果有）if result.stderr:waitTimer *= 2PrintError("错误输出：" + result.stderr)error_list[index] = code_direlse:PrintSuccess("完成输出：" + result.stdout)time.sleep(waitTimer)# print(code_dir + " Pull完毕！ " + full_path + '\n')os.chdir(dir_route)if len(error_list) > 0:for err in error_list:PrintError("未完成列表：" + err + " : " + error_list[err])else:PrintSuccess("完美！全部更新成功！！")input("按回车键退出...")project_addr = ['git@gitlab.com:project/mengxixing-h5.git','git@gitlab.com:project/waterinbrainopendomain.git','git@gitlab.com:project/waterinbrain.git'
]def BatchClone():os.mkdir('E:/code')os.chdir('E:/code')command = 'git clone 'for i in project_addr:os.popen(command + str(i))time.sleep(1)print(i + " Clone完毕！")print('当前项目组已全部拉取完毕')# BatchPull()if is_admin():# 初始化coloramacolorama.init()PrintSuccess("测试")user_input = input("将对 \" " + dir_root + " \" 文件夹下所有子文件执行批量Git操作，请确定路径正确。\n\n" +"请选择操作方式: \n1. 按1键 则全部执行Pull，获取最新代码。\n2. 按2键 则只执行Pull失败的库 " +"\n3. 按3键 暂时未开发\n \n \n 注意：\n1. 只有当1键执行后才会有Pull错误数据，这时候2键才有效果\n2. 按键后，再按回车键来确定\n" +"___________________________________________\n")if user_input == "1":BatchPull()elif user_input == "2":print("执行了B操作")elif user_input == "3":input("暂未任何操作，按回车键退出...")else:input("无效输入，按回车键退出...")else:ctypes.windll.shell32.ShellExecuteW(None, "runas", sys.executable, " ".join(sys.argv), None, 1)
