2
0
mirror of https://github.com/ComradCollective/Comrad synced 2024-11-09 13:10:44 +00:00
Comrad/app/etc/test.py

88 lines
2.5 KiB
Python
Raw Normal View History

2020-08-16 14:59:36 +00:00
import asyncio
from kademlia.network import Server
from kademlia.storage import *
import shelve
from collections import OrderedDict
import pickle,os
class HalfForgetfulStorage(ForgetfulStorage):
def __init__(self, ttl=604800):
"""
By default, max age is a week.
"""
self.fn='sto2.dat'
if not os.path.exists(self.fn):
self.data={}
else:
with open(self.fn,'rb') as f:
self.data=pickle.load(f)
print('loaded:',self.data)
#self.data = pickle.open('sto.dat','rb') #,writeback=True)
# self.data = self.store.get('OrderedDict',OrderedDict())
self.ttl = ttl
def __setitem__(self, key, value):
self.data[key] = (time.monotonic(), value)
self.write()
def write(self):
with open(self.fn,'wb') as f:
pickle.dump(self.data, f, protocol=pickle.HIGHEST_PROTOCOL)
def get(self, key, default=None):
# self.cull()
if key in self.data:
print('????',self.data[key])
return self[key]
return default
def __getitem__(self, key):
return self.data[key][1]
def __repr__(self):
return repr(self.data)
def iter_older_than(self, seconds_old):
min_birthday = time.monotonic() - seconds_old
zipped = self._triple_iter()
matches = takewhile(lambda r: min_birthday >= r[1], zipped)
return list(map(operator.itemgetter(0, 2), matches))
def _triple_iter(self):
ikeys = self.data.keys()
ibirthday = map(operator.itemgetter(0), self.data.values())
ivalues = map(operator.itemgetter(1), self.data.values())
return zip(ikeys, ibirthday, ivalues)
def __iter__(self):
ikeys = self.data.keys()
ivalues = map(operator.itemgetter(1), self.data.values())
return zip(ikeys, ivalues)
2020-08-16 17:50:17 +00:00
host="68.66.241.111"
# host="0.0.0.0"
2020-08-16 14:59:36 +00:00
async def run():
# Create a node and start listening on port 5678
node = Server() #storage=HalfForgetfulStorage())
await node.listen(8469)
# Bootstrap the node by connecting to other known nodes, in this case
# replace 123.123.123.123 with the IP of another node and optionally
# give as many ip/port combos as you can for other nodes.
await node.bootstrap([(host, 8468)])
# set a value for the key "my-key" on the network
# await node.set("my-key2", "my awesome value")
# get the value associated with "my-key" from the network
result = await node.get("my-key2")
print(result)
asyncio.run(run())