throw error in registration

Author: pfandzelter

celestial/host.py

@@ -17,6 +17,7 @@
 
 """Adapter for communication with the Celestial hosts over gRPC"""
 
+import os
 import typing
 import grpc
 import time
@@ -52,11 +53,16 @@ def register(self) -> proto.celestial.celestial_pb2.RegisterResponse:
 
         :return: The response from the host.
         """
-        request = proto.celestial.celestial_pb2.RegisterRequest(host=self.num)
+        try:
+            request = proto.celestial.celestial_pb2.RegisterRequest(host=self.num)
 
-        response: proto.celestial.celestial_pb2.RegisterResponse = self.stub.Register(
-            request
-        )
+            response: proto.celestial.celestial_pb2.RegisterResponse = (
+                self.stub.Register(request)
+            )
+
+        except grpc.RpcError as e:
+            logging.error(f"Error registering host {self.num}: {e}")
+            os.exit(1)
 
         # others currently not used
         self.peer_public_key = response.peer_public_key