def _get_sniffer_type()

in zktraffic/omni/omni_sniffer.py [0:0]

• 12 lines of code
• 5 McCabe index (conditional complexity)

  def _get_sniffer_type(self, sniffer):
    if isinstance(sniffer, Sniffer):
      if sniffer._msg_cls == FLE.Message:
        return 'fle'
      elif sniffer._msg_cls == ZAB.QuorumPacket:
        return 'zab'
      else:
        raise ValueError('Unknown sniffer msg cls %s' % sniffer._msg_cls)
    else:
      if isinstance(sniffer, ZKSniffer):
        return 'zk'
    raise ValueError('Unknown sniffer %s' % sniffer)