Merge pull request #13 from prius/dev-problemset-query

Added new graphql request (no backwards compatibility from here)

Author: fspv

README.generated.md

@@ -223,11 +223,16 @@ Class | Method | HTTP request | Description
 
 ## Documentation For Models
 
+ - [AnyOfGraphqlQueryVariables](docs/AnyOfGraphqlQueryVariables.md)
  - [AnyOfGraphqlQuestionDetailSolution](docs/AnyOfGraphqlQuestionDetailSolution.md)
  - [BaseSubmissionResult](docs/BaseSubmissionResult.md)
  - [Difficulty](docs/Difficulty.md)
  - [GraphqlData](docs/GraphqlData.md)
+ - [GraphqlProblemsetQuestionList](docs/GraphqlProblemsetQuestionList.md)
  - [GraphqlQuery](docs/GraphqlQuery.md)
+ - [GraphqlQueryGetQuestionDetailVariables](docs/GraphqlQueryGetQuestionDetailVariables.md)
+ - [GraphqlQueryProblemsetQuestionListVariables](docs/GraphqlQueryProblemsetQuestionListVariables.md)
+ - [GraphqlQueryProblemsetQuestionListVariablesFilterInput](docs/GraphqlQueryProblemsetQuestionListVariablesFilterInput.md)
  - [GraphqlQueryVariables](docs/GraphqlQueryVariables.md)
  - [GraphqlQuestionCodeSnippet](docs/GraphqlQuestionCodeSnippet.md)
  - [GraphqlQuestionContributor](docs/GraphqlQuestionContributor.md)

docs/AnyOfGraphqlQueryVariables.md

@@ -0,0 +1,8 @@
+# AnyOfGraphqlQueryVariables
+
+## Properties
+Name | Type | Description | Notes
+------------ | ------------- | ------------- | -------------
+
+[[Back to Model list]](../README.md#documentation-for-models) [[Back to API list]](../README.md#documentation-for-api-endpoints) [[Back to README]](../README.md)
+

docs/GraphqlData.md

@@ -5,6 +5,7 @@ Name | Type | Description | Notes
 ------------ | ------------- | ------------- | -------------
 **question** | [**GraphqlQuestionDetail**](GraphqlQuestionDetail.md) |  | [optional] 
 **user** | [**GraphqlUser**](GraphqlUser.md) |  | [optional] 
+**problemset_question_list** | [**GraphqlProblemsetQuestionList**](GraphqlProblemsetQuestionList.md) |  | [optional] 
 
 [[Back to Model list]](../README.md#documentation-for-models) [[Back to API list]](../README.md#documentation-for-api-endpoints) [[Back to README]](../README.md)
 

docs/GraphqlProblemsetQuestionList.md

@@ -0,0 +1,10 @@
+# GraphqlProblemsetQuestionList
+
+## Properties
+Name | Type | Description | Notes
+------------ | ------------- | ------------- | -------------
+**total** | **int** |  | [optional] 
+**questions** | [**list[GraphqlQuestionDetail]**](GraphqlQuestionDetail.md) |  | [optional] 
+
+[[Back to Model list]](../README.md#documentation-for-models) [[Back to API list]](../README.md#documentation-for-api-endpoints) [[Back to README]](../README.md)
+

docs/GraphqlQuery.md

@@ -4,7 +4,7 @@
 Name | Type | Description | Notes
 ------------ | ------------- | ------------- | -------------
 **query** | **str** |  | 
-**variables** | [**GraphqlQueryVariables**](GraphqlQueryVariables.md) |  | 
+**variables** | [**AnyOfGraphqlQueryVariables**](AnyOfGraphqlQueryVariables.md) |  | 
 **operation_name** | **str** |  | [optional] 
 
 [[Back to Model list]](../README.md#documentation-for-models) [[Back to API list]](../README.md#documentation-for-api-endpoints) [[Back to README]](../README.md)

docs/GraphqlQueryGetQuestionDetailVariables.md

@@ -0,0 +1,9 @@
+# GraphqlQueryGetQuestionDetailVariables
+
+## Properties
+Name | Type | Description | Notes
+------------ | ------------- | ------------- | -------------
+**title_slug** | **str** |  | [optional] 
+
+[[Back to Model list]](../README.md#documentation-for-models) [[Back to API list]](../README.md#documentation-for-api-endpoints) [[Back to README]](../README.md)
+

docs/GraphqlQueryProblemsetQuestionListVariables.md

@@ -0,0 +1,12 @@
+# GraphqlQueryProblemsetQuestionListVariables
+
+## Properties
+Name | Type | Description | Notes
+------------ | ------------- | ------------- | -------------
+**category_slug** | **str** |  | [optional] 
+**limit** | **int** |  | [optional] 
+**skip** | **int** |  | [optional] 
+**filters** | [**GraphqlQueryProblemsetQuestionListVariablesFilterInput**](GraphqlQueryProblemsetQuestionListVariablesFilterInput.md) |  | [optional] 
+
+[[Back to Model list]](../README.md#documentation-for-models) [[Back to API list]](../README.md#documentation-for-api-endpoints) [[Back to README]](../README.md)
+

docs/GraphqlQueryProblemsetQuestionListVariablesFilterInput.md

@@ -0,0 +1,13 @@
+# GraphqlQueryProblemsetQuestionListVariablesFilterInput
+
+## Properties
+Name | Type | Description | Notes
+------------ | ------------- | ------------- | -------------
+**tags** | **list[str]** |  | [optional] 
+**difficulty** | **str** |  | [optional] 
+**status** | **str** |  | [optional] 
+**list_id** | **str** |  | [optional] 
+**premium_only** | **bool** |  | [optional] 
+
+[[Back to Model list]](../README.md#documentation-for-models) [[Back to API list]](../README.md#documentation-for-api-endpoints) [[Back to README]](../README.md)
+

docs/GraphqlQueryVariables.md

@@ -3,7 +3,6 @@
 ## Properties
 Name | Type | Description | Notes
 ------------ | ------------- | ------------- | -------------
-**title_slug** | **str** |  | [optional] 
 
 [[Back to Model list]](../README.md#documentation-for-models) [[Back to API list]](../README.md#documentation-for-api-endpoints) [[Back to README]](../README.md)
 

example.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import os
 import sys
 from time import sleep
@@ -10,9 +12,9 @@
 
 # NOTE: cookies var is just a dict with `csrftoken` and `LEETCODE_SESSION`
 # fields which contain corresponding cookies from web browser
-leetcode_session = os.environ["LEETCODE_SESSION_ID"]
+leetcode_session: str = os.environ["LEETCODE_SESSION_ID"]
 
-csrf_token = leetcode.auth.get_csrf_cookie(leetcode_session)
+csrf_token: str = leetcode.auth.get_csrf_cookie(leetcode_session)
 
 configuration.api_key["x-csrftoken"] = csrf_token
 configuration.api_key["csrftoken"] = csrf_token

leetcode/__init__.py

@@ -22,13 +22,26 @@
 from leetcode.configuration import Configuration
 
 # import models into sdk package
+from leetcode.models.any_of_graphql_query_variables import AnyOfGraphqlQueryVariables
 from leetcode.models.any_of_graphql_question_detail_solution import (
     AnyOfGraphqlQuestionDetailSolution,
 )
 from leetcode.models.base_submission_result import BaseSubmissionResult
 from leetcode.models.difficulty import Difficulty
 from leetcode.models.graphql_data import GraphqlData
+from leetcode.models.graphql_problemset_question_list import (
+    GraphqlProblemsetQuestionList,
+)
 from leetcode.models.graphql_query import GraphqlQuery
+from leetcode.models.graphql_query_get_question_detail_variables import (
+    GraphqlQueryGetQuestionDetailVariables,
+)
+from leetcode.models.graphql_query_problemset_question_list_variables import (
+    GraphqlQueryProblemsetQuestionListVariables,
+)
+from leetcode.models.graphql_query_problemset_question_list_variables_filter_input import (
+    GraphqlQueryProblemsetQuestionListVariablesFilterInput,
+)
 from leetcode.models.graphql_query_variables import GraphqlQueryVariables
 from leetcode.models.graphql_question_code_snippet import GraphqlQuestionCodeSnippet
 from leetcode.models.graphql_question_contributor import GraphqlQuestionContributor

leetcode/models/__init__.py

@@ -14,13 +14,26 @@
 from __future__ import absolute_import
 
 # import models into model package
+from leetcode.models.any_of_graphql_query_variables import AnyOfGraphqlQueryVariables
 from leetcode.models.any_of_graphql_question_detail_solution import (
     AnyOfGraphqlQuestionDetailSolution,
 )
 from leetcode.models.base_submission_result import BaseSubmissionResult
 from leetcode.models.difficulty import Difficulty
 from leetcode.models.graphql_data import GraphqlData
+from leetcode.models.graphql_problemset_question_list import (
+    GraphqlProblemsetQuestionList,
+)
 from leetcode.models.graphql_query import GraphqlQuery
+from leetcode.models.graphql_query_get_question_detail_variables import (
+    GraphqlQueryGetQuestionDetailVariables,
+)
+from leetcode.models.graphql_query_problemset_question_list_variables import (
+    GraphqlQueryProblemsetQuestionListVariables,
+)
+from leetcode.models.graphql_query_problemset_question_list_variables_filter_input import (
+    GraphqlQueryProblemsetQuestionListVariablesFilterInput,
+)
 from leetcode.models.graphql_query_variables import GraphqlQueryVariables
 from leetcode.models.graphql_question_code_snippet import GraphqlQuestionCodeSnippet
 from leetcode.models.graphql_question_contributor import GraphqlQuestionContributor

leetcode/models/any_of_graphql_query_variables.py

@@ -0,0 +1,87 @@
+# coding: utf-8
+
+"""
+    Leetcode API
+
+    Leetcode API implementation.  # noqa: E501
+
+    OpenAPI spec version: 1.0.1-1
+    Contact: pv.safronov@gmail.com
+    Generated by: https://github.com/swagger-api/swagger-codegen.git
+"""
+from __future__ import annotations
+
+import pprint
+import re  # noqa: F401
+
+import six
+
+
+class AnyOfGraphqlQueryVariables(object):
+    """NOTE: This class is auto generated by the swagger code generator program.
+
+    Do not edit the class manually.
+    """
+
+    """
+    Attributes:
+      swagger_types (dict): The key is attribute name
+                            and the value is attribute type.
+      attribute_map (dict): The key is attribute name
+                            and the value is json key in definition.
+    """
+    swagger_types = {}
+
+    attribute_map = {}
+
+    def __init__(self) -> None:  # noqa: E501
+        """AnyOfGraphqlQueryVariables - a model defined in Swagger"""  # noqa: E501
+        self.discriminator = None
+
+    def to_dict(self):
+        """Returns the model properties as a dict"""
+        result = {}
+
+        for attr, _ in six.iteritems(self.swagger_types):
+            value = getattr(self, attr)
+            if isinstance(value, list):
+                result[attr] = list(
+                    map(lambda x: x.to_dict() if hasattr(x, "to_dict") else x, value)
+                )
+            elif hasattr(value, "to_dict"):
+                result[attr] = value.to_dict()
+            elif isinstance(value, dict):
+                result[attr] = dict(
+                    map(
+                        lambda item: (item[0], item[1].to_dict())
+                        if hasattr(item[1], "to_dict")
+                        else item,
+                        value.items(),
+                    )
+                )
+            else:
+                result[attr] = value
+        if issubclass(AnyOfGraphqlQueryVariables, dict):
+            for key, value in self.items():
+                result[key] = value
+
+        return result
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model"""
+        return pprint.pformat(self.to_dict())
+
+    def __repr__(self) -> str:
+        """For `print` and `pprint`"""
+        return self.to_str()
+
+    def __eq__(self, other: AnyOfGraphqlQueryVariables) -> bool:
+        """Returns true if both objects are equal"""
+        if not isinstance(other, AnyOfGraphqlQueryVariables):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other) -> bool:
+        """Returns true if both objects are not equal"""
+        return not self == other

leetcode/models/graphql_data.py

@@ -30,19 +30,32 @@ class GraphqlData(object):
       attribute_map (dict): The key is attribute name
                             and the value is json key in definition.
     """
-    swagger_types = {"question": "GraphqlQuestionDetail", "user": "GraphqlUser"}
-
-    attribute_map = {"question": "question", "user": "user"}
-
-    def __init__(self, question=None, user=None) -> None:  # noqa: E501
+    swagger_types = {
+        "question": "GraphqlQuestionDetail",
+        "user": "GraphqlUser",
+        "problemset_question_list": "GraphqlProblemsetQuestionList",
+    }
+
+    attribute_map = {
+        "question": "question",
+        "user": "user",
+        "problemset_question_list": "problemsetQuestionList",
+    }
+
+    def __init__(
+        self, question=None, user=None, problemset_question_list=None
+    ) -> None:  # noqa: E501
         """GraphqlData - a model defined in Swagger"""  # noqa: E501
         self._question = None
         self._user = None
+        self._problemset_question_list = None
         self.discriminator = None
         if question is not None:
             self.question = question
         if user is not None:
             self.user = user
+        if problemset_question_list is not None:
+            self.problemset_question_list = problemset_question_list
 
     @property
     def question(self):
@@ -86,6 +99,27 @@ def user(self, user):
 
         self._user = user
 
+    @property
+    def problemset_question_list(self):
+        """Gets the problemset_question_list of this GraphqlData.  # noqa: E501
+
+
+        :return: The problemset_question_list of this GraphqlData.  # noqa: E501
+        :rtype: GraphqlProblemsetQuestionList
+        """
+        return self._problemset_question_list
+
+    @problemset_question_list.setter
+    def problemset_question_list(self, problemset_question_list):
+        """Sets the problemset_question_list of this GraphqlData.
+
+
+        :param problemset_question_list: The problemset_question_list of this GraphqlData.  # noqa: E501
+        :type: GraphqlProblemsetQuestionList
+        """
+
+        self._problemset_question_list = problemset_question_list
+
     def to_dict(self):
         """Returns the model properties as a dict"""
         result = {}