lbry-sdk/lbrynet/dht/hashwatcher.py

from collections import Counter
import datetime
from twisted.internet import task


class HashWatcher(object):
    def __init__(self, clock=None):
        if not clock:
            from twisted.internet import reactor as clock
        self.ttl = 600
        self.hashes = []
        self.lc = task.LoopingCall(self._remove_old_hashes)
        self.lc.clock = clock

    def start(self):
        return self.lc.start(10)

    def stop(self):
        return self.lc.stop()

    def add_requested_hash(self, hashsum, contact):
        from_ip = contact.compact_ip
        matching_hashes = [h for h in self.hashes if h[0] == hashsum and h[2] == from_ip]
        if len(matching_hashes) == 0:
            self.hashes.append((hashsum, datetime.datetime.now(), from_ip))

    def most_popular_hashes(self, num_to_return=10):
        hash_counter = Counter([h[0] for h in self.hashes])
        return hash_counter.most_common(num_to_return)

    def _remove_old_hashes(self):
        remove_time = datetime.datetime.now() - datetime.timedelta(minutes=10)
        self.hashes = [h for h in self.hashes if h[1] < remove_time]