config file added for folloer notifications

This commit is contained in:
Chris kerr 2024-03-03 12:29:36 -05:00
parent 86ad3e5f6b
commit 6cfae6abdf
2 changed files with 19 additions and 9 deletions

1
.gitignore vendored Normal file
View File

@ -0,0 +1 @@
config.json

View File

@ -2,11 +2,16 @@ import json
import asyncio import asyncio
import websockets import websockets
# Replace these values with your own Nostr account details def load_config(file_path):
your_pubkey = "16f1a0100d4cfffbcc4230e8e0e4290cc5849c1adc64d6653fda07c031b1074b" try:
your_relay_url = "wss://nos.lol" with open(file_path, 'r') as config_file:
config = json.load(config_file)
return config
except FileNotFoundError:
print(f"Config file not found at {file_path}.")
return None
async def listen_to_relay(): async def listen_to_relay(your_pubkey,your_relay_url):
uri = f"{your_relay_url}/ws" uri = f"{your_relay_url}/ws"
async with websockets.connect(uri) as websocket: async with websockets.connect(uri) as websocket:
# Subscribe to all kind 3 events # Subscribe to all kind 3 events
@ -15,16 +20,16 @@ async def listen_to_relay():
while True: while True:
response = await websocket.recv() response = await websocket.recv()
await handle_event(response) # Await the handle_event coroutine await handle_event(response, your_pubkey) # Await the handle_event coroutine
async def handle_event(event_json): async def handle_event(event_json, your_pubkey):
event = json.loads(event_json) event = json.loads(event_json)
if event[0] == "EVENT": if event[0] == "EVENT":
event_data = event[2] event_data = event[2]
if event_data["kind"] == 3: # Follow list event if event_data["kind"] == 3: # Follow list event
await handle_follow_list(event_data) # Await the handle_follow_list coroutine await handle_follow_list(event_data, your_pubkey) # Await the handle_follow_list coroutine
async def handle_follow_list(follow_list_event): async def handle_follow_list(follow_list_event, your_pubkey):
if "tags" in follow_list_event: if "tags" in follow_list_event:
# Extract the list of "p" tags from the follow list event # Extract the list of "p" tags from the follow list event
p_tags = [tag for tag in follow_list_event["tags"] if tag[0] == "p"] p_tags = [tag for tag in follow_list_event["tags"] if tag[0] == "p"]
@ -34,4 +39,8 @@ async def handle_follow_list(follow_list_event):
print(f"You have a new follower: {follow_list_event.get('pubkey', '')}") print(f"You have a new follower: {follow_list_event.get('pubkey', '')}")
if __name__ == "__main__": if __name__ == "__main__":
asyncio.get_event_loop().run_until_complete(listen_to_relay()) config_path = 'config.json'
config = load_config(config_path)
if config:
asyncio.get_event_loop().run_until_complete(listen_to_relay(config["your_pubkey"], config["your_relay_url"]))