leetcode_table_maker.py

import os, pickle

class Table_maker:
    def __init__(self):
        self.path_from = "solutions"
        self.problems = sorted(os.listdir(self.path_from), key=lambda x : int(x))
        with open('problems_info', 'rb') as f:
            self.problems_info = pickle.load(f)

    def make_table(self, category):
        table = []
        table.append("| # | Title | Solutions | Analyses | Difficulty | Note |")
        table.append("| :---: | :---: | :---: | :---: | :---: | :---: |")
        for problem in self.problems:
            index = int(problem)
            if self.problems_info[index]['Category'] != category:
                continue
            row = []
            Analysis = []
            Solution = []
            for file in os.listdir(os.path.join(self.path_from, problem)):
                fmt = file.split('.')[-1]
                if fmt in ["md", "txt", "pdf"]:
                    Analysis.append(f'[{fmt}](solutions/{problem}/{file})')
                else:
                    Solution.append(f'[{fmt}](solutions/{problem}/{file})')

            Title = self.problems_info[index]['Title']
            link = self.problems_info[index]['href']
            Difficulty = self.problems_info[index]['Difficulty']
            Note = self.problems_info[index]['Note'] + " |"

            row.append("| " + problem)
            row.append(f'[{Title}]({link})')
            row.append(', '.join(sorted(Solution)))
            row.append(', '.join(sorted(Analysis)))
            row.append(Difficulty)
            row.append(Note)

            table.append(' | '.join(row))
        return table

def get_contents():
    maker = Table_maker()
    Algorithms_table = maker.make_table("Algorithms")
    Database_table = maker.make_table("Database")
    Shell_table = maker.make_table("Shell")
    Concurrency_table = maker.make_table("Concurrency")

    contents = []
    contents.append("## Algorithms")
    contents.extend(Algorithms_table)
    contents.append("\n")

    contents.append("## Database")
    contents.extend(Database_table)
    contents.append("\n")

    contents.append("## Shell")
    contents.extend(Shell_table)
    contents.append("\n")

    contents.append("## Concurrency")
    contents.extend(Concurrency_table)
    contents.append("\n")

    return "\n".join(contents)

if __name__ == "__main__":
    print(get_contents())