@@ -141,12 +141,13 @@ def as_mql(self, compiler, connection):
141
141
return {"$anyElementTrue" : lhs_mql }
142
142
143
143
144
- class ArrayAggregationSubqueryMixin :
144
+ @_EmbeddedModelArrayOutputField .register_lookup
145
+ class EmbeddedModelArrayFieldIn (EmbeddedModelArrayFieldBuiltinLookup , lookups .In ):
145
146
def get_subquery_wrapping_pipeline (self , compiler , connection , field_name , expr ):
146
147
return [
147
148
{
148
149
"$facet" : {
149
- "group " : [
150
+ "gathered_data " : [
150
151
{"$project" : {"tmp_name" : expr .as_mql (compiler , connection )}},
151
152
{
152
153
"$unwind" : "$tmp_name" ,
@@ -166,7 +167,7 @@ def get_subquery_wrapping_pipeline(self, compiler, connection, field_name, expr)
166
167
"$ifNull" : [
167
168
{
168
169
"$getField" : {
169
- "input" : {"$arrayElemAt" : ["$group " , 0 ]},
170
+ "input" : {"$arrayElemAt" : ["$gathered_data " , 0 ]},
170
171
"field" : "tmp_name" ,
171
172
}
172
173
},
@@ -178,13 +179,6 @@ def get_subquery_wrapping_pipeline(self, compiler, connection, field_name, expr)
178
179
]
179
180
180
181
181
- @_EmbeddedModelArrayOutputField .register_lookup
182
- class EmbeddedModelArrayFieldIn (
183
- EmbeddedModelArrayFieldBuiltinLookup , lookups .In , ArrayAggregationSubqueryMixin
184
- ):
185
- pass
186
-
187
-
188
182
@_EmbeddedModelArrayOutputField .register_lookup
189
183
class EmbeddedModelArrayFieldExact (EmbeddedModelArrayFieldBuiltinLookup , lookups .Exact ):
190
184
pass
0 commit comments