khelper.py

#!/usr/bin/python

from sys import argv
from subprocess import Popen, PIPE
from os import listdir
from urllib2 import urlopen
from StringIO import StringIO
from time import time

def print_usage(argv):
	print 'Khelper for kernel development'
	print 'Usage: %s [command] [arguments]' % argv[0]
	print 'test_sendemail(te) [patch_files]: send patch files only to me'
	print 'mapply(ma) [starting idx] [end idx]: apply multiple patch files which are prefixed by the number'
	print 'mapply_patchwork(mp) [url_for_patchwork_patch_list] [first patch id] [# of successive patches]: apply multiple patches which are found in patchwork'
	print 'apply_patchwork(ap) [patch id]: apply the patch which is found in patchwork'
	print 'mcheckpatch(mc) [starting idx] [end idx]: run the script checkpatch.pl for multiple patch files'

def get_usermail():
	p = Popen(['git', 'config', 'user.email'], stdout=PIPE)
	output = p.communicate()
	return output

def test_sendemail(argv):
	dest_cmd = ['--to', '--cc', '--bcc']
	email = get_usermail()[0][0:-1]
	print 'Your email address is <%s>' % email
	print 'The target patches are:'

	patches = []
	command = ['git', 'send-email', '--suppress-cc', 'all', '--to', email]
	i = 0
	for i in range(2, len(argv)):
		if argv[i] in dest_cmd:
			break
		patches.append(argv[i])
	command = command + patches
	Popen(command).communicate()

	print '\n##### Sent the patches to <%s> for testing'  % email
	print '##### Check your mailbox'

	command = ['git', 'send-email'] + patches
	for j in range(i, len(argv)):
		command.append(argv[j])

	response = raw_input('##### Is it OK to send them really? (y/N) ')
	if response == ('y' or 'Y'):
		Popen(command).communicate()

def get_patchfile_names(argv):
	i = int(argv[2])
	end = int(argv[3])
	files = listdir('.')
	ret_files = []

	files.sort()

	for file in files:
		if file.endswith('patch'):
			if file.startswith('%.04d' % i):
				ret_files.append(file)
				i += 1

		if i > end:
			break

	return ret_files

def mapply(argv):
	files = get_patchfile_names(argv)
	for file in files:
		Popen(['git', 'am', file]).communicate()

def mcheckpatch(argv):
	files = get_patchfile_names(argv)
	for file in files:
		Popen(['./scripts/checkpatch.pl', file]).communicate()

def get_patchname(filename):
	f = open(filename)
	lines = f.readlines()

	title = ''
	for i in range(0, len(lines)):
		line = lines[i]
		if line.startswith('Subject'):
			if line[9] == '[':
				if ']' not in line:
					while ']' not in line:
						title += line[:-1]
						i += 1
						line = lines[i]
				idx = line.index(']')
				title += line[idx+2:-1]
			else:
				title += line[9:-1]
			title += ' '
			i += 1
			line = lines[i]
			if not line.startswith('From'):
				title += line[1:-1]
			return title

def convert_to_filename(patchname, idx):
	replaced_chars = ['/', ':', ' ', '\'', '\"']

	replaced = False
	to_be_filename = '%.04d-' % idx
	for char in patchname:
		if char in replaced_chars:
			if replaced:
				continue
			replaced = True
			char = '-'
		else:
			replaced = False

		to_be_filename += char

	return to_be_filename[:57] + '.patch'

def apply_patchwork_id(id, idx):
	filename = 'patch-%s-%f.patch' % (id, time())
	p = Popen(['wget', '-q', '-O', filename , 'https://patchwork.kernel.org/patch/%s/mbox/' % id]).communicate()
	patchname = get_patchname(filename)
	to_be_filename = convert_to_filename(patchname, idx)
	Popen(['mv', filename , to_be_filename])
	Popen(['git', 'am', to_be_filename]).communicate()

def apply_patchwork(argv):
	id = argv[2]
	apply_patchwork_id(id, 0)

def mapply_patchwork(argv):
	url = argv[2]
	start_id = argv[3]
	len = int(argv[4])

	ids = []
	lines = StringIO(urlopen(url).read())
	for line in lines:
		if 'patch_row' in line:
			ids.append(line[20:27])

	ids.reverse()

	i = 1
	get_download = False
	for id in ids:
		if i < len+1 and (get_download or start_id == id):
			apply_patchwork_id(id, i)
			get_download = True
			i += 1

funcs = {
	'test_send-email': test_sendemail,
	'te': test_sendemail,
	'mapply': mapply,
	'ma': mapply,
	'mapply_patchwork': mapply_patchwork,
	'mp': mapply_patchwork,
	'apply_patchwork': apply_patchwork,
	'ap': apply_patchwork,
	'mcheckpatch': mcheckpatch,
	'mc': mcheckpatch,
}

def main():
	if len(argv) < 2:
		print_usage(argv)
		exit(0)

	funcs[argv[1]](argv)

if __name__ == "__main__":
	main()