1

#!/usr/bin/env python

     2

     3

"""

     4

    Update zone serials.

     5

"""

     6

     7

import logging; log = logging.getLogger('main')

     8

import optparse

     9

import os.path

    10

import pvl.args

    11

import pvl.dns.serial

    12

    13

from pvl.hosts import __version__

    14

    15

def parse_options (argv) :

    16

    """

    17

        Parse command-line arguments.

    18

    """

    19

    20

    prog = argv[0]

    21

    22

    parser = optparse.OptionParser(

    23

            prog        = prog,

    24

            usage       = '%prog: [options] <serial-path> ...',

    25

            version     = __version__,

    26

    27

            # module docstring

    28

            description = __doc__,

    29

    )

    30

    31

    # options

    32

    parser.add_option_group(pvl.args.parser(parser))

    33

    34

    parser.add_option('-n', '--noop',       action='store_true',

    35

            help="Do not actually update serial")

    36

    37

    # defaults

    38

    parser.set_defaults(

    39

    40

    )

    41

    42

    # parse

    43

    options, args = parser.parse_args(argv[1:])

    44

    45

    # apply

    46

    pvl.args.apply(options, prog)

    47

    48

    return options, args

    49

    50

def process_serial (options, path) :

    51

    """

    52

        Read old serial, generate new one, and update file.

    53

    """

    54

    55

    # read

    56

    if os.path.exists(path) :

    57

        serial = open(path).read().strip()

    58

        log.debug("%s: current: %s", path, serial)

    59

    60

    else :

    61

        log.warn("%s: Given .serial does not yet exist", path)

    62

        serial = None

    63

    64

    # update

    65

    serial = pvl.dns.serial.update_serial(serial)

    66

    67

    if options.noop :

    68

        log.warn("%s: %s", path, serial)

    69

    else :

    70

        # write

    71

        open(path, 'w').write(serial + '\n')

    72

    73

    return serial

    74

    75

def main (argv) :

    76

    options, args = parse_options(argv)

    77

    78

    # serial files to update

    79

    for path in args :

    80

        process_serial(options, path)

    81

    82

    return 0

    83

    84

if __name__ == '__main__':

    85

    import sys

    86

    87

    sys.exit(main(sys.argv))