使用Python对接Gitlab API批量设置镜像仓库

# Golang
./setting-gitlab-mirror-repo  1.12s user 0.94s system 5% cpu 34.574 total

# Python3
python3 ./setting-gitlab-mirror-repo.py  6.83s user 0.50s system 0% cpu 17:11.92 total

#! /usr/bin/env python3
# -*- coding: UTF-8 -*-

import requests


# 拥有管理员权限的访问令牌
headers = {"PRIVATE-TOKEN": "xxxxxx"}
# 源git服务器仓库地址
fromgitapiaddress = "https://git.oubayun.com/api/v4/projects"
# sync为同步服务器前缀，木子这里源服务器api接口地址为https://git.oubayun.com/api/v4/projects，目标服务器api接口地址为：https://sync.oubayun.com/api/v4/projects
togitapiaddress = "https://username:password@sync"


# 获取所有仓库地址
def get_all_repo(fromgitapiaddress, headers):
    # 存储所有仓库地址
    allrepoaddress = []
    # 通过返回的仓库数量确认是否还需要翻页
    allreponum = 1
    # 当前打开第P页
    pg = 1
    while allreponum != 0:
        payload = {"private": "true", "per_page": "100", "page": "%s" % pg}
        res = requests.get(url=fromgitapiaddress, params=payload, headers=headers)
        allreop = res.json()
        allreponum = len(allreop)
        for i in range(allreponum):
            allrepoaddress.append("%s,%s" % (allreop[i]["id"], allreop[i]["http_url_to_repo"]))
        pg = pg + 1
    return(allrepoaddress)

# 获取仓库是否已经设置了镜像地址
def get_mirrorurl(fromgitapiaddress, repoid, headers):
    gitpageaddress = "%s/%s/remote_mirrors" % (fromgitapiaddress, repoid)
    res = requests.get(url=gitpageaddress, headers=headers)
    return(res.json())

# 设置镜像仓库地址
def set_mirrorurl(fromgitapiaddress, repoid, mirrorurl, headers):
    gitpageaddress = "%s/%s/remote_mirrors" % (fromgitapiaddress, repoid)
    payload = {"url": mirrorurl, "enabled": "true"}
    res = requests.post(url=gitpageaddress, data=payload, headers=headers)
    return(res.text)


if __name__ == "__main__":
    allrepoaddress = get_all_repo(fromgitapiaddress, headers)
    print(allrepoaddress)
    for repoinfo in allrepoaddress:
        repoid = repoinfo.split(",")[0]
        repourl = repoinfo.split(",")[1]
        # 如果您返回的repourl是http就截取10位，如果是https就截取11位（与是否有用反向代理有关）
        mirrorurl = togitapiaddress + repourl[10:]
        print(mirrorurl)
        repo_mirror_url = get_mirrorurl(fromgitapiaddress, repoid, headers)
        print(repo_mirror_url)
        if len(repo_mirror_url) == 0:
            print("源仓库ID: %s, 源仓库地址: %s, 目标仓库地址L: %s" % (repoid, repourl, mirrorurl))
            print(set_mirrorurl(fromgitapiaddress, repoid, mirrorurl, headers))
    print("所有项目数量为: %s" % len(allrepoaddress))