Files
python-nostr/main.py
2023-02-07 23:07:56 +01:00

105 lines
2.6 KiB
Python

from nostr.client.client import NostrClient
from nostr.event import Event
from nostr.key import PublicKey
import asyncio
import threading
import time
import datetime
async def dm():
print("This is an example NIP-04 DM flow")
pk = input("Enter your privatekey to post from (enter nothing for a random one): ")
def callback(event: Event, decrypted_content):
"""
Callback to trigger when a DM is received.
"""
print(
f"From {event.public_key[:3]}..{event.public_key[-3:]}: {decrypted_content}"
)
client = NostrClient(privatekey_hex=pk)
filters = {
"since": int(
time.mktime(
(datetime.datetime.now() - datetime.timedelta(hours=1)).timetuple()
)
)
}
t = threading.Thread(
target=client.get_dm,
args=(
client.public_key,
callback,
filters,
),
)
t.start()
to_pubk_hex = (
input("Enter other pubkey to post to (enter nothing to DM yourself): ")
or client.public_key.hex()
)
print(f"Subscribing to DMs to {to_pubk_hex}")
while True:
msg = input("\nEnter message: ")
client.dm(msg, PublicKey(bytes.fromhex(to_pubk_hex)))
async def post():
print("This posts and reads a nostr note")
pk = input("Enter your privatekey to post from (enter nothing for a random one): ")
def callback(event: Event):
"""
Callback to trigger when post appers.
"""
print(f"From {event.public_key[:3]}..{event.public_key[-3:]}: {event.content}")
sender_client = NostrClient(privatekey_hex=pk)
to_pubk_hex = (
input(
"Enter other pubkey (enter nothing to read your own posts, enter * for all): "
)
or sender_client.public_key.hex()
)
if to_pubk_hex == "*":
to_pubk = None
else:
print(f"Subscribing to posts by {to_pubk_hex}")
to_pubk = PublicKey(bytes.fromhex(to_pubk_hex))
filters = {
"since": int(
time.mktime(
(datetime.datetime.now() - datetime.timedelta(hours=1)).timetuple()
)
)
}
t = threading.Thread(
target=sender_client.get_post,
args=(
to_pubk,
callback,
filters,
),
)
t.start()
while True:
msg = input("\nEnter post: ")
sender_client.post(msg)
if input("Enter 1 for DM, 2 for Posts (Default: 1)") or 1 == 1:
# write a DM and receive DMs
asyncio.run(dm())
else:
# make a post and subscribe to posts
asyncio.run(post())