QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-04-20 16:44:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: enforce more ruff rules (#31447)

Browse Source 
Co-authored-by: Elizabeth Thompson <eschutho@gmail.com>
This commit is contained in:
Maxime Beauchemin
2024-12-18 17:41:34 -08:00
committed by GitHub
parent 9da65d6bfd
commit e51b95ffa8
375 changed files with 1821 additions and 1718 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
superset/db_engine_specs/ocient.py
View File

@@ -214,7 +214,7 @@ def _find_columns_to_sanitize(cursor: Any) -> list[PlacedSanitizeFunc]:
:param cursor: the result set cursor
:returns: the list of tuples consisting of the column index and sanitization function
"""
""" # noqa: E501
return [
PlacedSanitizeFunc(i, _sanitized_ocient_type_codes[cursor.description[i][1]])
for i in range(len(cursor.description))
@@ -317,9 +317,7 @@ class OcientEngineSpec(BaseEngineSpec):
rows: list[tuple[Any, ...]] = super().fetch_data(cursor, limit)
except Exception:
with OcientEngineSpec.query_id_mapping_lock:
del OcientEngineSpec.query_id_mapping[
getattr(cursor, "superset_query_id")
]
del OcientEngineSpec.query_id_mapping[cursor.superset_query_id]
raise
# TODO: Unsure if we need to verify that we are receiving rows:
@@ -376,7 +374,7 @@ class OcientEngineSpec(BaseEngineSpec):
OcientEngineSpec.query_id_mapping[query.id] = cursor.query_id
# Add the query id to the cursor
setattr(cursor, "superset_query_id", query.id)
cursor.superset_query_id = query.id
return super().handle_cursor(cursor, query)
@classmethod