1

#!/usr/bin/env python

     2

# vim: set ft=python :

     3

     4

"""

     5

    Update zone serials.

     6

"""

     7

     8

from pvl.dns import __version__

     9

import pvl.dns.serial

    10

    11

import os.path

    12

    13

import pvl.args

    14

import optparse

    15

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

    16

    17

def parse_options (argv) :

    18

    """

    19

        Parse command-line arguments.

    20

    """

    21

    22

    prog = argv[0]

    23

    24

    parser = optparse.OptionParser(

    25

            prog        = prog,

    26

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

    27

            version     = __version__,

    28

    29

            # module docstring

    30

            description = __doc__,

    31

    )

    32

    33

    # options

    34

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

    35

    36

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

    37

            help="Do not actually update serial")

    38

    39

    # defaults

    40

    parser.set_defaults(

    41

    42

    )

    43

    44

    # parse

    45

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

    46

    47

    # apply

    48

    pvl.args.apply(options, prog)

    49

    50

    return options, args

    51

    52

def process_serial (options, path) :

    53

    """

    54

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

    55

    """

    56

    57

    # read

    58

    if os.path.exists(path) :

    59

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

    60

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

    61

    62

    else :

    63

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

    64

        serial = None

    65

    66

    # update

    67

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

    68

    69

    if options.noop :

    70

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

    71

    else :

    72

        # write

    73

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

    74

    75

    return serial

    76

    77

def main (argv) :

    78

    options, args = parse_options(argv)

    79

    80

    # serial files to update

    81

    for path in args :

    82

        process_serial(options, path)

    83

    84

    return 0

    85

    86

if __name__ == '__main__':

    87

    import sys

    88

    89

    sys.exit(main(sys.argv))