[Community] handle production db functions

Author: GuillaumeDSM

octobot/community/supabase_backend/community_supabase_client.py

@@ -424,14 +424,15 @@ async def upsert_portfolio_history(self, portfolio_histories) -> list:
         ).execute()).data
 
     async def fetch_candles_history_range(
-        self, exchange: str, symbol: str, time_frame: commons_enums.TimeFrames
+        self, exchange: str, symbol: str, time_frame: commons_enums.TimeFrames, use_production_db: bool
     ) -> (typing.Union[float, None], typing.Union[float, None]):
         params = {
             "exchange_internal_name": exchange,
             "symbol": symbol,
             "time_frame": time_frame.value,
         }
-        range_return = (await self.postgres_functions().invoke(
+        db_functions = self.production_anon_postgres_functions() if use_production_db else self.postgres_functions()
+        range_return = (await db_functions.invoke(
             "get_ohlcv_range",
             {"body": params}
         ))["data"]
@@ -511,6 +512,14 @@ async def get_production_anon_client(self):
             )
         return self.production_anon_client
 
+    def _get_production_anon_auth_headers(self):
+        return self._get_anon_auth_headers(constants.COMMUNITY_PRODUCTION_BACKEND_KEY)
+
+    def production_anon_postgres_functions(self):
+        return self.postgres_functions(
+            url=constants.COMMUNITY_PRODUCTION_BACKEND_URL, auth_headers=self._get_production_anon_auth_headers()
+        )
+
     async def _paginated_fetch_historical_data(
         self, client, table_name: str, select: str, matcher: dict,
         first_open_time: float, last_open_time: float

octobot/community/supabase_backend/supabase_client.py

@@ -136,8 +136,8 @@ async def close(self):
             pass
         await self.realtime.close()
 
-    def postgres_functions(self):
-        return postgres_functions.PostgresFunctions(self.supabase_url, self._get_auth_headers())
+    def postgres_functions(self, url=None, auth_headers=None):
+        return postgres_functions.PostgresFunctions(url or self.supabase_url, auth_headers or self._get_auth_headers())
 
     def remove_session_details(self):
         self.auth._remove_session()
@@ -158,9 +158,12 @@ def _use_auth_session(self, event: gotrue.AuthChangeEvent, _):
             # await self.storage.aclose()
             # self._init_storage_client(self.storage_url, self._get_auth_headers())
 
+    def _get_anon_auth_headers(self, supabase_key):
+        """Helper method to get anon auth headers."""
+        return self._format_auth_headers(supabase_key, supabase_key)
+
     def _get_auth_headers(self):
         """Helper method to get auth headers."""
-        # What's the corresponding method to get the token
         return self._format_auth_headers(self.supabase_key, self._get_auth_key())
 
     def _format_auth_headers(self, supabase_key, auth_token):