Skip to content
Snippets Groups Projects
Commit b6ad3e42 authored by Albert Feghaly's avatar Albert Feghaly
Browse files

Add dbid filter argument to dataset in GraphQL

parent b787e5ef
No related branches found
No related tags found
1 merge request!79Resolve "GraphQL select files using severals annotations (key and values)"
...@@ -38,7 +38,7 @@ class UserType(DjangoObjectType): ...@@ -38,7 +38,7 @@ class UserType(DjangoObjectType):
# Create a Query type # Create a Query type
class Query(graphene.ObjectType): class Query(graphene.ObjectType):
datafile = graphene.Field(DataFileType, id=graphene.String(), dbid=graphene.Int()) 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()) datafiles = graphene.List(DataFileType, key=graphene.String(), value=graphene.String(), dataset=graphene.String())
datasets = graphene.List(DataSetType) datasets = graphene.List(DataSetType)
...@@ -53,9 +53,12 @@ class Query(graphene.ObjectType): ...@@ -53,9 +53,12 @@ class Query(graphene.ObjectType):
def resolve_dataset(self, info, **kwargs): def resolve_dataset(self, info, **kwargs):
id = kwargs.get('id') id = kwargs.get('id')
dbid = kwargs.get('dbid')
if id is not None: if id is not None:
return DataSet.objects.get(iric_data_id=id) return DataSet.objects.get(iric_data_id=id)
elif dbid is not None:
return DataSet.objects.get(pk=dbid)
return None return None
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment