歩いたら休め

If the implementation is easy to explain, it may be a good idea.

【メモ】Pythonのosモジュールででtreeコマンドっぽい出力

windowsの共有フォルダの中身を、ファイルに書き出すスクリプトです。

バイト先でフォルダ構成を整理するため、共有フォルダの中身をExcelファイルに書き出してほしいと言われ、Pythonのosモジュールを使って楽しました。

# -*- coding: utf-8 -*-
import os

exportfile = "file.txt"
searchdir = "." #current directory

#def ls(directory):
#    tmp = os.listdir(directory)
#    list = []
#    for x in tmp:
#        if os.path.isfile(directory+"/"+x): #ls -F
#            list.append(x)
#    return list

g = os.walk(searchdir)
with open(exportfile,mode="w") as f:
    for dirpath, dirs, files in g:
        paths = dirpath.split("\\") #windows use "\\"
        f.write("\t"*(len(paths)-2)+ paths[-1] + "\n")
        #files = ls(dirpath)
        for x in files:
            f.write("\t"*(len(paths)-1)+ x + "\n")