makeProblemSets.py

import json
from collections import defaultdict
import copy
from getQuestionFromLC import getAllQuestions


def setProblems(setlist):
    problemNames = open(f"problemSetInfo/{setlist}problems.txt", "r")
    problemTopics = open(f"problemSetInfo/{setlist}Topics.txt", "r")
    probList = problemNames.read().split("\n")
    topicList = problemTopics.read().split("\n")
    topics = {}
    problemTopic = {}


    i = 0
    for b in topicList:
        topics[b] = []
        while i < len(probList) and probList[i] != "":
            topics[b].append(probList[i])
            problemTopic[probList[i].lower()] = b
            i += 1
        i += 1

    problemInfo = defaultdict(lambda: [])
    setlistProblems = {"ID": {}, "NAME": {}}
    probsFound = set()
    allQuestions = getAllQuestions()
    for problem in problemTopic:  # purposefully inefficient to preserve order, not really that important to optimize
        isFound = False
        for i, q in enumerate(allQuestions):

            if q["title"].lower() == problem.lower():
                newProblem = {}
                newProblem["TITLE"] = q["title"]
                newProblem["URL"] = "https://leetcode.com/problems/" + q["titleSlug"]
                newProblem["DIFFICULTY"] = q["difficulty"]
                setlistProblems["ID"][i+1] = copy.deepcopy(newProblem)
                setlistProblems["NAME"][newProblem["TITLE"]] = i + 1
                newProblem["ID"] = i + 1

                topic = problemTopic[q["title"].lower()]
                problemInfo[topic].append(newProblem)
                isFound = True
        if not isFound:
            print("NOT FOUND: ", problem)

    with open(f"problemSetInfo/{setlist}problems.json", "w") as writeFile:
        json.dump(problemInfo, writeFile)

    if setlist == "neetcode150":
        with open(f"problemSetInfo/setlistProblemIDs.json", "w") as writeFile:
            json.dump(setlistProblems, writeFile)
    elif setlist == "blind75":
        i = 0
        inBlind75 = set(setlistProblems["ID"].keys())
        print(inBlind75)
        with open(f"problemSetInfo/setlistProblemIDs.json", "r") as readFile:
            setlistProblems = json.load(readFile)
        for s in setlistProblems["ID"]:
            print(s)
            if int(s) in inBlind75:
                setlistProblems["ID"][s]["INBLIND75"] = True
                i += 1
            else:
                setlistProblems["ID"][s]["INBLIND75"] = False
        with open(f"problemSetInfo/setlistProblemIDs.json", "w") as writeFile:
            json.dump(setlistProblems, writeFile)
        print(i)

# setProblems("neetcode150")
# setProblems("blind75")

def makeTitleSlugToID():
    with open(f"problemSetInfo/setlistProblemIDs.json", "r") as readFile:
        setlist = json.load(readFile)
    titleSlugToID = {}
    for id in setlist["ID"]:
        url = setlist["ID"][id]["URL"]
        titleslug = url[url.find("s/") + 2:]
        titleSlugToID[titleslug] = id
    setlist["NAME"] = titleSlugToID
    with open(f"problemSetInfo/setlistProblemIDs.json", "w") as writeFile:
        json.dump(setlist, writeFile)


makeTitleSlugToID()