aljaz/python-nostr
1
0
Fork 0
mirror of https://github.com/aljazceru/python-nostr.git synced 2025-12-18 23:04:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

bech32 working

Browse Source
This commit is contained in:
callebtc
2023-02-13 22:11:23 +01:00
parent f598039e44
commit 0328fc35f5
2 changed files with 34 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
main.py
View File

@@ -27,11 +27,11 @@ async def dm():
f"\nFrom {event.public_key[:3]}..{event.public_key[-3:]}: {decrypted_content}"
)
client = NostrClient(privatekey_hex=pk)
client = NostrClient(private_key=pk)
if not pk:
print(f"Your private key: {client.private_key.hex()}")
print(f"Your private key: {client.private_key.bech32()}")
print(f"Your public key: {client.public_key.hex()}")
print(f"Your public key: {client.public_key.bech32()}")
# await asyncio.sleep(1)
t = threading.Thread(
@@ -43,16 +43,20 @@ async def dm():
)
t.start()
to_pubk_hex = (
input("Enter other pubkey to post to (enter nothing to DM yourself): ")
pubkey_to_str = (
input("Enter other pubkey to DM to (enter nothing to DM yourself): ")
or client.public_key.hex()
)
print(f"Subscribing to DMs to {to_pubk_hex}")
if pubkey_to_str.startswith("npub"):
pubkey_to = PublicKey().from_npub(pubkey_to_str)
else:
pubkey_to = PublicKey(bytes.fromhex(pubkey_to_str))
print(f"Sending DMs to {pubkey_to.bech32()}")
while True:
print_status(client)
await asyncio.sleep(1)
msg = input("\nEnter message: ")
client.dm(msg, PublicKey(bytes.fromhex(to_pubk_hex)))
client.dm(msg, pubkey_to)
async def post():
@@ -67,20 +71,23 @@ async def post():
f"\nFrom {event.public_key[:3]}..{event.public_key[-3:]}: {event.content}"
)
sender_client = NostrClient(privatekey_hex=pk)
sender_client = NostrClient(private_key=pk)
# await asyncio.sleep(1)
to_pubk_hex = (
pubkey_to_str = (
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
if pubkey_to_str == "*":
pubkey_to = None
elif pubkey_to_str.startswith("npub"):
pubkey_to = PublicKey().from_npub(pubkey_to_str)
else:
print(f"Subscribing to posts by {to_pubk_hex}")
to_pubk = PublicKey(bytes.fromhex(to_pubk_hex))
pubkey_to = PublicKey(bytes.fromhex(pubkey_to_str))
print(f"Subscribing to posts by {pubkey_to.bech32() if pubkey_to else 'everyone'}")
filters = {
"since": int(
@@ -93,7 +100,7 @@ async def post():
t = threading.Thread(
target=sender_client.get_post,
args=(
to_pubk,
pubkey_to,
callback,
filters,
),
@@ -107,9 +114,9 @@ async def 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:
if input("Enter '1' for DM, '2' for Posts (Default: 1): ") == "2":
# make a post and subscribe to posts
asyncio.run(post())
else:
# write a DM and receive DMs
asyncio.run(dm())

14
nostr/client/client.py
View File

@@ -30,8 +30,8 @@ class NostrClient:
private_key: PrivateKey
public_key: PublicKey
def __init__(self, privatekey_hex: str = "", relays: List[str] = [], connect=True):
self.generate_keys(privatekey_hex)
def __init__(self, private_key: str = "", relays: List[str] = [], connect=True):
self.generate_keys(private_key)
if len(relays):
self.relays = relays
@@ -48,9 +48,13 @@ class NostrClient:
def close(self):
self.relay_manager.close_connections()
def generate_keys(self, privatekey_hex: str = None):
pk = bytes.fromhex(privatekey_hex) if privatekey_hex else None
self.private_key = PrivateKey(pk)
def generate_keys(self, private_key: str = None):
if private_key.startswith("nsec"):
self.private_key = PrivateKey.from_nsec(private_key)
elif private_key:
self.private_key = PrivateKey(bytes.fromhex(private_key))
else:
self.private_key = PrivateKey() # generate random key
self.public_key = self.private_key.public_key
def post(self, message: str):