add release script

bench/cli.py

@@ -16,6 +16,7 @@ from .app import get_app as _get_app
 from .app import new_app as _new_app
 from .app import pull_all_apps
 from .config import generate_nginx_config, generate_supervisor_config
+from .release import release
 import os
 import sys
 import logging
@@ -174,6 +175,14 @@ def _prime_wheel_cache():
 	"Update wheel cache"
 	prime_wheel_cache(bench='.')
 
+@click.command('release')
+@click.argument('app', type=click.Choice(['frappe', 'erpnext', 'shopping_cart']))
+@click.argument('bump-type', type=click.Choice(['major', 'minor', 'patch']))
+def _release(app, bump_type):
+	"Release app (internal to the Frappe team)"
+	repo = os.path.join('apps', app)
+	release(repo, bump_type)
+
 ## Setup
 @click.group()
 def setup():
@@ -309,3 +318,4 @@ bench.add_command(shell)
 bench.add_command(_backup_all_sites)
 bench.add_command(_backup_site)
 bench.add_command(_prime_wheel_cache)
+bench.add_command(_release)

bench/release.py

@@ -0,0 +1,164 @@
+#! env python
+import json
+import os
+import semantic_version
+import git
+import json
+import requests
+import getpass
+import argparse
+import re
+from requests.auth import HTTPBasicAuth
+import requests.exceptions
+from time import sleep
+
+github_username = None
+github_password = None
+
+repo_map = {
+	'frappe': 'frappe',
+	'erpnext': 'erpnext',
+	'shopping_cart': 'shopping-cart'
+}
+
+def create_release(repo_path, version, remote='origin', develop_branch='develop', master_branch='master'):
+	repo = git.Repo(repo_path)
+	g = repo.git
+	g.checkout(master_branch)
+	g.merge(develop_branch, '--no-ff')
+	tag_name = 'v' + version
+	repo.create_tag(tag_name, message='Release {}'.format(version))
+	g.checkout(develop_branch)
+	g.merge(master_branch)
+	return tag_name
+
+def push_release(repo_path):
+	repo = git.Repo(repo_path)
+	g = repo.git
+	print g.push('upstream', 'master:master', 'develop:develop', '--tags')
+
+def create_github_release(owner, repo, tag_name, log, gh_username=None, gh_password=None):
+	global github_username, github_password
+	if not (gh_username and gh_password):
+		if not (github_username and github_password):
+			raise Exception, "No credentials"
+		gh_username = github_username
+		gh_password = github_password
+	data = {
+		'tag_name': tag_name,
+		'target_commitish': 'master',
+		'name': 'Release ' + tag_name,
+		'body': log,
+		'draft': 'false',
+		'prerelease': 'false'
+	}
+	for i in xrange(3):
+		try:
+                        r = requests.post('https://api.github.com/repos/{owner}/{repo}/releases'.format(owner=owner, repo=repo), auth=HTTPBasicAuth(gh_username, gh_password), data=json.dumps(data))
+                        r.raise_for_status()
+			break
+		except requests.exceptions.HTTPError:
+			print 'request failed, retrying....'
+			sleep(3*i + 1)
+			if i !=2:
+				continue
+			else:
+				raise
+	return r
+
+def update_branch(repo_path, branch, remote='origin'):
+	repo = git.Repo(repo_path)
+	g = repo.git
+	g.fetch(remote)
+	g.checkout(branch)
+	g.merge(remote+'/'+branch)
+
+def get_bumped_version(version, bump_type):
+	v = semantic_version.Version(version)
+	if bump_type == 'minor':
+		v.minor += 1
+		v.patch = 0
+	elif bump_type == 'major':
+		v.major += 1
+		v.minor = 0
+		v.patch = 0
+	elif bump_type == 'patch':
+		v.patch += 1
+	return unicode(v)
+
+def commit_changes(repo_path, version):
+	repo = git.Repo(repo_path)
+	repo_name = os.path.basename(repo_path)
+	repo.index.add(['setup.py'])
+	repo.index.add([os.path.join(repo_name, '__version__.py')])
+	repo.index.add([os.path.join(repo_name, 'hooks.py')])
+	repo.index.commit('bumped to version {}'.format(version))
+
+def set_filename_version(filename, version_number, pattern):
+	changed = []
+
+	def inject_version(match):
+		before, old, after = match.groups()
+		changed.append(True)
+		return before + version_number + after
+
+	with open(filename) as f:
+		contents = re.sub(r"^(\s*%s\s*=\s*['\\\"])(.+?)(['\"])(?sm)" % pattern,
+				inject_version, f.read())
+
+	if not changed:
+		raise Exception('Could not find %s in %s', pattern, filename)
+
+	with open(filename, 'w') as f:
+		f.write(contents)
+
+def set_setuppy_version(repo, version):
+	set_filename_version(os.path.join(repo, 'setup.py'), version, 'version')
+
+def set_versionpy_version(repo, version):
+	set_filename_version(os.path.join(repo, os.path.basename(repo),'__version__.py'), version, '__version__')
+
+def set_hooks_version(repo, version):
+	set_filename_version(os.path.join(repo, os.path.basename(repo),'hooks.py'), version, 'app_version')
+
+def set_version(repo, version):
+	set_setuppy_version(repo, version)
+	set_versionpy_version(repo, version)
+	set_hooks_version(repo, version)
+
+def get_current_version(repo):
+	filename = os.path.join(repo, 'setup.py')
+	with open(filename) as f:
+		contents = f.read()
+		match = re.search(r"^(\s*%s\s*=\s*['\\\"])(.+?)(['\"])(?sm)" % 'version',
+				contents)
+		return match.group(2)
+
+def bump_repo(repo, bump_type):
+		update_branch(repo, 'master', remote='upstream')
+		update_branch(repo, 'develop', remote='upstream')
+		git.Repo(repo).git.checkout('develop')
+		current_version = get_current_version(repo)
+		new_version = get_bumped_version(current_version, bump_type)
+		set_version(repo, new_version)
+		return new_version
+
+def bump(repo, bump_type):
+	assert bump_type in ['minor', 'major', 'patch']
+	new_version = bump_repo(repo, bump_type)
+	commit_changes(repo, new_version)
+	tag_name = create_release(repo, new_version)
+	#push_release(repo)
+	#create_github_release('frappe', repo, tag_name, '')
+	print 'Released {tag} for {repo}'.format(tag=tag_name, repo=repo)
+
+def release(repo, bump_type):
+	global github_username, github_password
+	github_username = raw_input('username:')
+	github_password = getpass.getpass()
+	r = requests.get('https://api.github.com/user', auth=HTTPBasicAuth(github_username, github_password))
+	r.raise_for_status()
+	bump(repo, bump_type)
+
+if __name__ == "__main__":
+	main()