Add dbid filter argument to dataset in GraphQL

portal/graphql_schema.py

@@ -38,7 +38,7 @@ class UserType(DjangoObjectType):
 # Create a Query type
 class Query(graphene.ObjectType):
     datafile = graphene.Field(DataFileType, id=graphene.String(), dbid=graphene.Int())
-    dataset = graphene.Field(DataSetType, id=graphene.String())
+    dataset = graphene.Field(DataSetType, id=graphene.String(), dbid=graphene.Int())
     datafiles = graphene.List(DataFileType, key=graphene.String(), value=graphene.String(), dataset=graphene.String())
     datasets = graphene.List(DataSetType)
 
@@ -53,9 +53,12 @@ class Query(graphene.ObjectType):
 
     def resolve_dataset(self, info, **kwargs):
         id = kwargs.get('id')
+        dbid = kwargs.get('dbid')
 
         if id is not None:
             return DataSet.objects.get(iric_data_id=id)
+        elif dbid is not None:
+            return DataSet.objects.get(pk=dbid)
 
         return None