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
|
# Copyright 2009 Canonical Ltd. This software is licensed under the
# GNU Affero General Public License version 3 (see the file LICENSE).
# pylint: disable-msg=E0611,W0212
"""Database classes for linking specifications and branches."""
__metaclass__ = type
__all__ = [
"SpecificationBranch",
"SpecificationBranchSet",
]
from sqlobject import (
ForeignKey,
IN,
StringCol,
)
from zope.interface import implements
from canonical.database.constants import UTC_NOW
from canonical.database.datetimecol import UtcDateTimeCol
from canonical.database.sqlbase import SQLBase
from lp.blueprints.interfaces.specificationbranch import (
ISpecificationBranch,
ISpecificationBranchSet,
)
from lp.registry.interfaces.person import validate_public_person
class SpecificationBranch(SQLBase):
"""See `ISpecificationBranch`."""
implements(ISpecificationBranch)
datecreated = UtcDateTimeCol(notNull=True, default=UTC_NOW)
specification = ForeignKey(dbName="specification",
foreignKey="Specification", notNull=True)
branch = ForeignKey(dbName="branch", foreignKey="Branch", notNull=True)
registrant = ForeignKey(
dbName='registrant', foreignKey='Person',
storm_validator=validate_public_person, notNull=True)
class SpecificationBranchSet:
"""See `ISpecificationBranchSet`."""
implements(ISpecificationBranchSet)
def getSpecificationBranchesForBranches(self, branches, user):
"""See `ISpecificationBranchSet`."""
branch_ids = [branch.id for branch in branches]
if not branch_ids:
return []
# When specification gain the ability to be private, this
# method will need to be updated to enforce the privacy checks.
return SpecificationBranch.select(
IN(SpecificationBranch.q.branchID, branch_ids))
|