pump_bfd2.py

import pump
import pump_bfd


class BFDSinkEx(pump_bfd.BFDSink):

    def __init__(self, opts, spec, source_bucket, source_node,
                 source_map, sink_map, ctl, cur):
        # pylint: disable=bad-super-call
        super(pump_bfd.BFDSink, self).__init__(opts, spec, source_bucket, source_node, source_map,
                                               sink_map, ctl, cur)
        self.mode = getattr(opts, "mode", "diff")
        self.init_worker(pump_bfd.BFDSink.run)

    @staticmethod
    def check_spec(source_bucket, source_node, opts, spec, cur):
        pump.Sink.check_spec(source_bucket, source_node, opts, spec, cur)

        seqno, dep, faillover_log, snapshot_markers = pump_bfd.BFD.find_seqno(opts, spec,
                                                        source_bucket['name'],
                                                        source_node['hostname'],
                                                        getattr(opts, "mode", "diff"))
        if 'seqno' in cur:
            cur['seqno'][(source_bucket['name'], source_node['hostname'])] = seqno
        else:
            cur['seqno'] = {(source_bucket['name'], source_node['hostname']): seqno}

        if 'failoverlog' in cur:
            cur['failoverlog'][(source_bucket['name'], source_node['hostname'])] = faillover_log
        else:
            cur['failoverlog'] = {(source_bucket['name'], source_node['hostname']): faillover_log}

        if 'snapshot' in cur:
            cur['snapshot'][(source_bucket['name'], source_node['hostname'])] = snapshot_markers
        else:
            cur['snapshot'] = {(source_bucket['name'], source_node['hostname']): snapshot_markers}