python简单模拟tree功能

写爬虫经常要生成小文件来存储爬到的测试信息，就模拟实现* tree 功能来熟悉 os模块和 sys模块*
的应用代码如下：

#!/usr/bin/env python3.6
# -*- coding: utf-8 -*-
__author__ = 'Timothy'
import os
import sys

def get_filename(dirname, d=0):
    print('  ' * d + r'|__' + dirname + '\n')
    for name in os.listdir(dirname):
        if not os.path.isdir(name):
            print('  ' * (d + 1) + r'|__' + name + '\n')
        else:
            get_filename(dirname + '/' + name, d + 1)

def cmd_run():
    if len(sys.argv) != 2:
        print('参数个数错误')
        exit()
    if not os.path.exists(sys.argv[1]):
        print('输入路径错误')
        exit()
    print(sys.argv[0], 'is running now\n')
    get_filename(sys.argv[1])

if __name__ == '__main__':
    cmd_run()