Skip to content

Commit 1bf5de8

Browse files
authored
Merge branch 'main' into bugfix/inhereted-defaults
2 parents c99ad77 + d0e3a68 commit 1bf5de8

File tree

2 files changed

+29
-12
lines changed

2 files changed

+29
-12
lines changed

aredis_om/model/model.py

Lines changed: 14 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -309,7 +309,7 @@ def score_field(self) -> str:
309309
class ExpressionProxy:
310310
def __init__(self, field: ModelField, parents: List[Tuple[str, "RedisModel"]]):
311311
self.field = field
312-
self.parents = parents
312+
self.parents = parents.copy() # Ensure a copy is stored
313313

314314
def __eq__(self, other: Any) -> Expression: # type: ignore[override]
315315
return Expression(
@@ -387,13 +387,14 @@ def __getattr__(self, item):
387387
attr = getattr(embedded_cls, item)
388388
else:
389389
attr = getattr(outer_type, item)
390+
390391
if isinstance(attr, self.__class__):
392+
# Clone the parents to ensure isolation
393+
new_parents = self.parents.copy()
391394
new_parent = (self.field.alias, outer_type)
392-
if new_parent not in attr.parents:
393-
attr.parents.append(new_parent)
394-
new_parents = list(set(self.parents) - set(attr.parents))
395-
if new_parents:
396-
attr.parents = new_parents + attr.parents
395+
if new_parent not in new_parents:
396+
new_parents.append(new_parent)
397+
attr.parents = new_parents
397398
return attr
398399

399400

@@ -632,10 +633,11 @@ def resolve_value(
632633
value: Any,
633634
parents: List[Tuple[str, "RedisModel"]],
634635
) -> str:
636+
# The 'field_name' should already include the correct prefix
637+
result = ""
635638
if parents:
636639
prefix = "_".join([p[0] for p in parents])
637640
field_name = f"{prefix}_{field_name}"
638-
result = ""
639641
if field_type is RediSearchFieldTypes.TEXT:
640642
result = f"@{field_name}_fts:"
641643
if op is Operators.EQ:
@@ -792,8 +794,6 @@ def resolve_redisearch_query(cls, expression: ExpressionOrNegated) -> str:
792794

793795
if expression.op is Operators.ALL:
794796
if encompassing_expression_is_negated:
795-
# TODO: Is there a use case for this, perhaps for dynamic
796-
# scoring purposes with full-text search?
797797
raise QueryNotSupportedError(
798798
"You cannot negate a query for all results."
799799
)
@@ -827,6 +827,11 @@ def resolve_redisearch_query(cls, expression: ExpressionOrNegated) -> str:
827827
f"or an expression enclosed in parentheses. Docs: {ERRORS_URL}#E7"
828828
)
829829

830+
if isinstance(expression.left, ModelField) and expression.parents:
831+
# Build field_name using the specific parents for this expression
832+
prefix = "_".join([p[0] for p in expression.parents])
833+
field_name = f"{prefix}_{field_name}"
834+
830835
right = expression.right
831836

832837
if isinstance(right, Expression) or isinstance(right, NegatedExpression):
@@ -842,8 +847,6 @@ def resolve_redisearch_query(cls, expression: ExpressionOrNegated) -> str:
842847

843848
if isinstance(right, NegatedExpression):
844849
result += "-"
845-
# We're handling the RediSearch operator in this call ("-"), so resolve the
846-
# inner expression instead of the NegatedExpression.
847850
right = right.expression
848851

849852
result += f"({cls.resolve_redisearch_query(right)})"

tests/test_json_model.py

Lines changed: 15 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1160,7 +1160,6 @@ class TestLiterals(JsonModel):
11601160
assert rematerialized.pk == item.pk
11611161

11621162

1163-
@py_test_mark_asyncio
11641163
async def test_child_class_expression_proxy():
11651164
# https://github.com/redis/redis-om-python/issues/669 seeing weird issue with child classes initalizing all their undefined members as ExpressionProxies
11661165
class Model(JsonModel):
@@ -1183,3 +1182,18 @@ class Child(Model):
11831182
assert rematerialized.age == 18
11841183
assert rematerialized.age != 19
11851184
assert rematerialized.bio is None
1185+
1186+
async def test_merged_model_error():
1187+
class Player(EmbeddedJsonModel):
1188+
username: str = Field(index=True)
1189+
1190+
class Game(JsonModel):
1191+
player1: Optional[Player]
1192+
player2: Optional[Player]
1193+
1194+
q = Game.find(
1195+
(Game.player1.username == "username") | (Game.player2.username == "username")
1196+
)
1197+
print(q.query)
1198+
assert q.query == "(@player1_username:{username})| (@player2_username:{username})"
1199+

0 commit comments

Comments
 (0)