Skip to content

Commit fc9dcea

Browse files
Merge pull request #211 from yunkst/support_db_alias
support db_alias
2 parents 9e9e5db + 60550a5 commit fc9dcea

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

graphene_mongo/fields.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -399,7 +399,10 @@ def default_resolver(self, _root, info, required_fields=None, resolved=None, **a
399399
args_copy[key] = args_copy[key].value
400400

401401
if PYMONGO_VERSION >= (3, 7):
402-
count = (mongoengine.get_db()[self.model._get_collection_name()]).count_documents(args_copy)
402+
if hasattr(self.model,'_meta') and 'db_alias' in self.model._meta:
403+
count = (mongoengine.get_db(self.model._meta['db_alias'])[self.model._get_collection_name()]).count_documents(args_copy)
404+
else:
405+
count = (mongoengine.get_db()[self.model._get_collection_name()]).count_documents(args_copy)
403406
else:
404407
count = self.model.objects(args_copy).count()
405408
if count != 0:

0 commit comments

Comments
 (0)