[presto] Removing patched presto (#4530)
This commit is contained in:
parent
06c5077691
commit
150768ee30
|
|
@ -541,12 +541,6 @@ class PrestoEngineSpec(BaseEngineSpec):
|
|||
"date_add('day', 1, CAST({col} AS TIMESTAMP))))"),
|
||||
)
|
||||
|
||||
@classmethod
|
||||
def patch(cls):
|
||||
from pyhive import presto
|
||||
from superset.db_engines import presto as patched_presto
|
||||
presto.Cursor.cancel = patched_presto.cancel
|
||||
|
||||
@classmethod
|
||||
def adjust_database_uri(cls, uri, selected_schema=None):
|
||||
database = uri.database
|
||||
|
|
|
|||
|
|
@ -1,27 +0,0 @@
|
|||
# -*- coding: utf-8 -*-
|
||||
from __future__ import absolute_import
|
||||
from __future__ import division
|
||||
from __future__ import print_function
|
||||
from __future__ import unicode_literals
|
||||
|
||||
from pyhive import presto
|
||||
|
||||
|
||||
# TODO(bogdan): Remove this when new pyhive release will be available.
|
||||
def cancel(self):
|
||||
if self._state == self._STATE_NONE:
|
||||
raise presto.ProgrammingError('No query yet')
|
||||
if self._nextUri is None:
|
||||
assert self._state == self._STATE_FINISHED, \
|
||||
'Should be finished if nextUri is None'
|
||||
return
|
||||
|
||||
response = presto.requests.delete(self._nextUri)
|
||||
|
||||
# pylint: disable=no-member
|
||||
if response.status_code != presto.requests.codes.no_content:
|
||||
fmt = 'Unexpected status code after cancel {}\n{}'
|
||||
raise presto.OperationalError(
|
||||
fmt.format(response.status_code, response.content))
|
||||
self._state = self._STATE_FINISHED
|
||||
return
|
||||
Loading…
Reference in New Issue