33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
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]
|