Upload 1.py

1.py

@@ -0,0 +1,54 @@
+import os
+
+# 定义要克隆的仓库列表和相应的目录名
+repositories = [
+    ("https://gitcode.net/overbill1683/stablediffusion", "stable-diffusion-stability-ai"),
+    ("https://gitcode.net/overbill1683/taming-transformers", "taming-transformers"),
+    ("https://gitcode.net/overbill1683/k-diffusion", "k-diffusion"),
+    ("https://gitcode.net/overbill1683/CodeFormer", "CodeFormer"),
+    ("https://gitcode.net/overbill1683/BLIP", "BLIP")
+]
+
+def update_repositories():
+    workspace_path = "/mnt/workspace/stable-diffusion-webui"
+    
+    # 切换到工作目录
+    os.chdir(workspace_path)
+    
+    # 更新主仓库
+    os.system("git pull")
+    
+    # 检查并更新子仓库
+    os.chdir(os.path.join(workspace_path, "repositories"))
+    for repo_url, repo_dir in repositories:
+        repo_path = os.path.join(workspace_path, "repositories", repo_dir)
+        
+        if os.path.exists(repo_path):
+            # 切换到子仓库目录
+            os.chdir(repo_path)
+            
+            # 获取远程仓库的最新信息
+            os.system("git fetch")
+            
+            # 检查本地和远程仓库的差异
+            diff_output = os.popen("git diff HEAD..origin/master").read().strip()
+            
+            if diff_output:
+                # 有更新，拉取最新代码
+                os.system("git pull")
+                print(f"子仓库 {repo_dir} 更新成功！")
+            else:
+                print(f"子仓库 {repo_dir} 已经是最新的，无需更新。")
+        else:
+            # 子仓库不存在，克隆仓库
+            os.chdir(workspace_path)
+            os.system(f"git clone {repo_url} {repo_dir}")
+            print(f"成功克隆子仓库 {repo_dir}！")
+    
+    # 下载配置文件
+    os.system("wget -O config.json https://gitcode.net/Akegarasu/sd-webui-configs/-/raw/master/config.json")
+    print("配置文件下载成功！")
+
+if __name__ == "__main__":
+    update_repositories()
+    print("主仓库和子仓库更新完成。")