import urllib.request
import gitlab

# ========================================配置開始=================================================

# ⚠️ 項(xiàng)目名, 無實(shí)際意義
project_name = "xxxx"
# ⚠️ 代表多個(gè)項(xiàng)目 ID, 平常一個(gè)項(xiàng)目即可, 多個(gè)項(xiàng)目需要保證 source_branch、target_branches 保持一致
project_id_list = ['xxxx']
# ⚠️ 生成 gitlab 對(duì)象
gitlab_url = 'xxxx'
# ⚠️ 需要生成自己的令牌 token
token = 'xxxx'
# ⚠️ 將 source_branch 分支 merge target_branches 分支集上
source_branch = 'test'
target_branches = ['master']

# =========================================配置結(jié)束================================================


# 驗(yàn)證登錄
gl = gitlab.Gitlab(gitlab_url, token)


def main():
 for project_id in project_id_list:
  project = gl.projects.get(project_id)
  print(' >>> 項(xiàng)目ID :: [%s], 項(xiàng)目名稱 :: [%s] ' % (str(project.id), project_name))
  for target in target_branches:
   print(" >>> 正在合并項(xiàng)目 :: [%s] 的 [%s] 分支到 [%s] 分支" % (project_name, source_branch, target))
   # mr合并請(qǐng)求的對(duì)象
   mr = None
   try:
    # 創(chuàng)建 merge request
    mr = project.mergerequests.create({'source_branch': source_branch,
             'target_branch': target,
             'title': "Merge branch '%s' into '%s'" % (
              source_branch, target)
             })

    # 接受合并請(qǐng)求
    url = '%s/api/v4/projects/%s/merge_requests/%s/merge' % (gitlab_url, project.id, mr.iid)
    print(' >>> Merge url :: [%s], title :: [%s] ' % (url, mr.title))

    # 個(gè)人/項(xiàng)目訪問令牌, 添加至請(qǐng)求頭
    headers = {"PRIVATE-TOKEN": token}
    req = urllib.request.Request(url=url, headers=headers, method="PUT")
    resp = urllib.request.urlopen(req)
    if resp.code == 200:
     print(" >>> 合并到目標(biāo)分支 [%s] 成功, 結(jié)束... \r\n" % target)
    else:
     mr.delete()
   except Exception as e:
    print(" >>> 合并出錯(cuò), 可能有沖突未解決或者 [%s] 分支并沒有更新, 異常信息:: \r\n" % source_branch, str(e))
    # 把創(chuàng)建的 merge 請(qǐng)求刪除
    mr.delete()

if __name__ == "__main__":
 main()