← Back to branch summary

~launchpad-pqm/launchpad/devel

« back to all changes in this revision

Viewing changes to scripts/ftpmaster-tools/archive-override-check.py

  • Committer: Steve Kowalik
  • Date: 2011-08-07 04:05:52 UTC
  • mto: This revision was merged to the branch mainline in revision 13626.
  • Revision ID: stevenk@ubuntu.com-20110807040552-mwnxo0flmhvl35e8
Correct the notification based on review comments, and remove request{,ed}
from the function names, switching to create{,d}.

Show diffs side-by-side

added added

removed removed

Lines of Context:
scripts/ftpmaster-tools/archive-override-check.py
11
11
 
12
12
import _pythonpath
13
13
 
14
 
import transaction
 
14
from optparse import OptionParser
 
15
import sys
 
16
 
15
17
from zope.component import getUtility
 
18
# Still needed fake import to stop circular imports.
 
19
import canonical.launchpad.interfaces
16
20
 
 
21
from canonical.config import config
 
22
from canonical.launchpad.scripts import (
 
23
    execute_zcml_for_scripts, logger, logger_options)
17
24
from lp.app.errors import NotFoundError
 
25
from canonical.lp import initZopeless
18
26
from lp.registry.interfaces.distribution import IDistributionSet
19
27
from lp.registry.interfaces.pocket import PackagePublishingPocket
20
 
from lp.services.config import config
21
 
from lp.services.scripts.base import LaunchpadScript
 
28
from lp.soyuz.scripts.ftpmaster import PubSourceChecker
22
29
from lp.soyuz.enums import PackagePublishingStatus
23
 
from lp.soyuz.scripts.ftpmaster import PubSourceChecker
24
 
 
25
 
 
26
 
class ArchiveOverrideCheckScript(LaunchpadScript):
27
 
 
28
 
    def add_my_options(self):
29
 
        self.parser.add_option(
30
 
            "-d", "--distribution", action="store",
31
 
            dest="distribution", metavar="DISTRO", default="ubuntu",
32
 
            help="Distribution to consider")
33
 
        self.parser.add_option(
34
 
            "-s", "--suite", action="store",
35
 
            dest="suite", metavar="SUITE", default=None,
36
 
            help=("Suite to consider, if not passed consider the "
37
 
                  "currentseries and the RELEASE pocket"))
38
 
 
39
 
    def main(self):
 
30
 
 
31
from contrib.glock import GlobalLock
 
32
 
 
33
def main():
 
34
    # Parse command-line arguments
 
35
    parser = OptionParser()
 
36
    logger_options(parser)
 
37
 
 
38
    parser.add_option("-d", "--distribution", action="store",
 
39
                      dest="distribution", metavar="DISTRO", default="ubuntu",
 
40
                      help="Distribution to consider")
 
41
 
 
42
    parser.add_option("-s", "--suite", action="store",
 
43
                      dest="suite", metavar="SUITE", default=None,
 
44
                      help=("Suite to consider, if not passed consider the "
 
45
                            "currentseries and the RELEASE pocket"))
 
46
 
 
47
    (options, args) = parser.parse_args()
 
48
 
 
49
    log = logger(options, "archive-override-check")
 
50
 
 
51
    log.debug("Acquiring lock")
 
52
    lock = GlobalLock('/var/lock/archive-override-check.lock')
 
53
    lock.acquire(blocking=True)
 
54
 
 
55
    log.debug("Initializing connection.")
 
56
    execute_zcml_for_scripts()
 
57
    ztm = initZopeless(dbuser=config.archivepublisher.dbuser)
 
58
 
 
59
    try:
40
60
        try:
41
 
            try:
42
 
                distribution = getUtility(IDistributionSet)[
43
 
                    self.options.distribution]
44
 
                if self.options.suite is None:
45
 
                    distroseries = distribution.currentseries
46
 
                    pocket = PackagePublishingPocket.RELEASE
47
 
                else:
48
 
                    distroseries, pocket = (
49
 
                        distribution.getDistroSeriesAndPocket(
50
 
                            self.options.suite))
51
 
 
52
 
                self.logger.debug(
53
 
                    "Considering: %s/%s/%s/%s."
54
 
                    % (distribution.name, distroseries.name, pocket.name,
55
 
                       distroseries.status.name))
56
 
 
57
 
                checkOverrides(distroseries, pocket, self.logger)
58
 
            except NotFoundError, info:
59
 
                self.logger.error('Not found: %s' % info)
60
 
        finally:
61
 
            self.logger.debug("Rolling back any remaining transactions.")
62
 
            transaction.abort()
 
61
            distribution = getUtility(IDistributionSet)[options.distribution]
 
62
            if options.suite is None:
 
63
                distroseries = distribution.currentseries
 
64
                pocket = PackagePublishingPocket.RELEASE
 
65
            else:
 
66
                distroseries, pocket = distribution.getDistroSeriesAndPocket(
 
67
                    options.suite)
 
68
 
 
69
            log.debug("Considering: %s/%s/%s/%s."
 
70
                      % (distribution.name, distroseries.name, pocket.name,
 
71
                         distroseries.status.name))
 
72
 
 
73
            checkOverrides(distroseries, pocket, log)
 
74
 
 
75
        except NotFoundError, info:
 
76
            log.error('Not found: %s' % info)
 
77
 
 
78
    finally:
 
79
        log.debug("Rolling back any remaining transactions.")
 
80
        ztm.abort()
 
81
        log.debug("Releasing lock")
 
82
        lock.release()
 
83
 
 
84
    return 0
63
85
 
64
86
 
65
87
def checkOverrides(distroseries, pocket, log):
94
116
            print report
95
117
 
96
118
if __name__ == '__main__':
97
 
    script = ArchiveOverrideCheckScript(
98
 
        'archive-override-check', config.archivepublisher.dbuser)
99
 
    script.lock_and_run()
 
119
    sys.exit(main())
 
120