Class Leetcode

src/leetcode.py:25–192

        return False

class Leetcode:
    def __init__(self):
        self.dict = self.init_db()
        self.finished = []
        self.flasks = []
        # read user
        p = util.get_root("user", "leetcode")
        entries = os.listdir(p)
        for k in entries:
            if k.endswith(".cpp") or k.endswith(".py"):
                self.finished.append(k)
            elif k.endswith(".md"):
                self.flasks.append(k)

    def init_db(self):
        d = SqliteDict(util.get_db('leetcode.sqlite'), autocommit=True)
        return d

    def close_db(self):
        self.dict.close()

    def get_tag_problems(self, tag):
        problems = self.get_all_problems()
        datas = [] 
        for k in problems:
            try:
                j = json.loads(problems[k])
                tags = j['data']['question']['topicTags']
                paid_only = j['data']['question']['paid_only']
                if len(tags) > 0:
                    for t in tags:
                        if t['slug'] == tag and paid_only == False:
                            datas.append(j)
                            break
            except Exception as e:
                print("unknow key:", k, e)
                pass
        return datas

    def get_all_problems(self):
        d = {}
        for k, v in self.dict.iteritems():
            if k.startswith("leetcode_") and k[9].isdigit():
                d[k] = v
        return d

    def save_problem(self, id, content):
        self.dict[leetcode_key(id)] = content
        self.dict.commit()

    def get_problem_content(self, id):
        v = self.dict.get(leetcode_key(id))
        return v

    def get_level(self, id):
        content = self.get_problem_content(id)
        if content == None:
            print("title not exist:", id)