import urllib.request
import gitlab

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

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

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


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


def main():
 for project_id in project_id_list:
  project = gl.projects.get(project_id)
  print(' >>> 項目ID :: [%s], 項目名稱 :: [%s] ' % (str(project.id), project_name))
  for target in target_branches:
   print(" >>> 正在合并項目 :: [%s] 的 [%s] 分支到 [%s] 分支" % (project_name, source_branch, target))
   # mr合并請求的對象
   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)
             })

    # 接受合并請求
    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))

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

if __name__ == "__main__":
 main()