forked from openlp/openlp
224 lines
9.1 KiB
Python
Executable File
224 lines
9.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
# vim: autoindent shiftwidth=4 expandtab textwidth=120 tabstop=4 softtabstop=4
|
|
|
|
###############################################################################
|
|
# OpenLP - Open Source Lyrics Projection #
|
|
# --------------------------------------------------------------------------- #
|
|
# Copyright (c) 2008-2018 OpenLP Developers #
|
|
# --------------------------------------------------------------------------- #
|
|
# This program is free software; you can redistribute it and/or modify it #
|
|
# under the terms of the GNU General Public License as published by the Free #
|
|
# Software Foundation; version 2 of the License. #
|
|
# #
|
|
# This program is distributed in the hope that it will be useful, but WITHOUT #
|
|
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or #
|
|
# FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for #
|
|
# more details. #
|
|
# #
|
|
# You should have received a copy of the GNU General Public License along #
|
|
# with this program; if not, write to the Free Software Foundation, Inc., 59 #
|
|
# Temple Place, Suite 330, Boston, MA 02111-1307 USA #
|
|
###############################################################################
|
|
|
|
"""
|
|
This script is used by developers to merge branches from launchpad.net into
|
|
other branches, which is basically what happens every time a merge proposal
|
|
is merged into trunk. This script simplifies the process and helps avoiding
|
|
merging to the wrong branch by doing some checks.
|
|
For the script to work it is assumed that the developer doing the merging
|
|
has a checkout (not branch) of the target branch.
|
|
|
|
Merge a branch
|
|
--------------
|
|
Once a branch has been approved for merging, go to the local folder with the
|
|
checkout of the target branch, copy the url from the merge proposal on launchpad
|
|
and use it like this:
|
|
|
|
script/lp-merge.py <url>
|
|
|
|
The url could look like this:
|
|
https://code.launchpad.net/~tomasgroth/openlp/doc22update4/+merge/271874
|
|
|
|
First you'll be asked whether to merge the branch from the url to the local
|
|
branch, which will be done if you answer 'y'. The script checks that the current
|
|
folder is the right one before merging.
|
|
Then you'll be asked whether to commit the changes to the branch. The script
|
|
shows the command it will run, including detected linked bugs and author. If you
|
|
wish to change it, you can chose to run 'qcommit', an bzr GUI. Note that you'll
|
|
have to install qbzr for this to work. If you choose to run qcommit the script
|
|
will print the detected bugs + author for easy copying into the GUI.
|
|
"""
|
|
import subprocess
|
|
import re
|
|
import os
|
|
from urllib.request import urlopen
|
|
from urllib.error import HTTPError
|
|
from argparse import ArgumentParser
|
|
from bs4 import BeautifulSoup
|
|
|
|
|
|
def parse_args():
|
|
"""
|
|
Parse command line arguments and return them
|
|
"""
|
|
parser = ArgumentParser(prog='lp-merge', description='A helper script to merge proposals on Launchpad')
|
|
parser.add_argument('url', help='URL to the merge proposal')
|
|
parser.add_argument('-y', '--yes-to-all', action='store_true', help='Presume yes for all queries')
|
|
parser.add_argument('-q', '--use-qcommit', action='store_true', help='Use qcommit for committing')
|
|
return parser.parse_args()
|
|
|
|
|
|
def is_merge_url_valid(url):
|
|
"""
|
|
Determine if the merge URL is valid
|
|
"""
|
|
match = re.match(r'.+?/\+merge/\d+', url)
|
|
if not url.startswith('https://code.launchpad.net/~') or match is None:
|
|
print('The url is not valid! It should look like this:\n '
|
|
'https://code.launchpad.net/~myusername/openlp/mybranch/+merge/271874')
|
|
return False
|
|
return True
|
|
|
|
|
|
def get_merge_info(url):
|
|
"""
|
|
Get all the merge information and return it as a dictionary
|
|
"""
|
|
merge_info = {}
|
|
print('Fetching merge information...')
|
|
# Try to load the page
|
|
try:
|
|
page = urlopen(url)
|
|
except HTTPError:
|
|
print('Unable to load merge URL: {}'.format(url))
|
|
return None
|
|
soup = BeautifulSoup(page.read(), 'lxml')
|
|
# Find this span tag that contains the branch url
|
|
# <span class="branch-url">
|
|
span_branch_url = soup.find('span', class_='branch-url')
|
|
if not span_branch_url:
|
|
print('Unable to find merge details on URL: {}'.format(url))
|
|
return None
|
|
merge_info['branch_url'] = span_branch_url.contents[0]
|
|
# Find this tag that describes the branch. We'll use that for commit message
|
|
# <meta name="description" content="...">
|
|
meta = soup.find('meta', attrs={"name": "description"})
|
|
merge_info['commit_message'] = meta.get('content')
|
|
# Find all tr-tags with this class. Makes it possible to get bug numbers.
|
|
# <tr class="bug-branch-summary"
|
|
bug_rows = soup.find_all('tr', class_='bug-branch-summary')
|
|
merge_info['bugs'] = []
|
|
for row in bug_rows:
|
|
id_attr = row.get('id')
|
|
merge_info['bugs'].append(id_attr[8:])
|
|
# Find target branch name using the tag below
|
|
# <div class="context-publication"><h1>Merge ... into...
|
|
div_branches = soup.find('div', class_='context-publication')
|
|
branches = div_branches.h1.contents[0]
|
|
merge_info['target_branch'] = '+branch/' + branches[(branches.find(' into lp:') + 9):]
|
|
# Find the authors email address. It is hidden in a javascript line like this:
|
|
# conf = {"status_value": "Needs review", "source_revid": "tomasgroth@yahoo.dk-20160921204550-gxduegmcmty9rljf",
|
|
# "user_can_edit_status": false, ...
|
|
script_tag = soup.find('script', attrs={"id": "codereview-script"})
|
|
content = script_tag.contents[0]
|
|
start_pos = content.find('source_revid') + 16
|
|
pattern = re.compile('.*\w-\d\d\d\d\d+')
|
|
match = pattern.match(content[start_pos:])
|
|
merge_info['author_email'] = match.group()[:-15]
|
|
# Launchpad doesn't supply the author's true name, so we'll just grab whatever they use for display on LP
|
|
a_person = soup.find('div', id='registration').find('a', 'person')
|
|
merge_info['author_name'] = a_person.contents[0]
|
|
return merge_info
|
|
|
|
|
|
def do_merge(merge_info, yes_to_all=False):
|
|
"""
|
|
Do the merge
|
|
"""
|
|
# Check that we are in the right branch
|
|
bzr_info_output = subprocess.check_output(['bzr', 'info'])
|
|
if merge_info['target_branch'] not in bzr_info_output.decode():
|
|
print('ERROR: It seems you are not in the right folder...')
|
|
return False
|
|
# Merge the branch
|
|
if yes_to_all:
|
|
can_merge = True
|
|
else:
|
|
user_choice = input('Merge ' + merge_info['branch_url'] + ' into local branch? (y/N/q): ').lower()
|
|
if user_choice == 'q':
|
|
return False
|
|
can_merge = user_choice == 'y'
|
|
if can_merge:
|
|
print('Merging...')
|
|
subprocess.call(['bzr', 'merge', merge_info['branch_url']])
|
|
return True
|
|
|
|
|
|
def qcommit(commit_message, author_name, author_email, bugs=[]):
|
|
"""
|
|
Use qcommit to make the commit
|
|
"""
|
|
# Setup QT workaround to make qbzr look right on my box
|
|
my_env = os.environ.copy()
|
|
my_env['QT_GRAPHICSSYSTEM'] = 'native'
|
|
# Print stuff that kan be copy/pasted into qbzr GUI
|
|
if bugs:
|
|
print('These bugs can be copy/pasted in: ' + ' '.join(['lp:{}'.format(bug) for bug in bugs]))
|
|
print('The authors email is: {} <{}>'.format(author_name, author_email))
|
|
# Run qcommit
|
|
subprocess.call(['bzr', 'qcommit', '-m', commit_message], env=my_env)
|
|
|
|
|
|
def do_commit(merge_info, yes_to_all=False, use_qcommit=False):
|
|
"""
|
|
Actually do the commit
|
|
"""
|
|
if use_qcommit:
|
|
qcommit(merge_info['commit_message'], merge_info['author_name'], merge_info['author_email'],
|
|
merge_info.get('bugs'))
|
|
return True
|
|
# Create commit command
|
|
commit_command = ['bzr', 'commit']
|
|
if 'bugs' in merge_info:
|
|
commit_command.extend(['--fixes=lp:{}'.format(bug) for bug in merge_info['bugs']])
|
|
commit_command.extend(['-m', merge_info['commit_message'],
|
|
'--author', '{author_name} <{author_email}>'.format(**merge_info)])
|
|
if yes_to_all:
|
|
can_commit = True
|
|
else:
|
|
print('About to run the bzr command below:\n')
|
|
print(' '.join(commit_command))
|
|
user_choice = input('Run the command (y), use qcommit (q) or cancel (C): ').lower()
|
|
if user_choice == 'q':
|
|
qcommit(merge_info['commit_message'], merge_info['author_name'], merge_info['author_email'],
|
|
merge_info.get('bugs'))
|
|
return True
|
|
can_commit = user_choice == 'y'
|
|
if can_commit:
|
|
print('Committing...')
|
|
subprocess.call(commit_command)
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
|
|
def main():
|
|
"""
|
|
Run the script
|
|
"""
|
|
args = parse_args()
|
|
if not is_merge_url_valid(args.url):
|
|
exit(1)
|
|
merge_info = get_merge_info(args.url)
|
|
if not merge_info:
|
|
exit(2)
|
|
if not do_merge(merge_info, args.yes_to_all):
|
|
exit(3)
|
|
if not do_commit(merge_info, args.yes_to_all, args.use_qcommit):
|
|
exit(4)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|