lbry-sdk/lbrynet/p2p/Peer.py

import datetime
from collections import defaultdict
from lbrynet.p2p import utils

# Do not create this object except through PeerManager
class Peer:
    def __init__(self, host, port):
        self.host = host
        self.port = port
        # If a peer is reported down, we wait till this time till reattempting connection
        self.attempt_connection_at = None
        # Number of times this Peer has been reported to be down, resets to 0 when it is up
        self.down_count = 0
        # Number of successful connections (with full protocol completion) to this peer
        self.success_count = 0
        self.score = 0
        self.stats = defaultdict(float)  # {string stat_type, float count}

    def is_available(self):
        if self.attempt_connection_at is None or utils.today() > self.attempt_connection_at:
            return True
        return False

    def report_up(self):
        self.down_count = 0
        self.attempt_connection_at = None

    def report_success(self):
        self.success_count += 1

    def report_down(self):
        self.down_count += 1
        timeout_time = datetime.timedelta(seconds=60 * self.down_count)
        self.attempt_connection_at = utils.today() + timeout_time

    def update_score(self, score_change):
        self.score += score_change

    def update_stats(self, stat_type, count):
        self.stats[stat_type] += count

    def __str__(self):
        return f'{self.host}:{self.port}'

    def __repr__(self):
        return f'Peer({self.host!r}, {self.port!r})'