You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
187 lines
7.0 KiB
187 lines
7.0 KiB
#! /usr/bin/python2.7
|
|
|
|
# Copyright (C) 2012 Canonical Ltd.
|
|
# Author: Colin Watson <cjwatson@ubuntu.com>
|
|
|
|
# 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 3 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, see <http://www.gnu.org/licenses/>.
|
|
|
|
"""Branch a set of Ubuntu seeds for the next release."""
|
|
|
|
from __future__ import print_function
|
|
|
|
from optparse import OptionParser
|
|
import os
|
|
import re
|
|
import time
|
|
import subprocess
|
|
try:
|
|
from urllib.parse import urlparse
|
|
except ImportError:
|
|
from urlparse import urlparse
|
|
|
|
from launchpadlib.launchpad import Launchpad
|
|
from enum import Enum
|
|
|
|
class VCS(Enum):
|
|
Git = 1
|
|
Bazaar = 2
|
|
|
|
@staticmethod
|
|
def detect_vcs(source):
|
|
if os.path.exists(os.path.join(source, ".git")):
|
|
return VCS.Git
|
|
elif os.path.exists(os.path.join(source, ".bzr")):
|
|
return VCS.Bazaar
|
|
else:
|
|
return None
|
|
|
|
|
|
def remote_bzr_branch(source):
|
|
# TODO: should really use bzrlib instead
|
|
info = subprocess.check_output(
|
|
["bzr", "info", source], universal_newlines=True)
|
|
for line in info.splitlines():
|
|
if "checkout of branch:" in line:
|
|
return line.split(": ")[1].rstrip("/")
|
|
else:
|
|
raise Exception("Unable to find remote branch for %s" % source)
|
|
|
|
|
|
def remote_git_repository(source, srcbranch):
|
|
fullbranch = subprocess.check_output(
|
|
["git", "rev-parse", "--symbolic-full-name",
|
|
srcbranch + "@{upstream}"],
|
|
universal_newlines=True, cwd=source)
|
|
return subprocess.check_output(
|
|
["git", "ls-remote", "--get-url", fullbranch.split("/")[2]],
|
|
universal_newlines=True, cwd=source).rstrip("\n")
|
|
|
|
|
|
def lp_branch(options, url):
|
|
return options.launchpad.branches.getByUniqueName(
|
|
unique_name=urlparse(url).path.lstrip("/"))
|
|
|
|
|
|
def branch(options, collection):
|
|
source = "%s.%s" % (collection, options.source_series)
|
|
dest = "%s.%s" % (collection, options.dest_series)
|
|
vcs = VCS.detect_vcs(source)
|
|
if vcs:
|
|
if vcs is VCS.Bazaar:
|
|
subprocess.check_call(["bzr", "up", source])
|
|
remote_source = remote_bzr_branch(source)
|
|
remote_dest = os.path.join(os.path.dirname(remote_source), dest)
|
|
subprocess.check_call(["bzr", "branch", source, dest])
|
|
subprocess.check_call(["bzr", "push", "-d", dest, remote_dest])
|
|
subprocess.check_call(["bzr", "bind", ":push"], cwd=dest)
|
|
|
|
lp_source = lp_branch(options, remote_source)
|
|
lp_source.lifecycle_status = "Mature"
|
|
lp_source.lp_save()
|
|
|
|
lp_dest = lp_branch(options, remote_dest)
|
|
lp_dest.lifecycle_status = "Development"
|
|
lp_dest.lp_save()
|
|
elif vcs is VCS.Git:
|
|
subprocess.check_call(["git", "fetch"], cwd=source)
|
|
subprocess.check_call(["git", "reset", "--hard", "FETCH_HEAD"], cwd=source)
|
|
os.rename(source, dest)
|
|
subprocess.check_call(["git", "checkout", "-b", options.dest_series], cwd=dest)
|
|
|
|
re_include_source = re.compile(
|
|
r"^(include )(.*)\.%s" % options.source_series)
|
|
new_lines = []
|
|
message = []
|
|
with open(os.path.join(dest, "STRUCTURE")) as structure:
|
|
for line in structure:
|
|
match = re_include_source.match(line)
|
|
if match:
|
|
new_lines.append(re_include_source.sub(
|
|
r"\1\2.%s" % options.dest_series, line))
|
|
message.append(
|
|
"%s.%s -> %s.%s" %
|
|
(match.group(2), options.source_series,
|
|
match.group(2), options.dest_series))
|
|
else:
|
|
new_lines.append(line)
|
|
if message:
|
|
with open(os.path.join(dest, "STRUCTURE.new"), "w") as structure:
|
|
for line in new_lines:
|
|
print(line, end="", file=structure)
|
|
os.rename(
|
|
os.path.join(dest, "STRUCTURE.new"),
|
|
os.path.join(dest, "STRUCTURE"))
|
|
if vcs is VCS.Bazaar:
|
|
subprocess.check_call(
|
|
["bzr", "commit", "-m", "; ".join(message)], cwd=dest)
|
|
elif vcs is VCS.Git:
|
|
subprocess.check_call(["git", "add", "STRUCTURE"], cwd=dest)
|
|
subprocess.check_call(
|
|
["git", "commit", "-m", "; ".join(message)], cwd=dest)
|
|
subprocess.check_call(
|
|
["git", "push", "origin", options.dest_series], cwd=dest)
|
|
|
|
remote = remote_git_repository(dest, options.source_series)
|
|
if "git.launchpad.net" in remote:
|
|
lp_git_repo = options.launchpad.git_repositories.getByPath(
|
|
path=urlparse(remote).path.lstrip("/"))
|
|
new_ref = "refs/heads/%s" % options.dest_series
|
|
# Sometimes it takes LP a while to notice the new ref
|
|
for i in range(10):
|
|
if lp_git_repo.getRefByPath(path=new_ref):
|
|
lp_git_repo.default_branch = new_ref
|
|
lp_git_repo.lp_save()
|
|
break
|
|
time.sleep(1)
|
|
else:
|
|
raise Exception(
|
|
"Was unable to set default_branch of %s after "
|
|
"multiple retries - proceed manually." % remote)
|
|
else:
|
|
raise Exception(
|
|
"Git remote URL must be on git.launchpad.net.")
|
|
|
|
|
|
def main():
|
|
parser = OptionParser(usage="usage: %prog [options] collection ...")
|
|
parser.add_option(
|
|
"-l", "--launchpad", dest="launchpad_instance", default="production")
|
|
parser.add_option(
|
|
"--source-series",
|
|
help="source series (default: current stable release)")
|
|
parser.add_option(
|
|
"--dest-series",
|
|
help="destination series (default: series in pre-release freeze)")
|
|
options, args = parser.parse_args()
|
|
if not args:
|
|
parser.error("You must specify at least one seed collection.")
|
|
|
|
options.launchpad = Launchpad.login_with(
|
|
"branch-seeds", options.launchpad_instance, version="devel")
|
|
|
|
distro = options.launchpad.distributions["ubuntu"]
|
|
if options.source_series is None:
|
|
options.source_series = [
|
|
series.name for series in distro.series
|
|
if series.status == "Current Stable Release"][0]
|
|
if options.dest_series is None:
|
|
options.dest_series = [
|
|
series.name for series in distro.series
|
|
if series.status == "Pre-release Freeze"][0]
|
|
|
|
for collection in args:
|
|
branch(options, collection)
|
|
|
|
|
|
main()
|