~launchpad-pqm/launchpad/devel

# Copyright 2010 Canonical Ltd.  This software is licensed under the

# GNU Affero General Public License version 3 (see the file LICENSE).

"""The implementation of the branch cloud."""

__metaclass__ = type

__all__ = [

    'BranchCloud',

    ]

from datetime import (

    datetime,

    timedelta,

    )

import pytz

from storm.expr import (

    Alias,

    Func,

    )

from storm.locals import (

    Count,

    Desc,

    Max,

    Not,

    )

from zope.interface import classProvides

from lp.code.interfaces.branch import IBranchCloud

from lp.code.model.revision import RevisionCache

from lp.registry.model.product import Product

from lp.services.database.lpstorm import ISlaveStore

class BranchCloud:

    """See `IBranchCloud`."""

    classProvides(IBranchCloud)

    @staticmethod

    def getProductsWithInfo(num_products=None):

        """See `IBranchCloud`."""

        distinct_revision_author = Func(

            "distinct", RevisionCache.revision_author_id)

        commits = Alias(Count(RevisionCache.revision_id))

        epoch = datetime.now(pytz.UTC) - timedelta(days=30)

        # It doesn't matter if this query is even a whole day out of date, so

        # use the slave store.

        result = ISlaveStore(RevisionCache).find(

            (Product.name,

             commits,

             Count(distinct_revision_author),

             Max(RevisionCache.revision_date)),

            RevisionCache.product == Product.id,

            Not(RevisionCache.private),

            RevisionCache.revision_date >= epoch)

        result = result.group_by(Product.name)

        result = result.order_by(Desc(commits))

        if num_products:

            result.config(limit=num_products)

        return result