~launchpad-pqm/launchpad/devel

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
#!/usr/bin/python -S
# Copyright 2010 Canonical Ltd.  This software is licensed under the
# GNU Affero General Public License version 3 (see the file LICENSE).

"""Populate the DatabaseTableStats and DatabaseCpuStats tables."""

__metaclass__ = type

import _pythonpath

from zope.component import getUtility

from canonical.launchpad.scripts import db_options
from canonical.launchpad.webapp.interfaces import (
    IStoreSelector, MAIN_STORE, MASTER_FLAVOR)
from lp.services.scripts.base import LaunchpadCronScript


class UpdateDatabaseStats(LaunchpadCronScript):
    """Populate the DatabaseTableStats and DatabaseCpuStats tables."""

    def main(self):
        "Run UpdateDatabaseTableStats."""
        store = getUtility(IStoreSelector).get(MAIN_STORE, MASTER_FLAVOR)

        # The logic is in a stored procedure because we want to run
        # ps(1) on the database server rather than the host this script
        # is running on.
        self.logger.debug("Invoking update_database_stats()")
        store.execute("SELECT update_database_stats()", noresult=True)

        self.logger.debug("Committing")
        store.commit()

    def add_my_options(self):
        """Add standard database command line options."""
        db_options(self.parser)

if __name__ == '__main__':
    script = UpdateDatabaseStats(
        'update-database-stats', dbuser='database_stats_update')
    script.lock_and_run()