1

#!/usr/bin/env python

     2

     3

"""

     4

    Update zone serials.

     5

"""

     6

     7

from pvl.dns import __version__

     8

import pvl.dns.serial

     9

    10

import os.path

    11

    12

import pvl.args

    13

import optparse

    14

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

    15

    16

def parse_options (argv) :

    17

    """

    18

        Parse command-line arguments.

    19

    """

    20

    21

    prog = argv[0]

    22

    23

    parser = optparse.OptionParser(

    24

            prog        = prog,

    25

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

    26

            version     = __version__,

    27

    28

            # module docstring

    29

            description = __doc__,

    30

    )

    31

    32

    # options

    33

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

    34

    35

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

    36

            help="Do not actually update serial")

    37

    38

    # defaults

    39

    parser.set_defaults(

    40

    41

    )

    42

    43

    # parse

    44

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

    45

    46

    # apply

    47

    pvl.args.apply(options, prog)

    48

    49

    return options, args

    50

    51

def process_serial (options, path) :

    52

    """

    53

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

    54

    """

    55

    56

    # read

    57

    if os.path.exists(path) :

    58

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

    59

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

    60

    61

    else :

    62

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

    63

        serial = None

    64

    65

    # update

    66

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

    67

    68

    if options.noop :

    69

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

    70

    else :

    71

        # write

    72

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

    73

    74

    return serial

    75

    76

def main (argv) :

    77

    options, args = parse_options(argv)

    78

    79

    # serial files to update

    80

    for path in args :

    81

        process_serial(options, path)

    82

    83

    return 0

    84

    85

if __name__ == '__main__':

    86

    import sys

    87

    88

    sys.exit(main(sys.argv))