add try when failing (#6)

main.py

@@ -7,18 +7,29 @@ from nostr.relay_manager import RelayManager
 from nostr.message_type import ClientMessageType
 from nostr.key import PrivateKey
 
-try:
-    # Read env variable and add relays
-    env_relays = os.getenv('RELAYS') # None
-    if env_relays is None:
-        env_relays = "wss://relay.nostr.band"
+def connect(relays):
     relay_manager = RelayManager()
-    for relay in env_relays.split(","):
+    for relay in relays:
         print("Adding relay: " + relay)
         relay_manager.add_relay(relay)
 
     relay_manager.open_connections({"cert_reqs": ssl.CERT_NONE}) # NOTE: This disables ssl certificate verification
     time.sleep(1.25) # allow the connections to open
+    print("Connected")
+
+    return relay_manager
+
+def disconnect(relay_manager):
+    print("Disconnecting...")
+    relay_manager.close_connections()
+    print("Disconnected")
+    
+try:
+    # Read env variable and add relays
+    env_relays = os.getenv('RELAYS') # None
+    if env_relays is None:
+        env_relays = "wss://relay.nostr.band"
+    relay_manager = connect(env_relays.split(","))
 
     env_private_key = os.environ.get("PRIVATE_KEY")
     if not env_private_key:
@@ -29,24 +40,36 @@ try:
 
     old_block_height = 0
     while True:
-        url = "https://blockchain.info/latestblock"
-        response = requests.get(url)
-        data = response.json()
-        block_height = data["height"]
+        failCount = 0
+        try:
+            url = "https://blockchain.info/latestblock"
+            response = requests.get(url)
+            data = response.json()
+            block_height = data["height"]
+
+            if(block_height > old_block_height):
+                message = "⚡️ " + str(block_height) + " ⚡️"
+                print(message)
+                event = Event(
+                    content=str(message),
+                    public_key=private_key.public_key.hex()
+                )
+                private_key.sign_event(event)
+                relay_manager.publish_event(event)
+                # relay_manager.close_connections() # NEEDED?!
+                old_block_height = block_height
+            time.sleep(5)
+        except Exception as e:
+            print(e)
+            failCount += 1
+            disconnect(relay_manager)
+            relay_manager = connect(env_relays.split(","))
+            if failCount > 5:
+                print("Too many fails, exiting...")
+                exit(1)
+            time.sleep(5)
 
-        if(block_height > old_block_height):
-            message = "⚡️ " + str(block_height) + " ⚡️"
-            print(message)
-            event = Event(
-                content=str(message),
-                public_key=private_key.public_key.hex()
-            )
-            private_key.sign_event(event)
-            relay_manager.publish_event(event)
-            # relay_manager.close_connections() # NEEDED?!
-            old_block_height = block_height
-        time.sleep(5)
 except Exception as e:
     print(e)
-    relay_manager.close_connections()
+    disconnect(relay_manager)
     exit(1)