gitlabreleaseuploader/gitlabreleaseuploader.py

#!/usr/bin/python

import requests
import json
import argparse
import sys


parser = argparse.ArgumentParser(description='Gitlab Release Uploader')
parser.add_argument('--privateToken', '-p',
                    help='Private token to access Gitlab', required=True)
parser.add_argument('--projectId', '-i',
                    help='ProjectID of the related project', required=True)
parser.add_argument('--projectUrl', '-u',
                    help='URL of the related project at Gitlab', required=True)
parser.add_argument('--file', '-f',
                    help='File to be released', required=True)
parser.add_argument('--releaseName', '-n',
                    help='Name of the release', required=False,
                    default='')
parser.add_argument('--releaseTag', '-t',
                    help='Tag of the release in the repo',
                    required=False,
                    default='')
parser.add_argument('--releaseTagTarget', '-T',
                    help='Commit or branch the tag should point to',
                    required=False,
                    default='')
parser.add_argument('--createReleaseTag', '-c',
                    help='Shall the release be created here',
                    required=False,
                    action='store_true',
                    default=False)
parser.add_argument('--description', '-d',
                    help='Description of the release', required=False,
                    default='')
parser.add_argument('--releaseInfoFile', '-F',
                    help='File containing JSON object with release info '
                         '(release tag, create release tag, description',
                    required=False,
                    default='')
parser.add_argument('--instanceUrl', '-I',
                    help='URL of your gitlab instance', required=False,
                    default='https://gitlab.com')
parser.add_argument('--verbose', '-v',
                    help='verbose output',
                    required=False,
                    action='store_true',
                    default=False)
args = parser.parse_args()

privateToken = args.privateToken
projectId = args.projectId
projectUrl = args.projectUrl
fileToUpload = args.file
releaseName = args.releaseName
releaseTag = args.releaseTag
releaseTagTarget = args.releaseTagTarget
releaseDescription = args.description
instanceUrl = args.instanceUrl
createReleaseTag = args.createReleaseTag
releaseInfoFilename = args.releaseInfoFile
verbose = args.verbose
releaseInfo = {}

if (releaseInfoFilename):
    with open(releaseInfoFilename, 'r') as releaseInfoFile:
        releaseInfo = releaseInfoFile.read()
    releaseInfo = json.loads(releaseInfo)
    if 'releaseName' in releaseInfo:
        releaseName = releaseInfo['releaseName']
    if 'releaseTag' in releaseInfo:
        releaseTag = releaseInfo['releaseTag']
    if 'releaseTagTarget' in releaseInfo:
        releaseTagTarget = releaseInfo['releaseTagTarget']
    if 'createReleaseTag' in releaseInfo:
        createReleaseTag = (releaseInfo['createReleaseTag'] in
                            ('true', 'True'))
    if 'description' in releaseInfo:
        releaseDescription = releaseInfo['description']

if releaseName == '':
    raise Exception('No release name given')
if releaseTag == '':
    raise Exception('No release tag given')
if releaseTagTarget == '':
    raise Exception('No release tag target given')
if releaseDescription == '':
    raise Exception('No release description given')


# --- upload the file
uploadUrl = "%s/api/v4/projects/%s/uploads" % (instanceUrl, projectId)
headers = {"PRIVATE-TOKEN": privateToken}
files = {"file": open(fileToUpload, 'rb')}

uploadResult = requests.post(uploadUrl, files=files, headers=headers)

if uploadResult.status_code != 201:
    print(uploadResult)
    print(uploadResult.text)
    raise Exception('Unable to upload file to Gitlab')

if verbose:
    print(uploadResult)
    print(uploadResult.text)
print('File successfully uploaded')

uploadResultJson = json.loads(uploadResult.text)

# --- create release tag
if createReleaseTag:
    createReleaseTagUrl = ("%s/api/v4//projects/%s/repository/tags" %
                           (instanceUrl, projectId))
    headers = {"PRIVATE-TOKEN": privateToken,
               "Content-Type": "application/json"}

    payloadCreateReleaseTag = {
        "tag_name": releaseTag,
        "id": projectId,
        "ref": releaseTagTarget,
        "message": "Tag for release %s" % releaseName
    }

    createReleaseTagResult = requests.post(createReleaseTagUrl,
                                           headers=headers,
                                           data=json.dumps(payloadCreateReleaseTag))

    if createReleaseTagResult.status_code != 201:
        print(createReleaseTagResult)
        print(createReleaseTagResult.text)
        raise Exception('Unable to create release tag')

    if verbose:
        print(createReleaseTagResult)
        print(createReleaseTagResult.text)
    print('Tag successfully created')


# --- create release
createReleaseUrl = "%s/api/v4/projects/%s/releases" % (instanceUrl, projectId)
headers = {"PRIVATE-TOKEN": privateToken, "Content-Type": "application/json"}
fullUrl = projectUrl + uploadResultJson['url']

payloadCreateRelease = {
  "name": releaseName,
  "tag_name": releaseTag,
  "description": releaseDescription,
  "assets": {
    "links": [
      {
        "name": fileToUpload,
        "url": fullUrl
      }
    ]
  }
}

createReleaseResult = requests.post(createReleaseUrl, headers=headers,
                                    data=json.dumps(payloadCreateRelease))

if createReleaseResult.status_code != 201:
    print(createReleaseResult)
    print(createReleaseResult.text)
    raise Exception('Unable to create release')

if verbose:
    print(createReleaseResult)
    print(createReleaseResult.text)
print('Release successfully created')