自动化软件测试 -- 文件的操作

一、文件的读取

file = open("text.txt","r")      #打开只读文件
print(file.read(5))             #第一次只读取5个数据
print(file.read())               #第二次把剩下的全部读出来


#当文件足够大的时候，内存不够加载整个文件，可以选择先加载一部分
print(file.readline())
print(file.readlines())

file.close()

二、文件的写入

file = open("test.txt","w")

#写入数据
file.write("hello world")

file = open("test.txt","a")

#添加数据
file.write("hello world\n")


file.close()

三、文件的复制

#复制文件 text.txt 文件为 text -复制.txt 
#1、获取源文件的名字（包含路径）
filename = input("请输入要备份的文件名称：")
#2、打开源文件
oldFile = open(filename,"r")
#3、获取备份文件的名字（源文件的名字 + “-复制” + 后缀名）
fileFlagnum = filename.rfind(".")
fileFlag = filename[fileFlagnum:]
print(fileFlag)
newFilename = filename[:fileFlagnum] + "-复制" +fileFlag
print(newfilename)
#4、导入备份文件
newfile = open(newFilename,"w")

for linecontent in oldFile.readlines():
    newfile.write(linecontent)

#5、关闭文件
oldFile.close()
newFile.close()

四、文件夹的相关操作

os模块中的rename（）可以完成对文件的重命名操作

rename(需要修改的文件名，新的文件名)

import os
os.rename("test.txt","毕业论文.txt")     #修改文件名

os.remove("test.txt")   #删除文件

os.mkdir("aa")         #创建文件夹

os.getcwd()           #获取当前目录

os.chidir("../")            #修改默认目录

os.listdir("./")           #获取目录列表

os.rmdir("./aa")          #删除文件夹

文件名批量修改

import os

listName = os.listdir(dirpath)

#flag为1，添加标记，flag为2，删除标记
flag = 1

for name in listName:
    if flag == 1:
        newName = "[cherry]-" + name
    if flag == 2:
        num = len("[cherry]-")
        newName = name[num:]

    os.rename(dirpath+name,dirpath+newName)  #重命名