模块化重构代码

This commit is contained in:
2026-02-14 01:08:38 +08:00
parent f07619c5b4
commit ca8ff7db82
9 changed files with 896 additions and 563 deletions

View File

@@ -1,6 +1,6 @@
# QuickGit - 萌芽一键Git管理工具
一个简单易用的Git命令行管理工具让Git操作更加便捷高效。
一个简单易用的模块化Git命令行管理工具让Git操作更加便捷高效。
## 功能特性
@@ -10,12 +10,33 @@
- **远程仓库管理** - 便捷地添加、删除、查看远程仓库
- **状态查看** - 快速查看仓库状态和提交历史
- **彩色界面** - 友好的彩色控制台输出
- **模块化设计** - 易于维护和扩展
## 项目结构
```
QuickGit/
├── quickgit/ # 核心模块
│ ├── __init__.py # 包初始化
│ ├── config.py # 配置模块
│ ├── utils.py # 工具类(命令执行、输出格式化、输入验证)
│ ├── git_operations.py # Git操作模块
│ ├── remote_manager.py # 远程仓库管理模块
│ └── ui.py # UI交互模块
├── quickgit.py # 主程序入口
├── mengya_git_manager.py # 旧版单文件脚本(兼容保留)
└── README.md # 项目文档
```
## 快速开始
### 运行脚本
```bash
# 使用新版模块化脚本(推荐)
python quickgit.py
# 或使用旧版单文件脚本
python mengya_git_manager.py
```
@@ -103,12 +124,40 @@ A: 使用 `5. 管理远程仓库` 功能添加或删除远程仓库
**Q: 支持哪些Git操作**
A: 目前支持init、add、commit、push、pull等常用操作
## 模块说明
### config.py - 配置模块
存储所有配置信息包括Gitea服务器地址、GitHub用户名、.gitignore模板等。
### utils.py - 工具类模块
- `Colors`: 控制台颜色定义
- `CommandExecutor`: 命令执行器
- `OutputFormatter`: 输出格式化器
- `InputValidator`: 输入验证器
### git_operations.py - Git操作模块
提供Git基本操作功能
- 初始化仓库
- 检查状态
- 添加/提交更改
- 推送/拉取代码
### remote_manager.py - 远程仓库管理模块
管理GitHub和Gitea远程仓库
- 添加/删除远程仓库
- 查看远程仓库列表
- 选择推送/拉取目标
### ui.py - UI交互模块
处理用户界面和交互逻辑,整合所有功能模块。
## 开发计划
- [x] 模块化架构重构
- [ ] 支持分支管理
- [ ] 支持标签管理
- [ ] 支持冲突解决辅助
- [ ] 支持配置文件
- [ ] 支持自定义配置文件
- [ ] 支持批量操作多个仓库
## 许可证

View File

@@ -1,561 +0,0 @@
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
萌芽一键Git管理工具
支持快速初始化Git仓库、提交更改、推送到GitHub和Gitea
"""
import os
import sys
import subprocess
from typing import Optional, List
class Colors:
"""控制台颜色"""
HEADER = '\033[95m'
BLUE = '\033[94m'
CYAN = '\033[96m'
GREEN = '\033[92m'
WARNING = '\033[93m'
FAIL = '\033[91m'
ENDC = '\033[0m'
BOLD = '\033[1m'
class GitManager:
"""Git管理器"""
def __init__(self):
self.gitea_host = "repo.shumengya.top"
self.gitea_port = "8022"
self.github_user = "shumengya"
self.current_dir = os.getcwd()
def print_header(self, text: str):
"""打印标题"""
print(f"\n{Colors.HEADER}{Colors.BOLD}{'='*50}")
print(f" {text}")
print(f"{'='*50}{Colors.ENDC}\n")
def print_success(self, text: str):
"""打印成功信息"""
print(f"{Colors.GREEN}{text}{Colors.ENDC}")
def print_error(self, text: str):
"""打印错误信息"""
print(f"{Colors.FAIL}{text}{Colors.ENDC}")
def print_info(self, text: str):
"""打印提示信息"""
print(f"{Colors.CYAN} {text}{Colors.ENDC}")
def print_warning(self, text: str):
"""打印警告信息"""
print(f"{Colors.WARNING}{text}{Colors.ENDC}")
def run_command(self, command: str, show_output: bool = True) -> tuple[bool, str]:
"""
执行命令
返回: (是否成功, 输出内容)
"""
try:
result = subprocess.run(
command,
shell=True,
capture_output=True,
text=True,
encoding='utf-8',
errors='ignore'
)
output = result.stdout + result.stderr
if show_output and output.strip():
print(output)
return result.returncode == 0, output
except Exception as e:
self.print_error(f"命令执行失败: {str(e)}")
return False, str(e)
def is_git_repo(self) -> bool:
"""检查当前目录是否是Git仓库"""
return os.path.isdir('.git')
def get_gitignore_template(self) -> str:
"""获取.gitignore模板"""
return """# Node/React
node_modules/
npm-debug.log*
yarn-debug.log*
yarn-error.log*
build/
dist/
coverage/
.env.local
.env.development.local
.env.test.local
.env.production.local
# Go
*.exe
*.exe~
*.test
*.out
*.dll
*.so
*.dylib
vendor/
# Python
__pycache__/
*.py[cod]
*$py.class
*.so
.Python
env/
venv/
ENV/
*.egg-info/
dist/
build/
# 数据文件
data/data.json
*.db
*.sqlite
# 日志文件
*.log
logs/
# 操作系统
.DS_Store
Thumbs.db
desktop.ini
# IDE
.idea/
.vscode/
*.swp
*.swo
*~
.project
.classpath
.settings/
# 其他
*.bak
*.tmp
*.temp
"""
def init_git_repo(self):
"""初始化Git仓库"""
self.print_header("初始化Git仓库")
if self.is_git_repo():
self.print_warning("当前目录已经是Git仓库")
return True
# 1. 初始化Git仓库
self.print_info("正在初始化Git仓库...")
success, _ = self.run_command("git init", show_output=False)
if not success:
self.print_error("Git初始化失败")
return False
self.print_success("Git仓库初始化成功")
# 2. 创建main分支
self.print_info("正在创建main分支...")
success, _ = self.run_command("git checkout -b main", show_output=False)
if success:
self.print_success("main分支创建成功")
else:
self.print_warning("main分支创建失败将使用默认分支")
# 3. 创建.gitignore文件
self.print_info("正在创建.gitignore文件...")
try:
with open('.gitignore', 'w', encoding='utf-8') as f:
f.write(self.get_gitignore_template())
self.print_success(".gitignore文件创建成功")
except Exception as e:
self.print_error(f".gitignore文件创建失败: {str(e)}")
# 4. 首次提交
self.print_info("正在进行首次提交...")
self.run_command("git add .", show_output=False)
success, _ = self.run_command('git commit -m "first commit"', show_output=False)
if success:
self.print_success("首次提交完成")
else:
self.print_warning("首次提交失败(可能没有文件可提交)")
# 5. 配置远程仓库
self.configure_remotes()
return True
def configure_remotes(self):
"""配置远程仓库"""
self.print_info("\n配置远程仓库...")
print("\n请选择要配置的远程仓库:")
print("1. GitHub")
print("2. Gitea")
print("3. 两者都配置")
print("4. 跳过")
choice = input("\n请选择 [1-4]: ").strip()
if choice == '1':
self.add_github_remote()
elif choice == '2':
self.add_gitea_remote()
elif choice == '3':
self.add_github_remote()
self.add_gitea_remote()
else:
self.print_info("跳过远程仓库配置")
def add_github_remote(self):
"""添加GitHub远程仓库"""
repo_name = input(f"\n请输入GitHub仓库名: ").strip()
if not repo_name:
self.print_warning("仓库名不能为空跳过GitHub配置")
return
remote_url = f"git@github.com:{self.github_user}/{repo_name}.git"
# 检查remote是否已存在
success, output = self.run_command("git remote", show_output=False)
if "github" in output:
self.run_command("git remote remove github", show_output=False)
success, _ = self.run_command(f'git remote add github {remote_url}', show_output=False)
if success:
self.print_success(f"GitHub远程仓库已添加: {remote_url}")
else:
self.print_error("GitHub远程仓库添加失败")
def add_gitea_remote(self):
"""添加Gitea远程仓库"""
user = input(f"\n请输入Gitea用户名: ").strip()
if not user:
self.print_warning("用户名不能为空跳过Gitea配置")
return
repo_name = input(f"请输入Gitea仓库名: ").strip()
if not repo_name:
self.print_warning("仓库名不能为空跳过Gitea配置")
return
remote_url = f"ssh://git@{self.gitea_host}:{self.gitea_port}/{user}/{repo_name}.git"
# 检查remote是否已存在
success, output = self.run_command("git remote", show_output=False)
if "gitea" in output:
self.run_command("git remote remove gitea", show_output=False)
success, _ = self.run_command(f'git remote add gitea {remote_url}', show_output=False)
if success:
self.print_success(f"Gitea远程仓库已添加: {remote_url}")
else:
self.print_error("Gitea远程仓库添加失败")
def commit_and_push(self):
"""提交并推送更改"""
self.print_header("提交并推送更改")
if not self.is_git_repo():
self.print_error("当前目录不是Git仓库请先初始化")
return False
# 1. 检查是否有更改
self.print_info("检查文件更改...")
success, output = self.run_command("git status --short", show_output=False)
if not output.strip():
self.print_warning("没有文件更改,无需提交")
return True
print("\n当前更改的文件:")
print(output)
# 2. 输入提交信息
commit_msg = input("\n请输入提交信息 (直接回车使用默认信息): ").strip()
if not commit_msg:
from datetime import datetime
commit_msg = f"update: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
# 3. 添加所有更改
self.print_info("正在添加文件...")
success, _ = self.run_command("git add .", show_output=False)
if not success:
self.print_error("添加文件失败")
return False
self.print_success("文件添加成功")
# 4. 提交更改
self.print_info("正在提交更改...")
success, _ = self.run_command(f'git commit -m "{commit_msg}"', show_output=True)
if not success:
self.print_error("提交失败")
return False
self.print_success("提交成功")
# 5. 推送到远程仓库
self.push_to_remote()
return True
def push_to_remote(self):
"""推送到远程仓库"""
# 获取当前分支
success, branch = self.run_command("git branch --show-current", show_output=False)
if not success or not branch.strip():
branch = "main"
else:
branch = branch.strip()
# 获取所有远程仓库
success, output = self.run_command("git remote", show_output=False)
if not success or not output.strip():
self.print_warning("没有配置远程仓库")
return
remotes = [r.strip() for r in output.strip().split('\n') if r.strip()]
if not remotes:
self.print_warning("没有配置远程仓库")
return
print("\n可用的远程仓库:")
for idx, remote in enumerate(remotes, 1):
print(f"{idx}. {remote}")
print(f"{len(remotes) + 1}. 全部推送")
choice = input(f"\n请选择要推送的远程仓库 [1-{len(remotes) + 1}]: ").strip()
try:
choice_idx = int(choice)
if choice_idx == len(remotes) + 1:
# 推送到所有远程仓库
for remote in remotes:
self.push_to_specific_remote(remote, branch)
elif 1 <= choice_idx <= len(remotes):
# 推送到指定远程仓库
remote = remotes[choice_idx - 1]
self.push_to_specific_remote(remote, branch)
else:
self.print_error("无效的选择")
except ValueError:
self.print_error("无效的输入")
def push_to_specific_remote(self, remote: str, branch: str):
"""推送到指定远程仓库"""
self.print_info(f"正在推送到 {remote}...")
# 检查是否需要设置上游分支
success, _ = self.run_command(f"git push {remote} {branch}", show_output=True)
if not success:
# 尝试设置上游分支
self.print_info(f"尝试设置上游分支...")
success, _ = self.run_command(f"git push -u {remote} {branch}", show_output=True)
if success:
self.print_success(f"成功推送到 {remote}")
else:
self.print_error(f"推送到 {remote} 失败")
def pull_from_remote(self):
"""从远程仓库拉取"""
self.print_header("从远程仓库拉取")
if not self.is_git_repo():
self.print_error("当前目录不是Git仓库")
return False
# 获取所有远程仓库
success, output = self.run_command("git remote", show_output=False)
if not success or not output.strip():
self.print_warning("没有配置远程仓库")
return False
remotes = [r.strip() for r in output.strip().split('\n') if r.strip()]
if not remotes:
self.print_warning("没有配置远程仓库")
return False
print("\n可用的远程仓库:")
for idx, remote in enumerate(remotes, 1):
print(f"{idx}. {remote}")
choice = input(f"\n请选择要拉取的远程仓库 [1-{len(remotes)}]: ").strip()
try:
choice_idx = int(choice)
if 1 <= choice_idx <= len(remotes):
remote = remotes[choice_idx - 1]
# 获取当前分支
success, branch = self.run_command("git branch --show-current", show_output=False)
if not success or not branch.strip():
branch = "main"
else:
branch = branch.strip()
self.print_info(f"正在从 {remote} 拉取 {branch} 分支...")
success, _ = self.run_command(f"git pull {remote} {branch}", show_output=True)
if success:
self.print_success(f"成功从 {remote} 拉取更新")
else:
self.print_error(f"{remote} 拉取失败")
else:
self.print_error("无效的选择")
except ValueError:
self.print_error("无效的输入")
def show_status(self):
"""显示仓库状态"""
self.print_header("仓库状态")
if not self.is_git_repo():
self.print_error("当前目录不是Git仓库")
return
print(f"{Colors.CYAN}当前目录:{Colors.ENDC} {self.current_dir}\n")
# Git状态
self.print_info("Git状态")
self.run_command("git status", show_output=True)
# 远程仓库
print(f"\n{Colors.CYAN}远程仓库:{Colors.ENDC}")
self.run_command("git remote -v", show_output=True)
# 最近提交
print(f"\n{Colors.CYAN}最近3次提交{Colors.ENDC}")
self.run_command("git log --oneline -3", show_output=True)
def manage_remotes(self):
"""管理远程仓库"""
self.print_header("管理远程仓库")
if not self.is_git_repo():
self.print_error("当前目录不是Git仓库")
return
while True:
print("\n远程仓库管理:")
print("1. 查看远程仓库")
print("2. 添加GitHub远程仓库")
print("3. 添加Gitea远程仓库")
print("4. 删除远程仓库")
print("5. 返回主菜单")
choice = input("\n请选择 [1-5]: ").strip()
if choice == '1':
print(f"\n{Colors.CYAN}当前远程仓库:{Colors.ENDC}")
self.run_command("git remote -v", show_output=True)
elif choice == '2':
self.add_github_remote()
elif choice == '3':
self.add_gitea_remote()
elif choice == '4':
self.remove_remote()
elif choice == '5':
break
else:
self.print_error("无效的选择")
def remove_remote(self):
"""删除远程仓库"""
success, output = self.run_command("git remote", show_output=False)
if not success or not output.strip():
self.print_warning("没有配置远程仓库")
return
remotes = [r.strip() for r in output.strip().split('\n') if r.strip()]
if not remotes:
self.print_warning("没有配置远程仓库")
return
print("\n当前远程仓库:")
for idx, remote in enumerate(remotes, 1):
print(f"{idx}. {remote}")
choice = input(f"\n请选择要删除的远程仓库 [1-{len(remotes)}]: ").strip()
try:
choice_idx = int(choice)
if 1 <= choice_idx <= len(remotes):
remote = remotes[choice_idx - 1]
confirm = input(f"确认删除远程仓库 '{remote}'? [y/N]: ").strip().lower()
if confirm == 'y':
success, _ = self.run_command(f"git remote remove {remote}", show_output=False)
if success:
self.print_success(f"远程仓库 '{remote}' 已删除")
else:
self.print_error("删除失败")
else:
self.print_error("无效的选择")
except ValueError:
self.print_error("无效的输入")
def run(self):
"""运行主程序"""
self.print_header("萌芽一键Git管理工具")
print(f"{Colors.CYAN}当前目录:{Colors.ENDC} {self.current_dir}")
print(f"{Colors.CYAN}Git仓库:{Colors.ENDC} {'' if self.is_git_repo() else ''}")
while True:
print(f"\n{Colors.BOLD}请选择操作:{Colors.ENDC}")
print("1. 初始化Git仓库")
print("2. 提交并推送更改")
print("3. 从远程仓库拉取")
print("4. 查看仓库状态")
print("5. 管理远程仓库")
print("6. 退出")
choice = input(f"\n{Colors.BOLD}请输入选项 [1-6]: {Colors.ENDC}").strip()
if choice == '1':
self.init_git_repo()
elif choice == '2':
self.commit_and_push()
elif choice == '3':
self.pull_from_remote()
elif choice == '4':
self.show_status()
elif choice == '5':
self.manage_remotes()
elif choice == '6':
self.print_success("感谢使用萌芽Git管理工具")
break
else:
self.print_error("无效的选择,请重新输入")
def main():
"""主函数"""
try:
manager = GitManager()
manager.run()
except KeyboardInterrupt:
print(f"\n\n{Colors.WARNING}程序被用户中断{Colors.ENDC}")
sys.exit(0)
except Exception as e:
print(f"\n{Colors.FAIL}发生错误: {str(e)}{Colors.ENDC}")
sys.exit(1)
if __name__ == "__main__":
main()

30
quickgit.py Normal file
View File

@@ -0,0 +1,30 @@
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
QuickGit - 萌芽一键Git管理工具
主程序入口
"""
import sys
from quickgit.ui import GitManagerUI
def main():
"""主函数"""
try:
ui = GitManagerUI()
ui.run()
except KeyboardInterrupt:
from quickgit.utils import Colors
print(f"\n\n{Colors.WARNING}程序被用户中断{Colors.ENDC}")
sys.exit(0)
except Exception as e:
from quickgit.utils import Colors
print(f"\n{Colors.FAIL}发生错误: {str(e)}{Colors.ENDC}")
import traceback
traceback.print_exc()
sys.exit(1)
if __name__ == "__main__":
main()

6
quickgit/__init__.py Normal file
View File

@@ -0,0 +1,6 @@
"""
QuickGit - 萌芽一键Git管理工具
"""
__version__ = "1.0.0"
__author__ = "shumengya"

84
quickgit/config.py Normal file
View File

@@ -0,0 +1,84 @@
"""
配置模块 - 存储项目配置信息
"""
class Config:
"""配置类"""
# Gitea服务器配置
GITEA_HOST = "repo.shumengya.top"
GITEA_PORT = "8022"
# GitHub配置
GITHUB_USER = "shumengya"
# Git配置
DEFAULT_BRANCH = "main"
# .gitignore模板
GITIGNORE_TEMPLATE = """# Node/React
node_modules/
npm-debug.log*
yarn-debug.log*
yarn-error.log*
build/
dist/
coverage/
.env.local
.env.development.local
.env.test.local
.env.production.local
# Go
*.exe
*.exe~
*.test
*.out
*.dll
*.so
*.dylib
vendor/
# Python
__pycache__/
*.py[cod]
*$py.class
*.so
.Python
env/
venv/
ENV/
*.egg-info/
dist/
build/
# 数据文件
data/data.json
*.db
*.sqlite
# 日志文件
*.log
logs/
# 操作系统
.DS_Store
Thumbs.db
desktop.ini
# IDE
.idea/
.vscode/
*.swp
*.swo
*~
.project
.classpath
.settings/
# 其他
*.bak
*.tmp
*.temp
"""

207
quickgit/git_operations.py Normal file
View File

@@ -0,0 +1,207 @@
"""
Git操作模块 - 提供Git基本操作功能
"""
import os
from datetime import datetime
from .utils import CommandExecutor, OutputFormatter
from .config import Config
class GitOperations:
"""Git操作类"""
def __init__(self):
self.executor = CommandExecutor()
self.current_dir = os.getcwd()
def is_git_repo(self) -> bool:
"""检查当前目录是否是Git仓库"""
return os.path.isdir('.git')
def init_repo(self) -> bool:
"""
初始化Git仓库
Returns:
是否成功
"""
if self.is_git_repo():
OutputFormatter.warning("当前目录已经是Git仓库")
return True
# 初始化Git仓库
OutputFormatter.info("正在初始化Git仓库...")
success, _ = self.executor.run("git init", show_output=False)
if not success:
OutputFormatter.error("Git初始化失败")
return False
OutputFormatter.success("Git仓库初始化成功")
# 创建main分支
OutputFormatter.info("正在创建main分支...")
success, _ = self.executor.run(f"git checkout -b {Config.DEFAULT_BRANCH}", show_output=False)
if success:
OutputFormatter.success("main分支创建成功")
else:
OutputFormatter.warning("main分支创建失败将使用默认分支")
# 创建.gitignore文件
self._create_gitignore()
# 首次提交
OutputFormatter.info("正在进行首次提交...")
self.executor.run("git add .", show_output=False)
success, _ = self.executor.run('git commit -m "first commit"', show_output=False)
if success:
OutputFormatter.success("首次提交完成")
else:
OutputFormatter.warning("首次提交失败(可能没有文件可提交)")
return True
def _create_gitignore(self):
"""创建.gitignore文件"""
OutputFormatter.info("正在创建.gitignore文件...")
try:
with open('.gitignore', 'w', encoding='utf-8') as f:
f.write(Config.GITIGNORE_TEMPLATE)
OutputFormatter.success(".gitignore文件创建成功")
except Exception as e:
OutputFormatter.error(f".gitignore文件创建失败: {str(e)}")
def get_status(self) -> tuple[bool, str]:
"""
获取Git状态
Returns:
(是否成功, 状态输出)
"""
return self.executor.run("git status --short", show_output=False)
def has_changes(self) -> bool:
"""
检查是否有更改
Returns:
是否有更改
"""
success, output = self.get_status()
return success and bool(output.strip())
def add_all(self) -> bool:
"""
添加所有更改
Returns:
是否成功
"""
OutputFormatter.info("正在添加文件...")
success, _ = self.executor.run("git add .", show_output=False)
if success:
OutputFormatter.success("文件添加成功")
else:
OutputFormatter.error("添加文件失败")
return success
def commit(self, message: str = None) -> bool:
"""
提交更改
Args:
message: 提交信息
Returns:
是否成功
"""
if not message:
message = f"update: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
OutputFormatter.info("正在提交更改...")
success, _ = self.executor.run(f'git commit -m "{message}"', show_output=True)
if success:
OutputFormatter.success("提交成功")
else:
OutputFormatter.error("提交失败")
return success
def get_current_branch(self) -> str:
"""
获取当前分支
Returns:
分支名
"""
success, branch = self.executor.run("git branch --show-current", show_output=False)
if success and branch.strip():
return branch.strip()
return Config.DEFAULT_BRANCH
def push(self, remote: str, branch: str = None) -> bool:
"""
推送到远程仓库
Args:
remote: 远程仓库名
branch: 分支名(默认为当前分支)
Returns:
是否成功
"""
if not branch:
branch = self.get_current_branch()
OutputFormatter.info(f"正在推送到 {remote}...")
# 先尝试直接推送
success, _ = self.executor.run(f"git push {remote} {branch}", show_output=True)
if not success:
# 如果失败,尝试设置上游分支
OutputFormatter.info("尝试设置上游分支...")
success, _ = self.executor.run(f"git push -u {remote} {branch}", show_output=True)
if success:
OutputFormatter.success(f"成功推送到 {remote}")
else:
OutputFormatter.error(f"推送到 {remote} 失败")
return success
def pull(self, remote: str, branch: str = None) -> bool:
"""
从远程仓库拉取
Args:
remote: 远程仓库名
branch: 分支名(默认为当前分支)
Returns:
是否成功
"""
if not branch:
branch = self.get_current_branch()
OutputFormatter.info(f"正在从 {remote} 拉取 {branch} 分支...")
success, _ = self.executor.run(f"git pull {remote} {branch}", show_output=True)
if success:
OutputFormatter.success(f"成功从 {remote} 拉取更新")
else:
OutputFormatter.error(f"{remote} 拉取失败")
return success
def show_status(self):
"""显示仓库状态"""
from .utils import Colors
print(f"{Colors.CYAN}当前目录:{Colors.ENDC} {self.current_dir}\n")
# Git状态
OutputFormatter.info("Git状态")
self.executor.run("git status", show_output=True)
# 最近提交
print(f"\n{Colors.CYAN}最近3次提交{Colors.ENDC}")
self.executor.run("git log --oneline -3", show_output=True)

210
quickgit/remote_manager.py Normal file
View File

@@ -0,0 +1,210 @@
"""
远程仓库管理模块 - 处理GitHub和Gitea远程仓库的管理
"""
from .utils import CommandExecutor, OutputFormatter, InputValidator
from .config import Config
class RemoteManager:
"""远程仓库管理器"""
def __init__(self):
self.executor = CommandExecutor()
def get_remotes(self) -> list[str]:
"""
获取所有远程仓库
Returns:
远程仓库列表
"""
success, output = self.executor.run("git remote", show_output=False)
if success and output.strip():
return [r.strip() for r in output.strip().split('\n') if r.strip()]
return []
def show_remotes(self):
"""显示所有远程仓库"""
from .utils import Colors
print(f"\n{Colors.CYAN}远程仓库:{Colors.ENDC}")
self.executor.run("git remote -v", show_output=True)
def add_github_remote(self, repo_name: str = None) -> bool:
"""
添加GitHub远程仓库
Args:
repo_name: 仓库名如果为None则提示用户输入
Returns:
是否成功
"""
if not repo_name:
repo_name = InputValidator.get_input("\n请输入GitHub仓库名: ")
remote_url = f"git@github.com:{Config.GITHUB_USER}/{repo_name}.git"
return self._add_remote("github", remote_url)
def add_gitea_remote(self, user: str = None, repo_name: str = None) -> bool:
"""
添加Gitea远程仓库
Args:
user: Gitea用户名如果为None则提示用户输入
repo_name: 仓库名如果为None则提示用户输入
Returns:
是否成功
"""
if not user:
user = InputValidator.get_input("\n请输入Gitea用户名: ")
if not repo_name:
repo_name = InputValidator.get_input("请输入Gitea仓库名: ")
remote_url = f"ssh://git@{Config.GITEA_HOST}:{Config.GITEA_PORT}/{user}/{repo_name}.git"
return self._add_remote("gitea", remote_url)
def _add_remote(self, name: str, url: str) -> bool:
"""
添加远程仓库
Args:
name: 远程仓库名称
url: 远程仓库URL
Returns:
是否成功
"""
# 检查remote是否已存在
remotes = self.get_remotes()
if name in remotes:
if InputValidator.confirm(f"远程仓库 '{name}' 已存在,是否覆盖?", default=False):
self.executor.run(f"git remote remove {name}", show_output=False)
else:
return False
success, _ = self.executor.run(f'git remote add {name} {url}', show_output=False)
if success:
OutputFormatter.success(f"{name.capitalize()}远程仓库已添加: {url}")
else:
OutputFormatter.error(f"{name.capitalize()}远程仓库添加失败")
return success
def remove_remote(self, name: str = None) -> bool:
"""
删除远程仓库
Args:
name: 远程仓库名如果为None则提示用户选择
Returns:
是否成功
"""
remotes = self.get_remotes()
if not remotes:
OutputFormatter.warning("没有配置远程仓库")
return False
if not name:
# 让用户选择要删除的远程仓库
print("\n当前远程仓库:")
for idx, remote in enumerate(remotes, 1):
print(f"{idx}. {remote}")
choice = InputValidator.get_choice(
f"\n请选择要删除的远程仓库 [1-{len(remotes)}]: ",
range(1, len(remotes) + 1)
)
name = remotes[choice - 1]
if not InputValidator.confirm(f"确认删除远程仓库 '{name}'?", default=False):
return False
success, _ = self.executor.run(f"git remote remove {name}", show_output=False)
if success:
OutputFormatter.success(f"远程仓库 '{name}' 已删除")
else:
OutputFormatter.error("删除失败")
return success
def configure_remotes_interactive(self):
"""交互式配置远程仓库"""
OutputFormatter.info("\n配置远程仓库...")
print("\n请选择要配置的远程仓库:")
print("1. GitHub")
print("2. Gitea")
print("3. 两者都配置")
print("4. 跳过")
choice = InputValidator.get_choice("\n请选择 [1-4]: ", range(1, 5))
if choice == 1:
self.add_github_remote()
elif choice == 2:
self.add_gitea_remote()
elif choice == 3:
self.add_github_remote()
self.add_gitea_remote()
else:
OutputFormatter.info("跳过远程仓库配置")
def select_remote_for_push(self) -> list[str]:
"""
选择要推送的远程仓库
Returns:
选中的远程仓库列表
"""
remotes = self.get_remotes()
if not remotes:
OutputFormatter.warning("没有配置远程仓库")
return []
print("\n可用的远程仓库:")
for idx, remote in enumerate(remotes, 1):
print(f"{idx}. {remote}")
print(f"{len(remotes) + 1}. 全部推送")
choice = InputValidator.get_choice(
f"\n请选择要推送的远程仓库 [1-{len(remotes) + 1}]: ",
range(1, len(remotes) + 2)
)
if choice == len(remotes) + 1:
return remotes
else:
return [remotes[choice - 1]]
def select_remote_for_pull(self) -> str:
"""
选择要拉取的远程仓库
Returns:
选中的远程仓库名
"""
remotes = self.get_remotes()
if not remotes:
OutputFormatter.warning("没有配置远程仓库")
return None
print("\n可用的远程仓库:")
for idx, remote in enumerate(remotes, 1):
print(f"{idx}. {remote}")
choice = InputValidator.get_choice(
f"\n请选择要拉取的远程仓库 [1-{len(remotes)}]: ",
range(1, len(remotes) + 1)
)
return remotes[choice - 1]

158
quickgit/ui.py Normal file
View File

@@ -0,0 +1,158 @@
"""
UI交互模块 - 处理用户界面和交互逻辑
"""
from .git_operations import GitOperations
from .remote_manager import RemoteManager
from .utils import OutputFormatter, InputValidator, Colors
class GitManagerUI:
"""Git管理器UI"""
def __init__(self):
self.git_ops = GitOperations()
self.remote_mgr = RemoteManager()
def show_welcome(self):
"""显示欢迎信息"""
OutputFormatter.header("萌芽一键Git管理工具 - QuickGit")
print(f"{Colors.CYAN}当前目录:{Colors.ENDC} {self.git_ops.current_dir}")
print(f"{Colors.CYAN}Git仓库:{Colors.ENDC} {'' if self.git_ops.is_git_repo() else ''}")
def show_main_menu(self):
"""显示主菜单"""
print(f"\n{Colors.BOLD}请选择操作:{Colors.ENDC}")
print("1. 初始化Git仓库")
print("2. 提交并推送更改")
print("3. 从远程仓库拉取")
print("4. 查看仓库状态")
print("5. 管理远程仓库")
print("6. 退出")
def handle_init_repo(self):
"""处理初始化仓库"""
OutputFormatter.header("初始化Git仓库")
if self.git_ops.init_repo():
# 配置远程仓库
self.remote_mgr.configure_remotes_interactive()
def handle_commit_and_push(self):
"""处理提交并推送"""
OutputFormatter.header("提交并推送更改")
if not self.git_ops.is_git_repo():
OutputFormatter.error("当前目录不是Git仓库请先初始化")
return
# 检查是否有更改
OutputFormatter.info("检查文件更改...")
if not self.git_ops.has_changes():
OutputFormatter.warning("没有文件更改,无需提交")
return
# 显示更改
print("\n当前更改的文件:")
_, status = self.git_ops.get_status()
print(status)
# 输入提交信息
commit_msg = InputValidator.get_input(
"\n请输入提交信息 (直接回车使用默认信息): ",
allow_empty=True
)
# 添加并提交
if not self.git_ops.add_all():
return
if not self.git_ops.commit(commit_msg if commit_msg else None):
return
# 推送到远程仓库
selected_remotes = self.remote_mgr.select_remote_for_push()
for remote in selected_remotes:
self.git_ops.push(remote)
def handle_pull(self):
"""处理拉取"""
OutputFormatter.header("从远程仓库拉取")
if not self.git_ops.is_git_repo():
OutputFormatter.error("当前目录不是Git仓库")
return
remote = self.remote_mgr.select_remote_for_pull()
if remote:
self.git_ops.pull(remote)
def handle_show_status(self):
"""处理显示状态"""
OutputFormatter.header("仓库状态")
if not self.git_ops.is_git_repo():
OutputFormatter.error("当前目录不是Git仓库")
return
self.git_ops.show_status()
# 显示远程仓库
self.remote_mgr.show_remotes()
def handle_manage_remotes(self):
"""处理远程仓库管理"""
OutputFormatter.header("管理远程仓库")
if not self.git_ops.is_git_repo():
OutputFormatter.error("当前目录不是Git仓库")
return
while True:
print("\n远程仓库管理:")
print("1. 查看远程仓库")
print("2. 添加GitHub远程仓库")
print("3. 添加Gitea远程仓库")
print("4. 删除远程仓库")
print("5. 返回主菜单")
choice = InputValidator.get_choice("\n请选择 [1-5]: ", range(1, 6))
if choice == 1:
self.remote_mgr.show_remotes()
elif choice == 2:
self.remote_mgr.add_github_remote()
elif choice == 3:
self.remote_mgr.add_gitea_remote()
elif choice == 4:
self.remote_mgr.remove_remote()
elif choice == 5:
break
def run(self):
"""运行主程序"""
self.show_welcome()
while True:
self.show_main_menu()
choice = InputValidator.get_choice(
f"\n{Colors.BOLD}请输入选项 [1-6]: {Colors.ENDC}",
range(1, 7)
)
if choice == 1:
self.handle_init_repo()
elif choice == 2:
self.handle_commit_and_push()
elif choice == 3:
self.handle_pull()
elif choice == 4:
self.handle_show_status()
elif choice == 5:
self.handle_manage_remotes()
elif choice == 6:
OutputFormatter.success("感谢使用萌芽Git管理工具")
break

150
quickgit/utils.py Normal file
View File

@@ -0,0 +1,150 @@
"""
工具类模块 - 提供命令执行、输出格式化等工具函数
"""
import subprocess
class Colors:
"""控制台颜色"""
HEADER = '\033[95m'
BLUE = '\033[94m'
CYAN = '\033[96m'
GREEN = '\033[92m'
WARNING = '\033[93m'
FAIL = '\033[91m'
ENDC = '\033[0m'
BOLD = '\033[1m'
class CommandExecutor:
"""命令执行器"""
@staticmethod
def run(command: str, show_output: bool = True) -> tuple[bool, str]:
"""
执行命令
Args:
command: 要执行的命令
show_output: 是否显示输出
Returns:
(是否成功, 输出内容)
"""
try:
result = subprocess.run(
command,
shell=True,
capture_output=True,
text=True,
encoding='utf-8',
errors='ignore'
)
output = result.stdout + result.stderr
if show_output and output.strip():
print(output)
return result.returncode == 0, output
except Exception as e:
OutputFormatter.error(f"命令执行失败: {str(e)}")
return False, str(e)
class OutputFormatter:
"""输出格式化器"""
@staticmethod
def header(text: str):
"""打印标题"""
print(f"\n{Colors.HEADER}{Colors.BOLD}{'='*50}")
print(f" {text}")
print(f"{'='*50}{Colors.ENDC}\n")
@staticmethod
def success(text: str):
"""打印成功信息"""
print(f"{Colors.GREEN}{text}{Colors.ENDC}")
@staticmethod
def error(text: str):
"""打印错误信息"""
print(f"{Colors.FAIL}{text}{Colors.ENDC}")
@staticmethod
def info(text: str):
"""打印提示信息"""
print(f"{Colors.CYAN} {text}{Colors.ENDC}")
@staticmethod
def warning(text: str):
"""打印警告信息"""
print(f"{Colors.WARNING}{text}{Colors.ENDC}")
class InputValidator:
"""输入验证器"""
@staticmethod
def get_choice(prompt: str, valid_range: range) -> int:
"""
获取用户选择(数字)
Args:
prompt: 提示信息
valid_range: 有效范围
Returns:
用户选择的数字
"""
while True:
try:
choice = input(prompt).strip()
choice_num = int(choice)
if choice_num in valid_range:
return choice_num
else:
OutputFormatter.error(f"请输入 {valid_range.start}-{valid_range.stop-1} 之间的数字")
except ValueError:
OutputFormatter.error("请输入有效的数字")
@staticmethod
def get_input(prompt: str, allow_empty: bool = False) -> str:
"""
获取用户输入
Args:
prompt: 提示信息
allow_empty: 是否允许空输入
Returns:
用户输入的字符串
"""
while True:
user_input = input(prompt).strip()
if user_input or allow_empty:
return user_input
else:
OutputFormatter.error("输入不能为空")
@staticmethod
def confirm(prompt: str, default: bool = False) -> bool:
"""
获取用户确认
Args:
prompt: 提示信息
default: 默认值
Returns:
用户确认结果
"""
suffix = " [Y/n]: " if default else " [y/N]: "
user_input = input(prompt + suffix).strip().lower()
if not user_input:
return default
return user_input in ['y', 'yes']