Skip to content

Commit d89f823

Browse files
committed
test: Make both python2 and python3 work
1 parent f913cfa commit d89f823

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

graphene_mongo/fields.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)