Skip to content

Commit 1fa737b

Browse files
committed
chore: updates operator reference to rely on json operator
1 parent bea6a5d commit 1fa737b

1 file changed

Lines changed: 3 additions & 10 deletions

File tree

  • packages/sqlalchemy-bigquery/sqlalchemy_bigquery

packages/sqlalchemy-bigquery/sqlalchemy_bigquery/_struct.py

Lines changed: 3 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
import sqlalchemy.sql.default_comparator
2121
import sqlalchemy.sql.sqltypes
2222
import sqlalchemy.types
23+
from sqlalchemy.sql import operators
2324

2425
from . import base
2526

@@ -90,7 +91,7 @@ def _setup_getitem(self, name):
9091
subtype = self.expr.type._STRUCT_byname.get(name.lower())
9192
if subtype is None:
9293
raise KeyError(name)
93-
operator = struct_getitem_op
94+
operator = operators.json_getitem_op
9495
index = _field_index(self, name, operator)
9596
return operator, index, subtype
9697

@@ -113,16 +114,8 @@ def _field_index(self, name, operator):
113114
)
114115

115116

116-
def struct_getitem_op(a, b):
117-
raise NotImplementedError()
118-
119-
120-
sqlalchemy.sql.default_comparator.operator_lookup[
121-
struct_getitem_op.__name__
122-
] = sqlalchemy.sql.default_comparator.operator_lookup["json_getitem_op"]
123-
124117

125118
class SQLCompiler:
126-
def visit_struct_getitem_op_binary(self, binary, operator_, **kw):
119+
def visit_json_getitem_op_binary(self, binary, operator_, **kw):
127120
left = self.process(binary.left, **kw)
128121
return f"{left}.{binary.right.value}"

0 commit comments

Comments
 (0)