LeetCode id -> title slug

KM_LeetCode_ID.py

#!/opt/bin/python3

import os
import requests

question_id = os.environ.get('KMVAR_LeetCodeID', 'add-two-numbers').strip()

cookies={
    "LEETCODE_SESSION": ''
}

def get_csrf_cookie() -> str:
    response = requests.get(
        "https://leetcode.com/",
        cookies=cookies,
    )

    return response.cookies["csrftoken"]

HEADERS = {
    'x-csrftoken': get_csrf_cookie(),
}




def leetcode_id_to_slug(num_id: int):
    query = '''query problemsetQuestionList($categorySlug: String, $limit: Int, $skip: Int, $filters: QuestionListFilterInput) {
  problemsetQuestionList(
    categorySlug: $categorySlug
    limit: $limit
    skip: $skip
    filters: $filters
  ) {
    hasMore
    total
    questions {
      acRate
      difficulty
      freqBar
      frontendQuestionId
      isFavor
      paidOnly
      solutionNum
      status
      title
      titleCn
      titleSlug
      topicTags {
        name
        nameTranslated
        id
        slug
      }
      extra {
        hasVideoSolution
        topCompanyTags {
          imgUrl
          slug
          numSubscribed
        }
      }
    }
  }
}'''

    request_body = {
        'query': query,
        'variables': {
            'categorySlug': '',
            'filters': {
                'searchKeywords': '{}.'.format(num_id)
            },
        }
    }


    response = requests.post('https://leetcode-cn.com/graphql/', headers=HEADERS, cookies=cookies, json=request_body)
    
    questions = response.json()['data']['problemsetQuestionList']['questions']
    
    for question in questions:
        if str(question['frontendQuestionId']) == str(num_id):
            return question['titleSlug']

    return questions[0]['titleSlug']

if question_id.isnumeric():
    question_id = leetcode_id_to_slug(int(question_id))

print(question_id)