#!/usr/bin/python3
#coding=utf-8
# 自動合并項目dev分支到其他分支，當合并失敗刪除merge請求
import sys
import datetime
import urllib.request
import gitlab


#項目名和項目ID 
project_name = "python-jmmei"
project_list=[265]

# 生成gitlab對象
gitlab_url = 'http://gitlab.baidu.com'
token = 'xxxxxxxxxxxx'
old_branch="hotfix"
new_branches=["test","uat","uat_match"]
#驗證登錄
gl = gitlab.Gitlab(gitlab_url, token)


for project_id in project_list:
  project = gl.projects.get(project_id)
  print(project.id)
  for proj in new_branches:
    print("正在合并項目：%s的%s分支到%s分支"%(project_name,old_branch,proj))
    # mr合并請求的對象
    mr = None
    try:
      # 創(chuàng)建mr
      mr = project.mergerequests.create({'source_branch': old_branch,
                      'target_branch': proj,
                      'title': "%s to %s ---"%(old_branch,proj) + datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
                     })
    
      # 接受合并請求
      url = "%s/api/v3/projects/%s/merge_request/%s/merge"%(gitlab_url,project.id, mr.id)
      headers = {
        "PRIVATE-TOKEN":token
      }
      req = urllib.request.Request(url=url,headers=headers,method="PUT")
      resp = urllib.request.urlopen(req)
      # print(resp)
      print("合并到分支%s成功,結束...\r\n"%(proj))
    except Exception as e:
      print("合并出錯,可能有沖突未解決或者%s分支并沒有更新,異常信息：\r\n"%(old_branch))
      print(e)
      # 把剛創(chuàng)建的mr請求刪除
      #v4版本支持 project.mergerequests.delete(mr.id)
      mr.delete()