소스 검색

HUE-9561 [SQLAlchemy] Default selected DB does not seem to be used

ayush.goyal 5 년 전
부모
커밋
08a0dad0cc
1개의 변경된 파일3개의 추가작업 그리고 0개의 파일을 삭제
  1. 3 0
      desktop/libs/notebook/src/notebook/connectors/sql_alchemy.py

+ 3 - 0
desktop/libs/notebook/src/notebook/connectors/sql_alchemy.py

@@ -227,6 +227,9 @@ class SqlAlchemyApi(Api):
         self.interpreter.get('dialect_properties') and self.interpreter['dialect_properties']['trim_statement_semicolon']:
       statement = statement.strip().rstrip(';')
 
+    if snippet.get('database'):
+      connection.execute('USE ' + snippet['database'])
+
     result = connection.execute(statement)
 
     cache = {