os模块os模块负责程序与操作系统的交互,提供了访问操作系统底层的接口;即os模块提供了非常丰富的方法用来处理文件和目录。
使用的时候需要导入该模块:import os
常用方法如下:方法名作用
遍历所有文件及文件夹
获得当前python程序运行路径
import os
print(os.getcwd()) # 输出当前代码所在模块的路径#
/Users/running/PycharmProjects/chapter04/novel
切换目录到chapter03项目下:
os.chdir('/Users/running/PycharmProjects/chapter03')
查看当前目录下的所有文件和文件夹,并判断是否是文件和文件夹
import os
files = os.listdir()
for file in files:
print(file, os.path.isdir(file),
os.path.isfile(file))
输出结果:
或者使用:
forfileinos.scandir():
print(file.name, file.path,
file.is_dir())
可以调用file.stat()获取更加详细的文件信息。如:
其中:st_size:文件的体积大小(单位:bytes),除以1024就是KBst_atime:文件的最近访问时间st_mtime:文件的最近修改时间st_ctime:Windows下表示创建时间
案例1:
1.键盘输入一个路径
2.统计该路径下的文件和文件夹,以及分别的数量
3.统计当前路径下包含文件名称中包含demo的文件数量,注意不区分大小写
import ospath = input('输入要查询的路径:')
os.chdir(path)
file_list = []
dir_list = []
for file inos.scandir():
if file.is_dir():
dir_list.append(file.name)
else:
file_list.append(file.name)
print("文件夹的总量是{},\n文件为别为{}".format(len(dir_list),dir_list))
print('{}'.format('-'*30))
print("文件总量是{},\n文件为别为{}".format(len(file_list),file_list))
print('{}'.format('-'*30))
python_list = []
for name in file_list:
if'demo'in name.lower():
python_list.append(name)
print('含有python单词的文件数量有{}个,\n文件分别为{}'.format(len(python_list),python_list))
如果要遍历一个目录下的所有子目录,里面指定的文件,我们可以通过递归函数遍历每个目录进行查找或者使用os.walk(),递归实现代码如下:
import os
import sys
from stat import *
defwalktree(top, callback):for f in os.listdir(top):
pathname = os.path.join(top, f)
try:
mode = os.stat(pathname, follow_symlinks=False).st_mode
except:
continueif S_ISDIR(mode):
# directory, recurse into it
walktree(pathname, callback)
else:
# file, whatever type, make the call back function
callback(pathname)
returndefprintfile(file):
print('get to', file)
if __name__ == '__main__':
if (os.path.isabs(sys.argv[1]) and
os.path.exists(sys.argv[1])):
walktree(sys.argv[1], printfile)
Python os模块的walk()函数,顾名思义,就是用来遍历目录树的,此函数可以很方便的遍历以输入的路径为root的所有子目录和其中的文件。walk函数是一个Python生成器(generator),调用方式是在一个for...in...循环中,walk生成器每次返回的是一个含有3个元素的tuple,分别是 (dirpath, dirnames, filenames)
fordirpath, dirnames, filesinos.walk('./'):
print(dirpath,dirnames,files)
当然也可以使用glob模块快速实现,假设要获取主目录中所有的mp4文件:
此时使用glob更加方便
import glob
t_list = glob.glob('t*')
print(t_list) # 获取当前目录下t开头的文件,但是不能获取子文件夹里面的内容
Python的glob还支持规则(从3.5开始),当规则与另一个参数recursive=True配合的时候,可以深入的路径的子目录当中去匹配:
glob.glob('**') # 获取当前目录的文件和文件夹
glob.glob('**/') # 获取当前目录的文件夹# 如果**结合recursive=True使用,但是如果结合一个*则不能实现递归
glob.glob('**/',recursive=True) # 可以遍历出所有的子目录# 若要获取子目录中的文件则需要
glob.glob('**/*.mp4',recursive=True)
案例2:
1.键盘输入一个路径
2.搜索该路径下文件大小超过50M的zip文件
3.搜索该路径下最后修改日期在30天前的文件
4.打印显示2,3的文件import os
import datetime
path = input('输入要查询的路径:')
os.chdir(path)
paths = glob.glob('**/*.zip',recursive = True)
forpathin paths:
file_size = os.stat(path).st_size/1024/1024
file_modify = datetime.datetime.fromtimestamp(os.stat(path).st_mtime)
days = (datetime.datetime.now() - file_modify).days
if (file_size > 50) and (days > 30):
print('压缩包的路径名称是:{},大小为{:.2f}MB,创建文件年份为:{}'.format(path,file_size,file_build_year))
批量操作文件及文件夹
创建文件夹和多层文件夹
import osifnotos.path.exists('新文件夹名'):
os.mkdir('新文件夹名')
# 创建多层文件夹
os.makedirs('第一层文件夹/第二层文件夹/第三层文件夹')
复制文件或者文件夹
复制、移动、删除文件夹需要借助另一个模块:shutil模块
# 首先导入模块import shutil
import os
shutil.copy('文件名', '目标文件夹') # 将文件复制到指定文件夹
shutil.copy('文件名', '目标文件夹/新文件名.txt') # 复制并重命名文件#如果复制的是文件夹
shutil.copytree('源文件夹','目标文件夹')
# 如果不想复制文件而是移动文件,可以使用过move方法 使用方法类似copy,也可以移动文件夹
shutil.move('文件名', '目标文件夹/')
shutil.move('源文件夹','目标文件夹/') # 文件夹后面记住最后加斜杠/# 若要删除文件夹
shutil.rmtree('文件夹名')
# 删除文件
os.remove('文件名')
案例3:
1.键盘输入一个路径
2.获取里面所有的mp4文件
3.重命名mp4文件在每个文件前面添加前缀,前缀就是文件最后修改的年月日(如:2021-08-18_西游记01.mp4)
4.新建文件夹:最新视频
5.将重命名的视频批量移动到最新视频文件夹
import os
import datetime
import shutil
import glob
path = input('输入要查询的路径:')
os.chdir(path)
ifnotos.path.exists('最新视频'):
os.mkdir('最新视频')
for dirpath,dirnames,files inos.walk('./'):
for file inos.scandir(dirpath):
if file.name.endswith(".mp4"):
tm = datetime.datetime.fromtimestamp(file.stat().st_mtime)
new_file = str(tm.year)+'-'+str(tm.month)+'-'+str(tm.day)+'-'+file.name
os.rename(dirpath + '/' + file.name,new_file)
file_ls = glob.glob('*.mp4')
for name in file_ls:
shutil.move(name,'最新视频/')
print('over!!!')