符號分割文字轉化為markdown表格形式

寫部落格,做筆記的時候,有時需要複製表格到markdown文件,數量一多一個個新增格式還是很麻煩,然後就寫了一個小工具,先將表格複製貼上到文字,用工具轉換,再複製貼上到markdown文件..

def to_table(filename='test', rept=':'):
    '''將符號分割文字轉化為markdown表格形式
    
    parameter:需要轉化的檔名(str),文字的分割符號(str)
    '''
    file_obj_r = open(filename, 'r')
    file_obj_w = open('new'+filename, 'w')
    count = 0  # 哨兵,實現檔案的第二行另做處理
    for line in file_obj_r:
        rec = line.split(rept)
        new_rec = [i.strip() for i in rec]
        new_rec = '|'+ '|'.join(new_rec) +'|'+'\n'
        file_obj_w.write(new_rec)
        if count == 0:
            head_rec = '|'
            for i in rec:
                head_rec += '---|'
            head_rec += '\n'
            file_obj_w.write(head_rec)
            count += 1
    file_obj_r.close()
    file_obj_w.close()


if __name__ == '__main__':
    import sys
    try:
        to_table(sys.argv[1],sys.argv[2])
    except:
        to_table()