direct.py

from .base import Router

class DirectRouter(Router):
    def __call__(self, packet):
        if packet.destination in self.node.neighbours:
            return packet.destination, 'direct', True
        return None, None, False

    def on_send_failure(self, target, packet):
        self.node.buffer.add(packet)