上述实验题若改为由运行参数给出指定目录,执行格式为: 程序名 [目录名],若执行时位置参数没有给出目录名,则默认为当前目录。 脚本要怎么改?
时间: 2024-03-23 09:43:31 浏览: 23
Python获取运行目录与当前脚本目录的方法
可以使用Python中的sys.argv来获取运行参数。修改脚本如下:
```python
import os
import sys
# 获取目录名参数,如果没有则默认为当前目录
if len(sys.argv) > 1:
path = sys.argv[1]
else:
path = '.'
# 遍历目录下的所有文件和子目录
for root, dirs, files in os.walk(path):
for file in files:
# 输出文件路径
print(os.path.join(root, file))
```
在命令行中执行时,输入以下命令即可:
```
python script.py [目录名]
```
其中,`script.py`为脚本文件名,`[目录名]`为可选的目录名参数。如果不提供目录名参数,则默认为当前目录。
阅读全文