lbry-sdk/lbrynet/core/Peer.py

36 lines
1 KiB
Python
Raw Normal View History

2015-08-20 17:27:15 +02:00
from collections import defaultdict
import datetime
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
datetime.datetime.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 = datetime.datetime.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 self.host + ":" + str(self.port)