mirror of
https://git.launchpad.net/ubuntu-dev-tools
synced 2025-03-13 08:01:09 +00:00
148 lines
4.7 KiB
Python
Executable File
148 lines
4.7 KiB
Python
Executable File
#!/usr/bin/python
|
|
# -*- coding: utf-8 -*-
|
|
#
|
|
# Copyright (C) 2007 Canonical Ltd.
|
|
# Created by Daniel Holbach <daniel.holbach@ubuntu.com>
|
|
# Modified by Jonathan Patrick Davies <jpds@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.
|
|
#
|
|
# 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.
|
|
#
|
|
# See file /usr/share/common-licenses/GPL-3 for more details.
|
|
#
|
|
# ##################################################################
|
|
#
|
|
# This script is used to checkout or branch all the Bazaar branches
|
|
# of a Launchpad team.
|
|
#
|
|
|
|
import os
|
|
import re
|
|
import subprocess
|
|
import sys
|
|
import urllib2
|
|
from optparse import OptionParser
|
|
|
|
def main():
|
|
usage = "Usage: %prog [-d <directory>] -t <team> [-o <operation>]"
|
|
usage += "\nUsage: %prog <team>"
|
|
optParser = OptionParser(usage)
|
|
|
|
# Our options.
|
|
optParser.add_option("-d", "--directory", action = "store", type = "string",
|
|
dest = "directory", default = os.getcwd(),
|
|
help = "Directory to download branches to.")
|
|
optParser.add_option("-t", "--team", action = "store", type = "string",
|
|
dest = "lpteam", help = "Launchpad team to download branches from.")
|
|
optParser.add_option("-o", "--operation", action = "store", type = "string",
|
|
dest = "operation", default = "branch",
|
|
help = "Whether to branch or checkout the Bazaar branches. May be " \
|
|
"either 'branch' or 'checkout'.")
|
|
|
|
(options, args) = optParser.parse_args()
|
|
|
|
# Parse our options.
|
|
if len(args) != 1 and options.lpteam == None:
|
|
optParser.error("No team has been specified.")
|
|
|
|
# Launchpad team setting.
|
|
if options.lpteam:
|
|
team = str(options.lpteam).lower()
|
|
if args:
|
|
team = str(args[0]).lower()
|
|
|
|
directory = options.directory
|
|
|
|
# Dictionary settings.
|
|
if not os.path.isdir(directory): # Check that it is a directory.
|
|
optParser.error("%s is not a valid directory." % directory)
|
|
|
|
# Type of Bazaar operation to perform.
|
|
operation_type = str(options.operation).lower()
|
|
|
|
# Got an argument, check if it is valid.
|
|
if operation_type not in ("branch", "checkout"):
|
|
optParser.error("Invalid operation '%s' for '-o' flag." % \
|
|
operation_type)
|
|
|
|
|
|
# Fetch our current directory to return to later.
|
|
pwd = os.getcwd()
|
|
|
|
# Change to the specified directory.
|
|
os.chdir(directory)
|
|
|
|
# Try to open the teams code page.
|
|
try:
|
|
sock = urllib2.urlopen("https://code.launchpad.net/~%s" % team)
|
|
except urllib2.HTTPError:
|
|
print >> sys.stderr, "The page https://code.launchpad.net/~%s does " \
|
|
"not exist." % team
|
|
print >> sys.stderr, "Perhaps invalid team name?"
|
|
sys.exit(1)
|
|
|
|
branch_list_page = sock.read()
|
|
sock.close()
|
|
|
|
branch_page_urls_regex = r'.*href="/(~%s/.*)">lp:.*' % team
|
|
branch_page_urls = re.findall(branch_page_urls_regex, branch_list_page)
|
|
|
|
# Check the team actually has branches.
|
|
if len(branch_page_urls) == 0:
|
|
print "The team '%s' does not have any branches on Launchpad." % team
|
|
sys.exit(0)
|
|
|
|
print "Downloading all branches for the team '%s'. This may take some " \
|
|
"time." % team
|
|
|
|
try:
|
|
os.makedirs(team)
|
|
except:
|
|
pass
|
|
|
|
os.chdir(team)
|
|
|
|
for url in branch_page_urls:
|
|
sock = urllib2.urlopen("https://code.launchpad.net/%s" % url)
|
|
branch_page = sock.read()
|
|
sock.close()
|
|
|
|
branch_url_regex = r'<dd>bzr branch <span.*>lp:(.*)</span></dd>'
|
|
branch_url = re.findall(branch_url_regex, branch_page)
|
|
|
|
if branch_url[0]:
|
|
print "Downloading branch: lp:%s (%s)." % (branch_url[0], url)
|
|
product = url.split("/")[-2]
|
|
branch_nick = url.split("/")[-1]
|
|
else:
|
|
continue
|
|
|
|
print branch_nick, product, os.getcwd()
|
|
if not os.path.exists(product):
|
|
os.makedirs(product)
|
|
os.chdir(product)
|
|
|
|
if not os.path.exists(branch_nick):
|
|
subprocess.call(["bzr", operation_type, "lp:%s" % url])
|
|
else:
|
|
os.chdir(branch_nick)
|
|
subprocess.call(["bzr", "merge", "--pull", "--remember"])
|
|
os.chdir(os.path.join(directory, team))
|
|
|
|
os.chdir(pwd)
|
|
sys.exit(0)
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
main()
|
|
except KeyboardInterrupt:
|
|
print "Operation was interrupted by user."
|