PySecretHandshake/examples/test_server.py
Gergely Polonkai 85cea58067
build: Change yaml.load to safe_load
It’s safer.  PyYAML also changed the load method so it requires a loader parameter.
2023-10-31 07:34:56 +01:00

29 lines
625 B
Python

import os
from asyncio import get_event_loop
from base64 import b64decode
import yaml
from nacl.signing import SigningKey
from secret_handshake import SHSServer
with open(os.path.expanduser('~/.ssb/secret')) as f:
config = yaml.safe_load(f)
async def _on_connect(conn):
async for msg in conn:
print(msg)
async def main():
server_keypair = SigningKey(b64decode(config['private'][:-8])[:32])
server = SHSServer('localhost', 8008, server_keypair)
server.on_connect(_on_connect)
await server.listen()
loop = get_event_loop()
loop.run_until_complete(main())
loop.run_forever()
loop.close()