From 4f52498a0688da67d93913a29795a994840d122f Mon Sep 17 00:00:00 2001 From: Syrus Akbary Date: Sat, 13 Aug 2016 17:32:07 -0700 Subject: [PATCH] Added required option in Argument --- graphene/types/argument.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/graphene/types/argument.py b/graphene/types/argument.py index b16aef47..964d42d6 100644 --- a/graphene/types/argument.py +++ b/graphene/types/argument.py @@ -2,12 +2,17 @@ from collections import OrderedDict from itertools import chain from ..utils.orderedtype import OrderedType +from .structures import NonNull class Argument(OrderedType): - def __init__(self, type, default_value=None, description=None, name=None, _creation_counter=None): + def __init__(self, type, default_value=None, description=None, name=None, required=False, _creation_counter=None): super(Argument, self).__init__(_creation_counter=_creation_counter) + + if required: + type = NonNull(type) + self.name = name self.type = type self.default_value = default_value