mirror of
https://gitee.com/coder-xiaomo/gitee2github
synced 2025-01-10 11:48:14 +08:00
197 lines
9.1 KiB
Python
197 lines
9.1 KiB
Python
|
||
# #####################################################################################
|
||
# 脚本逻辑
|
||
# 首先在当前目录下创建一个临时目录，然后让用户授权Gitee和GitHub账号以获得账号中的所有Git仓库信息
|
||
# 最后使用 `git clone --mirror xxx` 命令克隆裸仓库，并使用 `git push --mirror xxx` 推送，
|
||
# 以达到转移仓库的目的
|
||
# #####################################################################################
|
||
# 脚本使用前置条件
|
||
# - 电脑上已安装Python，并且安装了下方import的Python库
|
||
# - 没有安装的话可以使用 `pip install <库名>` 进行安装
|
||
# - 建议先配置好pip国内镜像，再使用pip安装，这样安装速度会快一些
|
||
# - 电脑上已安装Git
|
||
# - 已经配置好本地SSH信息
|
||
# - 即可以直接使用git命令克隆或推送SSH地址git@git___.com:xxx.git
|
||
# - 特别是对于您的私有仓库，只有配置好了SSH公钥的情况下才能直接使用git命令完成克隆
|
||
# - 对于大多数在国内的用户，需要准备一个梯子，否则GitHub请求很容易超时，无法获取仓库信息
|
||
# - 如果GitHub不需要使用代理，请将下方 githubProxies 变量中的代理配置设置为空
|
||
# - 使用梯子请求的话，请将下方 githubProxies 变量中的代理配置设置为梯子的IP地址
|
||
# - 如果Git提示SSL验证失败，可以使用 `git config --global http.sslVerify false` 关闭SSL验证
|
||
# #####################################################################################
|
||
# 注意事项
|
||
# - 使用时，需要使用Gitee和GitHub进行授权
|
||
# - 使用前，请注意备份数据，虽然我们已经使用我们自己的账号测试脚本没有问题，
|
||
# 但受由于执行环境的不同，不排除脚本出现错误等情况。
|
||
# 若因脚本执行错误或者您的失误导致数据丢失，我们不承担责任，感谢您的理解！
|
||
# #####################################################################################
|
||
|
||
import os # 执行系统命令
|
||
import requests # 发送请求
|
||
import signal # 捕获Ctrl+C退出信号
|
||
import webbrowser # 打开浏览器
|
||
import time # 时间模块 (倒计时退出，创建带时间戳的工作目录)
|
||
import shutil # 文件夹操作 （递归删除文件夹）
|
||
import json # 读写JSON
|
||
import configparser # 读取ini配置文件
|
||
import copy # 字典深拷贝
|
||
|
||
# ######################################## 工作路径 ########################################
|
||
# 当前目录
|
||
CurrentDir = os.path.dirname(os.path.abspath(__file__))
|
||
|
||
# 临时工作目录
|
||
WorkingDirName = "WorkingDir" # './temp-' + str(int(time.time() * 10 ** 6))
|
||
WorkingDir = os.path.abspath(os.path.join(CurrentDir, WorkingDirName))
|
||
|
||
# 设置保存路径
|
||
GlobalVarsSavePath = os.path.abspath(os.path.join(WorkingDir, "GlobalVars.json"))
|
||
|
||
# 准备工作目录
|
||
from assets.prepareWorkingDir import prepareWorkingDir
|
||
prepareWorkingDir(CurrentDir = CurrentDir, WorkingDir = WorkingDir)
|
||
|
||
# ######################################## 读取配置 ########################################
|
||
# 读取配置文件
|
||
from assets.readConfigFile import readConfigFile
|
||
config = readConfigFile(os.path.abspath("config.ini"))
|
||
|
||
if not 'Gitee' in config or not 'GitHub' in config:
|
||
print("[error] 配置文件读取失败，请检查配置文件是否正确")
|
||
exit()
|
||
|
||
if not 'ClientID' in config['Gitee'] or not 'ClientSecret' in config['Gitee']:
|
||
print("[error] Gitee ClientID或ClientSecret配置错误，请检查配置文件是否正确")
|
||
exit()
|
||
if not 'ClientID' in config['GitHub'] or not 'ClientSecret' in config['GitHub']:
|
||
print("[error] GitHub ClientID或ClientSecret配置错误，请检查配置文件是否正确")
|
||
exit()
|
||
|
||
# ClientID 与 ClientSecret
|
||
# Gitee
|
||
GiteeClientID = config['Gitee']['ClientID']
|
||
GiteeClientSecret = config['Gitee']['ClientSecret']
|
||
# GitHub
|
||
GitHubClientID =config['GitHub']['ClientID']
|
||
GitHubClientSecret = config['GitHub']['ClientSecret']
|
||
|
||
# Protocol: https or ssh
|
||
GiteeProtocol = 'https'
|
||
GitHubProtocol = 'https'
|
||
# Gitee
|
||
if 'Protocol' in config['Gitee'] and config['Gitee']['Protocol'] == 'ssh':
|
||
GiteeProtocol = "ssh"
|
||
# GitHub
|
||
if 'Protocol' in config['GitHub'] and config['GitHub']['Protocol'] == 'ssh':
|
||
GitHubProtocol = "ssh"
|
||
|
||
# 代理
|
||
# Gitee请求代理，默认为不使用代理 { "http": None, "https": None }
|
||
giteeProxies = { "http": None, "https": None }
|
||
if 'Proxy' in config['Gitee'] and config['Gitee']['Proxy'] != "":
|
||
giteeProxies['http'] = config['Gitee']['Proxy']
|
||
giteeProxies['https'] = config['Gitee']['Proxy']
|
||
# GitHub请求代理，默认为使用本地代理 { "http": "127.0.0.1:15732", "https": "127.0.0.1:15732" }
|
||
# python3 requests使用proxy代理,cookies https://www.cnblogs.com/lshan/p/11878638.html
|
||
githubProxies = { "http": None, "https": None }
|
||
if 'Proxy' in config['GitHub'] and config['GitHub']['Proxy'] != "":
|
||
githubProxies['http'] = config['GitHub']['Proxy']
|
||
githubProxies['https'] = config['GitHub']['Proxy']
|
||
|
||
# 请求超时时间
|
||
timeout = 10
|
||
if 'RequestTimeout' in config['Common'] and config['Common']['RequestTimeout'].isdigit():
|
||
timeout = int(config['Common']['RequestTimeout'])
|
||
print("[info] 请求超时时间设置为：" + str(timeout) + "秒")
|
||
|
||
# ######################################## 主函数 ########################################
|
||
if __name__ == '__main__':
|
||
"""
|
||
主函数
|
||
"""
|
||
from assets.common import saveJSON, readJSON, fileExists
|
||
|
||
GlobalVars = {
|
||
'lastRunTime': time.time(),
|
||
'giteeRepos': {},
|
||
'githubRepos': {},
|
||
'RepoMatch': {},
|
||
}
|
||
|
||
# 如果存在之前的获取结果，则首先导入先前的结果
|
||
importFileOrNot = ''
|
||
if not fileExists(WorkingDir): # 如果工作目录不存在，那么不可导入
|
||
importFileOrNot = 'n'
|
||
if fileExists(GlobalVarsSavePath): # 如果JSON文件存在，那么询问用户否导入
|
||
while(importFileOrNot != 'y' and importFileOrNot != 'n'):
|
||
importFileOrNot = input("[info] 发现上次运行结果，是否读取？(y/n) 默认为y ")
|
||
if importFileOrNot == '':
|
||
importFileOrNot = 'y'
|
||
else: # 如果JSON文件不存在，那么不可导入
|
||
importFileOrNot = 'n'
|
||
|
||
if importFileOrNot == 'y': # 导入
|
||
# 读取上次的结果
|
||
GlobalVars = readJSON(GlobalVarsSavePath)
|
||
GlobalVars['RepoMatch'] = {}
|
||
print("[info] 导入JSON文件成功")
|
||
|
||
# 注册 Ctrl+C 退出处理程序
|
||
# from assets.defineWorkingDirCleanFunction import defineWorkingDirCleanFunction
|
||
# defineWorkingDirCleanFunction(saveJSONFunction = saveJSON)
|
||
|
||
# 询问用户是否重新获取Gitee仓库列表
|
||
if GlobalVars['giteeRepos']:
|
||
userInput = 'unknown'
|
||
while userInput != 'y' and userInput != 'n' and userInput != '':
|
||
userInput = input("[info] 已经获取过Gitee仓库列表，是否重新获取？(y/n) 默认为n ")
|
||
if userInput == 'y':
|
||
GlobalVars['giteeRepos'] = {} # 清空之前的记录
|
||
|
||
if not GlobalVars['giteeRepos']:
|
||
# 读取 Gitee 仓库列表
|
||
from assets.giteeOauth import giteeOauth
|
||
[GlobalVars['giteeReposOrigin'], GlobalVars['giteeRepos']] = \
|
||
giteeOauth(GiteeClientID = GiteeClientID, GiteeClientSecret = GiteeClientSecret, giteeProxies = giteeProxies, timeout = timeout)
|
||
print("[info] 获取到 {} 个Git仓库".format(len(GlobalVars['giteeRepos'])))
|
||
# 保存JSON文件
|
||
saveJSON(GlobalVars, GlobalVarsSavePath)
|
||
print("[info] 保存JSON文件完成")
|
||
|
||
# 询问用户是否重新获取Gitee仓库列表
|
||
if GlobalVars['githubRepos']:
|
||
userInput = 'unknown'
|
||
while userInput != 'y' and userInput != 'n' and userInput != '':
|
||
userInput = input("[info] 已经获取过GitHub仓库列表，是否重新获取？(y/n) 默认为n ")
|
||
if userInput == 'y':
|
||
GlobalVars['githubRepos'] = {} # 清空之前的记录
|
||
|
||
if not GlobalVars['githubRepos']:
|
||
# 读取 GitHub 仓库列表
|
||
from assets.githubOauth import githubOauth
|
||
[GlobalVars['githubReposOrigin'], GlobalVars['githubRepos']] = \
|
||
githubOauth(GitHubClientID = GitHubClientID, GitHubClientSecret = GitHubClientSecret, githubProxies = githubProxies, timeout = timeout)
|
||
print("[info] 获取到 {} 个Git仓库".format(len(GlobalVars['githubRepos'])))
|
||
# 保存JSON文件
|
||
saveJSON(GlobalVars, GlobalVarsSavePath)
|
||
print("[info] 保存JSON文件完成")
|
||
|
||
# 匹配仓库
|
||
from assets.repoMatching import repoMatching, printMatchintInfo
|
||
matchList = repoMatching(repo1 = GlobalVars['giteeRepos'], repo2 = GlobalVars['githubRepos'])
|
||
GlobalVars['RepoMatch'] = matchList
|
||
|
||
# 保存JSON文件
|
||
saveJSON(GlobalVars, GlobalVarsSavePath)
|
||
print("[info] 保存JSON文件完成")
|
||
|
||
# 打印匹配信息
|
||
printMatchintInfo(matchList, repo1Name = "Gitee", repo2Name = "GitHub")
|
||
|
||
# 转移仓库
|
||
input("[info] 确认无误后按回车继续: ")
|
||
print("[info] 开始同步仓库")
|
||
from assets.transferRepos import transferRepos
|
||
transferRepos(matchList.get('match'), WorkingDir, fromRepoProtocol = GiteeProtocol, toRepoProtocol = GitHubProtocol)
|
||
print("程序结束，将退出")
|
||
os.system("pause")
|