mirror of
https://github.com/heqin-zhu/algorithm.git
synced 2024-03-22 13:30:46 +08:00
55 lines
1.7 KiB
Python
55 lines
1.7 KiB
Python
''' mbinary
|
|
#########################################################################
|
|
# File : tree.py
|
|
# Author: mbinary
|
|
# Mail: zhuheqin1@gmail.com
|
|
# Blog: https://mbinary.coding.me
|
|
# Github: https://github.com/mbinary
|
|
# Created Time: 2018-12-11 15:56
|
|
# Description:
|
|
#########################################################################
|
|
'''
|
|
# coding: utf-8
|
|
import os
|
|
from argparse import ArgumentParser
|
|
|
|
#命令行输入参数处理
|
|
parser = ArgumentParser()
|
|
|
|
parser.add_argument('-p','--path',default='.',help='path to walk')
|
|
parser.add_argument('-f','--fileinclude',action='store_true',help='if has, list files and dirs, else only dirs')
|
|
parser.add_argument('-d','--depth', type = int, default = 2)
|
|
#获取参数
|
|
args = parser.parse_args()
|
|
FILE = args.fileinclude
|
|
PATH = args.path
|
|
DEPTH = args.depth
|
|
|
|
def mklink(path):
|
|
return '* [{name}]({path})'.format(name=os.path.basename(path),path=path)
|
|
def clean(paths):
|
|
ret = []
|
|
for path in paths:
|
|
name = os.path.basename(path)
|
|
if not ( name.startswith('.') or name.startswith('__')):
|
|
ret.append(path)
|
|
return ret
|
|
|
|
def tree(path='.',depth=2,showfile=False):
|
|
li = []
|
|
if os.path.isdir(path):li = os.listdir(path)
|
|
else:li=[path]
|
|
items = [os.path.join(path,i) for i in li if not i.startswith('.')]
|
|
items = clean(items)
|
|
items = sorted(items)
|
|
if not showfile: items = [i for i in items if os.path.isdir(i)]
|
|
if depth==1:
|
|
return [mklink(path)] + [' '*4 + mklink(i) for i in items]
|
|
else:
|
|
uls = [tree(i,depth-1,showfile) for i in items]
|
|
ret = [' '*4 + li for ul in uls for li in ul]
|
|
return [mklink(path)] + ret
|
|
|
|
if __name__ =='__main__':
|
|
print('\n'.join(tree(PATH,DEPTH,FILE)))
|