写爬虫经常要生成小文件来存储爬到的测试信息,就模拟实现* 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()