先直接贴代码
import sys
import os
def print_file_name(file_dir, file_format = "None"):
if (file_format == "None"):
for root, dirs, files in os.walk(file_dir):
for ele in files:
print(ele)
break
else:
for root, dirs, files in os.walk(file_dir):
for ele in files:
temp_format = os.path.splitext(ele)[1]
if (temp_format.lstrip(".") == file_format.strip().lstrip(".")):
print(ele)
break
params_ = sys.argv
lens_ = len(params_)
if lens_ > 3:
print("Error: too many params.")
else:
if (lens_ == 1):
abs_path = os.path.split(os.path.abspath(__file__))
print_file_name(abs_path[0])
sys.exit()
elif(lens_ == 2):
cur_format = params_[1]
abs_path = os.path.split(os.path.abspath(__file__))
print_file_name(abs_path[0], cur_format)
sys.exit()
else:
cur_path = params_[2]
cur_format = params_[1]
print_file_name(cur_path, cur_format)
调用方式
- 先将上述代码保存成.py文件，比如保存成print_file_name.py
- 在命令行调用
-
打印当前目录所有文件列表
-
打印当前目录的所有指定格式的文件列表
-
打印指定目录的指定格式的文件列表
