Merge pull request #59 from ricekot/update-graphql

Update GraphQL API

Author: kingthorin

CHANGELOG.md

@@ -4,7 +4,8 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 
 ## [Unreleased]
-
+### Changed
+- Update APIs from GraphQL add-on.
 
 ## [0.0.17] - 2020-10-14
 ### Added

src/zapv2/graphql.py

@@ -27,6 +27,54 @@ class graphql(object):
     def __init__(self, zap):
         self.zap = zap
 
+    @property
+    def option_args_type(self):
+        """
+        Returns how arguments are currently specified.
+        This component is optional and therefore the API will only work if it is installed
+        """
+        return six.next(six.itervalues(self.zap._request(self.zap.base + 'graphql/view/optionArgsType/')))
+
+    @property
+    def option_max_args_depth(self):
+        """
+        Returns the current maximum arguments generation depth.
+        This component is optional and therefore the API will only work if it is installed
+        """
+        return six.next(six.itervalues(self.zap._request(self.zap.base + 'graphql/view/optionMaxArgsDepth/')))
+
+    @property
+    def option_max_query_depth(self):
+        """
+        Returns the current maximum query generation depth.
+        This component is optional and therefore the API will only work if it is installed
+        """
+        return six.next(six.itervalues(self.zap._request(self.zap.base + 'graphql/view/optionMaxQueryDepth/')))
+
+    @property
+    def option_optional_args_enabled(self):
+        """
+        Returns whether or not optional arguments are currently specified.
+        This component is optional and therefore the API will only work if it is installed
+        """
+        return six.next(six.itervalues(self.zap._request(self.zap.base + 'graphql/view/optionOptionalArgsEnabled/')))
+
+    @property
+    def option_query_split_type(self):
+        """
+        Returns the current level for which a single query is generated.
+        This component is optional and therefore the API will only work if it is installed
+        """
+        return six.next(six.itervalues(self.zap._request(self.zap.base + 'graphql/view/optionQuerySplitType/')))
+
+    @property
+    def option_request_method(self):
+        """
+        Returns the current request method.
+        This component is optional and therefore the API will only work if it is installed
+        """
+        return six.next(six.itervalues(self.zap._request(self.zap.base + 'graphql/view/optionRequestMethod/')))
+
     def import_file(self, endurl, file, apikey=''):
         """
         Imports a GraphQL Schema from a File.
@@ -43,3 +91,45 @@ def import_url(self, endurl, url=None, apikey=''):
         if url is not None:
             params['url'] = url
         return six.next(six.itervalues(self.zap._request(self.zap.base + 'graphql/action/importUrl/', params)))
+
+    def set_option_args_type(self, string, apikey=''):
+        """
+        Sets how arguments are specified.
+        This component is optional and therefore the API will only work if it is installed
+        """
+        return six.next(six.itervalues(self.zap._request(self.zap.base + 'graphql/action/setOptionArgsType/', {'String': string, 'apikey': apikey})))
+
+    def set_option_query_split_type(self, string, apikey=''):
+        """
+        Sets the level for which a single query is generated.
+        This component is optional and therefore the API will only work if it is installed
+        """
+        return six.next(six.itervalues(self.zap._request(self.zap.base + 'graphql/action/setOptionQuerySplitType/', {'String': string, 'apikey': apikey})))
+
+    def set_option_request_method(self, string, apikey=''):
+        """
+        Sets the request method.
+        This component is optional and therefore the API will only work if it is installed
+        """
+        return six.next(six.itervalues(self.zap._request(self.zap.base + 'graphql/action/setOptionRequestMethod/', {'String': string, 'apikey': apikey})))
+
+    def set_option_max_args_depth(self, integer, apikey=''):
+        """
+        Sets the maximum arguments generation depth.
+        This component is optional and therefore the API will only work if it is installed
+        """
+        return six.next(six.itervalues(self.zap._request(self.zap.base + 'graphql/action/setOptionMaxArgsDepth/', {'Integer': integer, 'apikey': apikey})))
+
+    def set_option_max_query_depth(self, integer, apikey=''):
+        """
+        Sets the maximum query generation depth.
+        This component is optional and therefore the API will only work if it is installed
+        """
+        return six.next(six.itervalues(self.zap._request(self.zap.base + 'graphql/action/setOptionMaxQueryDepth/', {'Integer': integer, 'apikey': apikey})))
+
+    def set_option_optional_args_enabled(self, boolean, apikey=''):
+        """
+        Sets whether or not Optional Arguments should be specified.
+        This component is optional and therefore the API will only work if it is installed
+        """
+        return six.next(six.itervalues(self.zap._request(self.zap.base + 'graphql/action/setOptionOptionalArgsEnabled/', {'Boolean': boolean, 'apikey': apikey})))