PySecretHandshake/examples/test_client.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

26 lines
547 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 SHSClient
with open(os.path.expanduser('~/.ssb/secret')) as f:
config = yaml.safe_load(f)
async def main():
server_pub_key = b64decode(config['public'][:-8])
client = SHSClient('localhost', 8008, SigningKey.generate(), server_pub_key)
await client.open()
async for msg in client:
print(msg)
loop = get_event_loop()
loop.run_until_complete(main())
loop.close()