leetcode_graphql.py

import requests
import time
import json
import sys

url = "https://leetcode.com/graphql/"

problem = "add-two-numbers"

def getTopicID(problem, target_lang="python3"):
    variables = {
        "query": "",                         # optional search string
        "languageTags": [target_lang],         # filter by language
        "topicTags": [problem],      # filter by topic
        "questionSlug": problem,           # slug of the problem
        "skip": 0,                           # pagination offset
        "first": 1,                          # number of results
        "orderBy": "hot"                     # sort option (e.g. "hot", "newest")
    }

    # GraphQL query string
    query = """
    query communitySolutions(
    $questionSlug: String!,
    $skip: Int!,
    $first: Int!,
    $query: String,
    $orderBy: TopicSortingOption,
    $languageTags: [String!],
    $topicTags: [String!]
    ) {
    questionSolutions(
        filters: {
        questionSlug: $questionSlug,
        skip: $skip,
        first: $first,
        query: $query,
        orderBy: $orderBy,
        languageTags: $languageTags,
        topicTags: $topicTags
        }
    ) {
        hasDirectResults
        totalNum
        solutions {
        id
        title
        commentCount
        viewCount
        solutionTags {
            name
            slug
        }
        post {
            id
            status
            voteCount
            creationDate
            author {
            username
            profile {
                userAvatar
                reputation
            }
            }
        }
        }
    }
    }
    """

    # Build payload
    payload = {
        "query": query,
        "variables": variables
    }

    headers = {"Content-Type": "application/json"}
    
    response = extractResponse(headers, payload, "community.json", "all_community.json")
    return response["data"]["questionSolutions"]["solutions"][0]["id"]


def getContent(saveContent = True, path1="content.json", path2="all_content.json"):
    # 1. Load JSON file
    with open("solution.json", "r", encoding="utf-8") as f:
        data = json.load(f)

    # 2. Extract the content field
    content = data["data"]["topic"]["post"]["content"]

    # 3. Decode escape sequences so \n -> actual newlines
    # First encode to bytes, then decode with "unicode_escape"
    #cleaned_content = content.encode("utf-8").decode("unicode_escape")
    
    try:
        # Wrap in quotes and use json.loads to handle all escape sequences properly
        cleaned_content = json.loads(f'"{content}"')
    except json.JSONDecodeError:
        # Fallback to manual replacement if json.loads fails
        cleaned_content = content.replace('\\n', '\n').replace('\\t', '\t').replace('\\"', '"').replace('\\\\', '\\')
    
    
    full_content = {
        problem: {
            "content": cleaned_content
        }
    }

    if saveContent:
        with open("content.txt", "w", encoding="utf-8") as f:
            f.write(cleaned_content)
            
        # Load existing content or create empty dict if file doesn't exist
        try:
            with open(path2, "r", encoding="utf-8") as f:
                all_content = json.load(f)
        except (FileNotFoundError, json.JSONDecodeError):
            all_content = {}
        
        # Update the dictionary with new content (this will merge, not overwrite)
        all_content.update(full_content)
        
        with open(path2, "w", encoding="utf-8") as f:
            json.dump(all_content, f, indent=4, ensure_ascii=False)
    
    # 4. Print nicely
    #print(cleaned_content)


def getSolution(topicId):
    # define your variable here
    #topicId = 3678229  

    # build the query and variables as Python objects
    payload = {
        "query": """
        query communitySolution($topicId: Int!) {
        topic(id: $topicId) {
            id
            viewCount
            topLevelCommentCount
            subscribed
            title
            pinned
            solutionTags {
            name
            slug
            }
            hideFromTrending
            commentCount
            isFavorite
            post {
            id
            voteCount
            voteStatus
            content
            updationDate
            creationDate
            status
            isHidden
            author {
                isDiscussAdmin
                isDiscussStaff
                username
                nameColor
                activeBadge {
                displayName
                icon
                }
                profile {
                userAvatar
                reputation
                }
                isActive
            }
            authorIsModerator
            isOwnPost
            }
        }
        }
        """,
        "variables": {"topicId": topicId}
    }

    headers = {
        "Content-Type": "application/json"
    }
    
    return extractResponse(headers, payload, "solution.json", "all_solutions.json")
    
    
def extractResponse(headers, payload, path1, path2):
    # use json= to auto-convert dict to JSON
    time.sleep(0.5)
    response = requests.post(url, headers=headers, json=payload)

    data = response.json()

    with open(path1, "w", encoding="utf-8") as f:
        json.dump(data, f, indent=4)
        
    solutions = []
    with open(path2, "r", encoding="utf-8") as f:
        solutions = json.load(f)
    
    solutions.append(data)
    
    with open(path2, "w", encoding="utf-8") as f:
        json.dump(solutions, f, indent=4)
        
    #print(json.dumps(data, indent=4))

    return data

def main(arg=None, problem_slug=None, target_lang="python"):
    # Use provided problem_slug or default to global 'problem'
    global problem
    problem = problem_slug

    if arg is None:
        topicID = getTopicID(problem)
        getSolution(topicID)
        getContent()
    elif arg == "1":
        getContent()
    elif arg == "2":
        topicID = getTopicID(problem)
        getSolution(topicID)
    else:
        print("Usage: python leetcode-graphql.py [1|2]")
        
    return 1
    
if __name__ == "__main__":
    arg = sys.argv[1] if len(sys.argv) > 1 else None
    main(arg)
    #topicID = getTopicID("two-sum")
    #solution = getSolution(topicID)
    #getContent()