# coding=utf-8
import os.path
import sys
import time
link1 = "─┬─"
link2 = "├──"
link3 = "───"
link4 = "└──"
link5 = " │ "
link6 = " "
link7 = "│ "
class File(object):
def __init__(self, abs_path, name, children=[]):
self.abs_path = abs_path
self.name = name
self.children = children
self.next = None
def __repr__(self) -> str:
return str({'name': self.name,
'children': self.children
})
def create_path(parent, file):
return parent + os.sep + file
def sub_dirs(path):
return [x for x in os.listdir(path) if os.path.isdir(create_path(path, x))]
def sub_dirs_abs(path):
return [create_path(path, x) for x in os.listdir(path) if os.path.isdir(create_path(path, x))]
def dir_tree(path):
parent = File(path, os.path.basename(path), [])
dirs = sub_dirs_abs(path)
for dir in dirs:
kid = dir_tree(dir)
parent.children.append(kid)
for i in range(len(parent.children) - 1):
parent.children[i].next = parent.children[i + 1]
return parent
def space(num=0):
return " " * num
def print_prefix():
pass
def print_dir_tree(dir, link='', deep=0, prefix_print=''):
print(prefix_print, end="")
print(link, end="")
print(dir.name)
if dir.next:
prefix_print += link7
else:
prefix_print += link6
for i in range(len(dir.children) - 1):
print_dir_tree(dir.children[i], link2, deep + 1, prefix_print)
if len(dir.children) >= 1:
print_dir_tree(dir.children[-1], link4, deep + 1, prefix_print)
def main(argv):
base_path = argv[1] if len(argv) >= 2 else os.path.abspath('.')
root = dir_tree(base_path)
print_dir_tree(root)
if __name__ == '__main__':
start = time.time()
main(sys.argv)
end = time.time()
print("print dir tree cost {:.2f}s".format(end - start))
Python 打印目錄樹
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
- 文/潘曉璐 我一進(jìn)店門往弓，熙熙樓的掌柜王于貴愁眉苦臉地迎上來，“玉大人蓄氧，你說我怎么就攤上這事函似。” “怎么了喉童？”我有些...
- 文/不壞的土叔 我叫張陵撇寞，是天一觀的道長。 經(jīng)常有香客問我，道長蔑担，這世上最難降的妖魔是什么牌废？ 我笑而不...
- 正文 為了忘掉前任，我火速辦了婚禮啤握，結(jié)果婚禮上鸟缕，老公的妹妹穿的比我還像新娘。我一直安慰自己排抬，他們只是感情好懂从，可當(dāng)我...
- 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著蹲蒲，像睡著了一般番甩。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上届搁，一...
- 文/蒼蘭香墨 我猛地睜開眼，長吁一口氣：“原來是場噩夢啊……” “哼浩嫌！你這毒婦竟也來了檐迟？” 一聲冷哼從身側(cè)響起，我...
- 序言：老撾萬榮一對情侶失蹤码耐，失蹤者是張志新（化名）和其女友劉穎追迟，沒想到半個月后，有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體骚腥，經(jīng)...
- 正文 獨(dú)居荒郊野嶺守林人離奇死亡敦间，尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
- 正文 我和宋清朗相戀三年，在試婚紗的時候發(fā)現(xiàn)自己被綠了束铭。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片廓块。...
- 正文 年R本政府宣布拴清，位于F島的核電站靶病，受9級特大地震影響，放射性物質(zhì)發(fā)生泄漏口予。R本人自食惡果不足惜娄周，卻給世界環(huán)境...
- 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望沪停。 院中可真熱鬧煤辨，春花似錦、人聲如沸牙甫。這莊子的主人今日做“春日...
- 文/蒼蘭香墨 我抬頭看了看天上的太陽窟哺。三九已至泻轰，卻和暖如春，著一層夾襖步出監(jiān)牢的瞬間且轨，已是汗流浹背浮声。 一陣腳步聲響...
- 正文 我出身青樓，卻偏偏與公主長得像至朗，于是被迫代替她去往敵國和親屉符。 傳聞我的和親對象是個殘疾皇子，可洞房花燭夜當(dāng)晚...
推薦閱讀更多精彩內(nèi)容
- 假設(shè)有這樣一個任務(wù)锹引，希望對某個文件夾（包括所有子文件夾與文件）中的所有文件進(jìn)行處理矗钟。這就需要遍歷整理目錄樹， 處理...
- 根據(jù) 《Think in java》 io 章的前兩節(jié)嫌变，簡單寫出來的代碼吨艇， 還有不完善的地方，需要修改腾啥。 得到結(jié)果如下：
- 最近看到vscode tree生成目錄樹結(jié)構(gòu)插件挺好玩的东涡，自己琢磨實(shí)現(xiàn)了一個簡單版本，支持文件倘待，文件夾過濾 最終效...
