@@ -64,7 +64,7 @@ def model(self):
6464 def args (self ):
6565 return to_arguments (
6666 self ._base_args or OrderedDict (),
67- dict (self .field_args . items () + self .reference_args . items () )
67+ dict (self .field_args , ** self .reference_args )
6868 )
6969
7070 @args .setter
@@ -109,13 +109,15 @@ def get_query(cls, model, info, **args):
109109 objs = model .objects ()
110110 if args :
111111 reference_fields = get_model_reference_fields (model )
112- for arg_name , arg in args .items ():
112+ reference_args = {}
113+ for arg_name , arg in args .copy ().items ():
113114 if arg_name in reference_fields :
114115 reference_model = model ._fields [arg_name ]
115116 pk = from_global_id (args .pop (arg_name ))[- 1 ]
116117 reference_obj = reference_model .document_type_obj .objects (pk = pk ).get ()
117- args [arg_name ] = reference_obj
118+ reference_args [arg_name ] = reference_obj
118119
120+ args .update (reference_args )
119121 first = args .pop ('first' , None )
120122 last = args .pop ('last' , None )
121123 id = args .pop ('id' , None )
@@ -143,7 +145,6 @@ def get_query(cls, model, info, **args):
143145 # https://github.com/graphql-python/graphene-mongo/issues/20
144146 objs = objs [- (last + 1 ):]
145147
146- print (objs )
147148 return objs
148149
149150 # noqa
0 commit comments