tool.py

#!/usr/bin/env python

import glob
# from xlrd import open_workbook

def combiner(file_list):
    f = open('solutions.md', 'w')
    for i, answer in enumerate(file_list):
        with open(answer) as file:
            title = answer.split('/')[-1].split('.py')[0].split('_')
            folder = answer.split('/')[2]
            content = file.read()
            sections = content.split("\"\"\"")
            if len(sections) < 3 or len(sections[2]) < 10:
                continue
            if folder == 'Leetcode':
                whole_name = answer.split('/')[-1].split('.py')[0]
                leetcode_num = whole_name.split('_')[0]
                leetcode_title = ' '.join(whole_name.split('_')[1:])
                leetcode_address = leetcode_title.replace(' ', '-').lower()
                f.write('### [%d. %s](https://oj.leetcode.com/problems/%s/)\n' % (int(leetcode_num), leetcode_title, leetcode_address))
            else:
                f.write('### %d. %s\n' % (i+1,title))
            if '\\' in sections[1]:
                new_section = sections[1].split('\n')
                min_index = len(new_section) - 1
                max_index = 0
                for i, line in enumerate(new_section):
                    if '\\' in line or '/' in line:
                        min_index = min(min_index, i)
                        max_index = max(max_index, i)
                new_section.insert(min_index - 1, '```')
                new_section.insert(max_index + 3, '```')
                f.write('\n'.join(new_section))
            else:
                f.write(sections[1])
            f.write('\n```python')
            f.write(''.join(sections[2:]))
            f.write('```\n')
            f.write('-----\n\n')
    f.close()

def title_convert(title):
    title = str(title)
    title = title.translate(None, "()'")
    title = title.replace(', ', '-')
    return title.lower()

def frequency_creator(file_list):
    f = open('frequency.md', 'w')
    f.write('## Leetcode Order by Frequency\n')
    book = open_workbook('Leetcode Order by Frequency.xlsx')
    sheet = book.sheet_by_index(0)

    titles = {}
    for file_name in file_list:
        title = ' '.join(file_name.split('/')[-1].split('.py')[0].split('_')).lower()
        titles[title] = file_name

    for row_index in range(sheet.nrows):
        if row_index == 1:
            f.write('|---|:---:|---:|---:|---|---|\n')
        f.write('| ')
        for col_index in range(sheet.ncols):
            if row_index != 0 and col_index in [0,2,3]:
                f.write(str(sheet.cell(row_index,col_index).value).split('.')[0])
            elif row_index != 0 and col_index == 1:
                name = str(sheet.cell(row_index, 1).value)
                file_name = titles.get(title_convert(name), None)
                link = '[%s](%s)' % (name, file_name)
                f.write(link)
            else:
                f.write(str(sheet.cell(row_index,col_index).value))
            if col_index == 5:
                f.write(' |')
            else:
                f.write(' | ')
        f.write('\n')
    f.close()

# type 1 == Data Structure, 2 == Algorithms
def type_searchor(target, type):
    book = open_workbook('Leetcode Order by Frequency.xlsx')
    sheet = book.sheet_by_index(0)
    if type == 1:
        col_index = 4
    else:
        col_index = 5
    for row_index in range(sheet.nrows):
        if target in str(sheet.cell(row_index, col_index).value):
            print(str(sheet.cell(row_index, 1).value))


if __name__ == '__main__':
    file_list = glob.glob('./Ninja/Leetcode/*.py')
    file_list.remove('./Ninja/Leetcode/__init__.py')
    file_list.sort()
    interview_list = glob.glob('./Interviews/*.py')
    interview_list.sort()
    file_list.extend(interview_list)

    combiner(file_list)
    # frequency_creator(file_list)
    # type 1 == Data Structure, 2 == Algorithms
    # type_searchor('dp', 2)