1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
# Copyright 2011 Canonical Ltd. This software is licensed under the
# GNU Affero General Public License version 3 (see the file LICENSE).
__metaclass__ = type
__all__ = [
'AlreadyRegisteredError',
'BranchPopupWidget',
'NoProductError',
]
import sys
from lazr.uri import (
InvalidURIError,
URI,
)
from zope.app.form.interfaces import ConversionError
from zope.component import getUtility
from canonical.launchpad.webapp.interfaces import ILaunchBag
from canonical.launchpad.webapp.menu import structured
from lp.app.browser.tales import BranchFormatterAPI
from lp.app.validators import LaunchpadValidationError
from lp.app.widgets.popup import VocabularyPickerWidget
from lp.code.enums import BranchType
from lp.code.interfaces.branch import IBranch
from lp.code.interfaces.branchlookup import IBranchLookup
from lp.code.interfaces.branchnamespace import get_branch_namespace
class AlreadyRegisteredError(Exception):
"""Raised when we try to register an already-registered branch."""
class NoProductError(Exception):
"""Raised when we need a product and can't find one."""
class BranchPopupWidget(VocabularyPickerWidget):
"""Custom popup widget for choosing branches."""
displayWidth = '35'
def getBranchNameFromURL(self, url):
"""Return a branch name based on `url`.
The name is based on the last path segment of the URL. If there is
already another branch of that name on the product, then we'll try to
find a unique name by appending numbers.
"""
return URI(url).ensureNoSlash().path.split('/')[-1]
def getPerson(self):
"""Return the person in the context, if any."""
return getUtility(ILaunchBag).user
def getProduct(self):
"""Return the product in the context, if there is one."""
return getUtility(ILaunchBag).product
def makeBranchFromURL(self, url):
"""Make a mirrored branch for `url`.
The product and owner of the branch are derived from information in
the launchbag. The name of the branch is derived from the last segment
of the URL and is guaranteed to be unique for the product.
:param url: The URL to mirror.
:return: An `IBranch`.
"""
# XXX: JonathanLange 2008-12-08 spec=package-branches: This method
# needs to be rewritten to get the sourcepackage and distroseries out
# of the launch bag.
url = unicode(URI(url).ensureNoSlash())
if getUtility(IBranchLookup).getByUrl(url) is not None:
raise AlreadyRegisteredError('Already a branch for %r' % url)
# Make sure the URL is valid.
IBranch['url'].validate(url)
product = self.getProduct()
if product is None:
raise NoProductError("Could not find product in LaunchBag.")
owner = self.getPerson()
name = self.getBranchNameFromURL(url)
namespace = get_branch_namespace(person=owner, product=product)
branch = namespace.createBranchWithPrefix(
BranchType.MIRRORED, name, owner, url=url)
branch.requestMirror()
self.request.response.addNotification(
structured('Registered %s' %
BranchFormatterAPI(branch).link(None)))
return branch
def _toFieldValue(self, form_input):
try:
return super(BranchPopupWidget, self)._toFieldValue(form_input)
except ConversionError, exception:
# Save the initial error so we can re-raise it later.
exc_class, exc_obj, exc_tb = sys.exc_info()
# Try to register a branch, assuming form_input is a URL.
try:
return self.makeBranchFromURL(form_input)
except (InvalidURIError, NoProductError, AlreadyRegisteredError,
LaunchpadValidationError):
# If it's not a URL or we can't figure out a product, then we
# re-raise the initial error.
raise exc_class, exc_obj, exc_tb
|