Add get_url_of_login_page to LeetCodeService

To enable logging in to LeetCode using
https://github.com/online-judge-tools/oj/

Author: usk83

onlinejudge/service/leetcode.py

@@ -55,6 +55,9 @@ def from_url(cls, url: str) -> Optional['LeetCodeService']:
             return None
         return cls()
 
+    def get_url_of_login_page(self) -> str:
+        return f'{self.get_url()}accounts/login/'
+
     def is_logged_in(self, *, session: Optional[requests.Session] = None) -> bool:
         session = self._set_request_header(session)
         json_body = {

request_test.py

@@ -0,0 +1,70 @@
+import json
+import requests
+
+s = requests.Session()
+
+cookies = {
+    'LEETCODE_SESSION': 'eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJfYXV0aF91c2VyX2lkIjoiMzQ4NjE5MSIsIl9hdXRoX3VzZXJfYmFja2VuZCI6ImRqYW5nby5jb250cmliLmF1dGguYmFja2VuZHMuTW9kZWxCYWNrZW5kIiwiX2F1dGhfdXNlcl9oYXNoIjoiYjBhNDA5NmNhMWFkZTc4ZDk1ZDA2ZTgzZDhiMmI2YjYzOWIwNzE1ZiIsImlkIjozNDg2MTkxLCJlbWFpbCI6Imx0LWNsaS1kZXZAc3V0ZS5qcCIsInVzZXJuYW1lIjoibHQtY2xpLWRldiIsInVzZXJfc2x1ZyI6Imx0LWNsaS1kZXYiLCJhdmF0YXIiOiJodHRwczovL3d3dy5ncmF2YXRhci5jb20vYXZhdGFyLzZiM2QyMmRkZGY5NjlhYjNjOTdlYTk1NDE1OTU1NjM2LnBuZz9zPTIwMCIsInJlZnJlc2hlZF9hdCI6MTYwMjgxMTkxNSwiaXAiOiI2NC40Ni4xMC4zOCIsImlkZW50aXR5IjoiZmZhZWE1YWZlZDUxZjFhMWIyY2RjOTk2YWVkNTQ1YmUiLCJzZXNzaW9uX2lkIjoyOTMwNjY1LCJfc2Vzc2lvbl9leHBpcnkiOjEyMDk2MDB9.MW-LzxO5l1HPsw3fKOtd2A1n3HJF47HywD3j0CU-0Kw',
+    'csrftoken': 'Lhp4Arh6FxIXPGh59WOdGz1eOqa3AVAzoQDbzjWSGfBTlLCz8gk2SpnMVguR7b75',
+}
+
+headers = {
+    'X-CSRFToken': 'Lhp4Arh6FxIXPGh59WOdGz1eOqa3AVAzoQDbzjWSGfBTlLCz8gk2SpnMVguR7b75',
+    'Origin': 'https://leetcode.com',
+    'Referer': 'https://leetcode.com/',
+    # 'Referer': 'https://leetcode.com/problems/two-sum/',
+    # 'Referer': 'https://leetcode.com/problems/sellers-with-no-sales/',
+    # 'Referer': 'https://leetcode.com/problems/split-two-strings-to-make-palindrome/',
+    'Content-type': 'application/json',
+}
+
+# query = '\n'.join([
+#   'query getQuestionDetail($titleSlug: String!) {',
+#   '  question(titleSlug: $titleSlug) {',
+#   'codeSnippets {',
+#      'lang',
+#      'langSlug',
+#   '}',
+#   # 'title',
+#   # 'questionFrontendId',
+#   # 'sampleTestCase',
+#   # 'metaData',
+#   # 'isPaidOnly',
+#   # '    content',
+#   # '    stats',
+#   # '    codeDefinition',
+#   # '    sampleTestCase',
+#   # '    enableRunCode',
+#   # '    metaData',
+#   # '    translatedContent',
+#   '  }',
+#   '}',
+# ])
+
+# json_data = {
+#     'operationName': 'getQuestionDetail',
+#     'query': query,
+#     'variables': {
+#         # 'titleSlug': 'two-sum',
+#         # 'titleSlug': 'sellers-with-no-sales',
+#         'titleSlug': 'split-two-strings-to-make-palindrome',
+#     },
+# }
+
+json_data = {
+    'operationName': 'globalData',
+    'query': 'query globalData {\n  feature {\n    questionTranslation\n    subscription\n    signUp\n    discuss\n    mockInterview\n    contest\n    store\n    chinaProblemDiscuss\n    socialProviders\n    studentFooter\n    enableChannels\n    dangerZone\n    cnJobs\n    enableSharedWorker\n    enableRecaptchaV3\n    enableDebugger\n    enableDebuggerPremium\n    enableAutocomplete\n    enableAutocompletePremium\n    enableAllQuestionsRaw\n    autocompleteLanguages\n    enableIndiaPricing\n    enableReferralDiscount\n    maxTimeTravelTicketCount\n    __typename\n  }\n  userStatus {\n    isSignedIn\n    isAdmin\n    isStaff\n    isSuperuser\n    isTranslator\n    isPremium\n    isVerified\n    checkedInToday\n    username\n    realName\n    userSlug\n    avatar\n    optedIn\n    requestRegion\n    region\n    activeSessionId\n    permissions\n    notificationStatus {\n      lastModified\n      numUnread\n      __typename\n    }\n    completedFeatureGuides\n    __typename\n  }\n  siteRegion\n  chinaHost\n  websocketUrl\n  recaptchaKey\n  recaptchaKeyV2\n  sitewideAnnouncement\n  userCountryCode\n}\n',
+}
+
+{'data': {'feature': {'questionTranslation': False, 'subscription': True, 'signUp': True, 'discuss': True, 'mockInterview': True, 'contest': True, 'store': True, 'chinaProblemDiscuss': False, 'socialProviders': '[{"id": "linkedin_oauth2", "login_url": "/accounts/linkedin_oauth2/login/", "deprecated": true}, {"id": "google", "login_url": "/accounts/google/login/", "deprecated": false}, {"id": "github", "login_url": "/accounts/github/login/", "deprecated": false}, {"id": "facebook", "login_url": "/accounts/facebook/login/", "deprecated": false}]', 'studentFooter': True, 'enableChannels': True, 'dangerZone': True, 'cnJobs': False, 'enableSharedWorker': True, 'enableRecaptchaV3': True, 'enableDebugger': True, 'enableDebuggerPremium': True, 'enableAutocomplete': True, 'enableAutocompletePremium': True, 'enableAllQuestionsRaw': True, 'autocompleteLanguages': '["c", "cpp", "java", "python", "python3", "javascript"]', 'enableIndiaPricing': True, 'enableReferralDiscount': True, 'maxTimeTravelTicketCount': 3, '__typename': 'FeatureNode'}, 'userStatus': {'isSignedIn': True, 'isAdmin': False, 'isStaff': False, 'isSuperuser': False, 'isTranslator': False, 'isPremium': False, 'isVerified': True, 'checkedInToday': True, 'username': 'lt-cli-dev', 'realName': '', 'userSlug': 'lt-cli-dev', 'avatar': 'https://www.gravatar.com/avatar/6b3d22dddf969ab3c97ea95415955636.png?s=200', 'optedIn': True, 'requestRegion': 'US', 'region': '', 'activeSessionId': 3701518, 'permissions': [], 'notificationStatus': {'lastModified': 1602738260, 'numUnread': 1, '__typename': 'UserNotificationNode'}, 'completedFeatureGuides': ['QUESTION_DETAIL_TOUR'], '__typename': 'MeNode'}, 'siteRegion': 'US', 'chinaHost': 'https://leetcode-cn.com', 'websocketUrl': 'wss://sockets.leetcode.com/ws/', 'recaptchaKey': '6LdBpsIUAAAAAKAYWjZfIpn4cJHVIk_tsmxpl7cz', 'recaptchaKeyV2': '6LdBX8MUAAAAAAI4aZHi1C59OJizaJTvPNvWH2wz', 'sitewideAnnouncement': None, 'userCountryCode': 'CA'}}
+
+
+# json_data = {
+#     'lang': 'golang',
+#     'question_id': '1',
+#     'typed_code': 'func twoSum(nums []int, target int) []int {\n\tcache := map[int]int{}\n\tfor index, num := range nums {\n\t\tif pairIndex, ok := cache[target-num]; ok {\n\t\t\treturn []int{pairIndex, index}\n\t\t}\n\t\tcache[num] = index\n\t}\n\treturn nil\n}\n',
+# }
+
+resp = s.post('https://leetcode.com/graphql', cookies=cookies, headers=headers, json=json_data)
+# resp = s.post('https://leetcode.com/problems/two-sum/submit/', cookies=cookies, headers=headers, json=json_data)
+print(resp.json())