Selaa lähdekoodia

sentry] Fix Hive autocomplete error on columns with API v2

Romain Rigaux 9 vuotta sitten
vanhempi
commit
570e2a0
1 muutettua tiedostoa jossa 7 lisäystä ja 3 poistoa
  1. 7 3
      apps/security/src/security/api/sentry.py

+ 7 - 3
apps/security/src/security/api/sentry.py

@@ -48,18 +48,22 @@ def _fetch_hive_path(request):
 
   database = None
   table = None
+  column = None
+
   if path:
     database = path
   if '/' in path:
-    database, table = path.split('/')
+    database, table = path.split('/', 1)
+    if '.' in table:
+      table, column  = table.split('.', 1)
 
-  resp = autocomplete(request, database, table)
+  resp = autocomplete(request, database, table, column)
 
   if database and request.GET['doas'] != request.user.username:
     request.GET = request.GET.copy()
     request.GET['doas'] = request.GET['doas']
 
-    resp = autocomplete(request, database, table)
+    resp = autocomplete(request, database, table, column)
 
   return resp