Realtime integration (#878)

Author: juancarlospaco

supabase/_async/client.py

@@ -10,6 +10,7 @@
     AsyncRPCFilterRequestBuilder,
 )
 from postgrest.constants import DEFAULT_POSTGREST_CLIENT_TIMEOUT
+from realtime.connection import Socket
 from storage3 import AsyncStorageClient
 from storage3.constants import DEFAULT_TIMEOUT as DEFAULT_STORAGE_CLIENT_TIMEOUT
 from supafunc import AsyncFunctionsClient
@@ -80,12 +81,7 @@ def __init__(
             auth_url=self.auth_url,
             client_options=options,
         )
-        # TODO: Bring up to parity with JS client.
-        # self.realtime: SupabaseRealtimeClient = self._init_realtime_client(
-        #     realtime_url=self.realtime_url,
-        #     supabase_key=self.supabase_key,
-        # )
-        self.realtime = None
+        self.realtime = self._init_realtime_client(self.realtime_url, self.supabase_key)
         self._postgrest = None
         self._storage = None
         self._functions = None
@@ -273,6 +269,11 @@ def _init_postgrest_client(
             verify=verify,
         )
 
+    @staticmethod
+    def _init_realtime_client(realtime_url: str, supabase_key: str) -> Socket:
+        """Private helper for creating an instance of the Socket client."""
+        return Socket(realtime_url, supabase_key)
+
     def _create_auth_header(self, token: str):
         return f"Bearer {token}"
 

supabase/_sync/client.py

@@ -10,6 +10,7 @@
     SyncRPCFilterRequestBuilder,
 )
 from postgrest.constants import DEFAULT_POSTGREST_CLIENT_TIMEOUT
+from realtime.connection import Socket
 from storage3 import SyncStorageClient
 from storage3.constants import DEFAULT_TIMEOUT as DEFAULT_STORAGE_CLIENT_TIMEOUT
 from supafunc import SyncFunctionsClient
@@ -80,12 +81,7 @@ def __init__(
             auth_url=self.auth_url,
             client_options=options,
         )
-        # TODO: Bring up to parity with JS client.
-        # self.realtime: SupabaseRealtimeClient = self._init_realtime_client(
-        #     realtime_url=self.realtime_url,
-        #     supabase_key=self.supabase_key,
-        # )
-        self.realtime = None
+        self.realtime = self._init_realtime_client(self.realtime_url, self.supabase_key)
         self._postgrest = None
         self._storage = None
         self._functions = None
@@ -273,6 +269,11 @@ def _init_postgrest_client(
             verify=verify,
         )
 
+    @staticmethod
+    def _init_realtime_client(realtime_url: str, supabase_key: str) -> Socket:
+        """Private helper for creating an instance of the Socket client."""
+        return Socket(realtime_url, supabase_key)
+
     def _create_auth_header(self, token: str):
         return f"Bearer {token}"