lbry-sdk/lbrynet/core/Peer.py

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


class Peer(object):
    def __init__(self, host, port):
        self.host = host
        self.port = port
        self.attempt_connection_at = None
        self.down_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_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 '{}:{}'.format(self.host, self.port)

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