Patch pre_select supabase

Author: moshemorad

src/robusta/core/sinks/robusta/dal/supabase_dal.py

@@ -10,7 +10,7 @@
 from cachetools import TTLCache
 import requests
 from postgrest._sync.request_builder import SyncQueryRequestBuilder
-from postgrest.base_request_builder import BaseFilterRequestBuilder
+from postgrest.base_request_builder import BaseFilterRequestBuilder, QueryArgs
 from postgrest.exceptions import APIError as PostgrestAPIError
 from postgrest.types import ReturnMethod
 from postgrest.utils import sanitize_param
@@ -37,6 +37,8 @@
     AccountResourceStatusType,
     ResourceKind,
 )
+from postgrest._sync import request_builder as supabase_request_builder
+
 
 SERVICES_TABLE = "Services"
 NODES_TABLE = "Nodes"
@@ -55,6 +57,18 @@
 OPENSHIFT_GROUPS_TABLE = "OpenshiftGroups"
 SESSION_TOKENS_TABLE = "AuthTokens"
 
+
+logging.info("Patching supabase_request_builder.pre_select")
+original_pre_select = supabase_request_builder.pre_select
+def pre_select_patched(*args, **kwargs):
+    args: QueryArgs = original_pre_select(*args, **kwargs)
+    if not args.json:
+        args = QueryArgs(args.method, args.params, args.headers, None)
+
+    return args
+supabase_request_builder.pre_select = pre_select_patched
+
+
 class SupabaseDal(AccountResourceFetcher):
     def __init__(
         self,