python自动操作脚本_用Python操纵文件（自动化脚本的第一步）

用Python操纵文件（自动化脚本的第一步）

@(Python相关)

os模块相关

路径的跨平台（解决windows系统和Unix系统正反斜杠的问题）

import os

spam_path = os.path.join('usr','bin','spam')

控制工作目录

获取当前工作目录

cwd_path = os.getcwd()

更换当前工作目录

os.chdir(new_cwd_path)

创建新文件夹

os.makedirs(os.path.join(os.getcwd(),newdir,subdir))

此函数会沿着指定的路径连续创建不存在的文件夹，以确保路径的完整性

绝对路径与相对路径

检查给定路径是否是绝对路径

os.path.isabs(path)

将相对路径转化为绝对路径

abs_path = os.path.abspath(relative_path)

提取两个路径之间的相对路径（从start到end）

rel_path = os.path.relpath(end,start)

分离提取文件的绝对路径里的文件夹路径与文件名

dir_path = os.path.dirname(path)

file_name = os.path.basename(path)

dir_path, file_name = os.path.split(path)

若要逐级分离（当作字符串处理）

path_string.split(os.path.sep)

"""

此种用法在windows上第一个是磁盘的名称

而在Mac和Linux上第一个是空字符串，因为字符串的第一个字符就是分隔符~

"""

os.getcwd() -> '/mnt/lvm/dingning/datasets/coco'

os.getcwd().split(os.path.sep) -> ['', 'mnt', 'lvm', 'dingning', 'datasets', 'coco']

查看文件大小和文件夹内容

查看指定路径下的文件的大小（返回对应文件的字节数）

os.path.getsize(file_path)

查看指定文件夹路径对应的文件夹中的文件名称（返回文件名列表）

os.listdir(dir_path)

检查文件有效性

检查path对应的文件或文件夹是否存在（返回布尔变量）

os.path.exists(path)

检查file_path对应的文件路径是否存在，并且是否是一个文件（返回布尔变量）

os.path.isfile(file_path)

检查dir_path对应的文件夹路径是否存在，并且是一个文件夹（返回布尔变量）

os.path.isdir(dir_path)

遍历目录树 os.walk()

传入一个文件夹路径，在每次迭代时返回三个值：

当前文件夹名称的字符串

当前文件夹中子文件夹的字符串列表

当前文件夹中问价的字符串列表

import os

for folderName, subfolders, filenames in os.walk(fold_path):

print 'the current folder is ' + folderName

for subfolder in subfolders:

print 'something'

for filename in filenames:

print 'something'

File对象相关

文件读写

三个步骤：

针对指定文件路径字符串调用open()返回一个File对象

调用File对象的read() write()方法

调用File对象的close()方法，关闭文件

一般使用with ... as ... :方式使用文件

读取文件内容：

read() 将整个文件读取为一个字符串

readlines() 将文本的每一行读取为一个字符串，返回整个文件所有行字符串对应的列表

写入文件（只有一个write()方法）：

以'w'方式打开文件，若该文件不存在，则创建，若存在，则覆盖

以'a'方式打开文件，若该文件不存在，则创建，若存在，则在文件内容的最后开始追加写入

write()不会自动加入\n，所以写入文件的时候如果需要换行，则需要自己添加换行分隔符

用shelve模块保存Python变量（将Python变量保存到二进制shelf文件中）

import shelve

shelfFile = shelve.open('mydata')

cats = ['a','b','c']

shelfFile['cats'] = cats

shelfFile.close()

shelve文件就像一个列表，对每一个变量都可以取一个名字作为key，然后其内容则为value

list(shelfFile.keys()) -> ['cats']

list(shelfFile.values()) -> [['a','b','c']] """注意此处是一个列表，里面第一个元素正好是一个列表对象"""

pprint 模块--漂亮打印

pprint.pprint()将列表和字典漂亮打印到屏幕

pprint.pformat()将上述要打印到屏幕的文本以字符串的方式返回，利用此可以简单用程序编写python脚本

import pprint

cats = ['a','b','c']

fileObj = open('myCats.py','w')

fileObj.write('cats = ' + pprint.pformat(cats) + '\n')

fileObj.close()

在另一个python脚本中可以这样使用（用Python程序编写更加庞大的Python程序的基础）

from myCats import cats

print cats -> ['a','b','c']

shutil模块相关（shell utilize）（复制，移动，改名，删除）

复制文件和文件夹

shutil.copy(source, destination) 将source处文件复制到destination处

destination如果是文件夹，则新文件名称与原始文件名称保持一致

destination如果是文件名，则以此文件名作为复制后的文件名复制

shutil.copytree(source, destination)将source处文件夹复制到destination处

此时，source和destination都必须是文件夹的路径

文件夹的移动和改名 shutil.move(source, destination) 使用需谨慎~

若destination指定的是文件夹，则必须判定该文件夹是否存在，否则会将路径当作是文件名重命名

若其存在，则将文件移动到该文件夹下

若destination指定的是文件名，若沿途的文件夹均存在，则移动到对应路径下，否则会抛出异常

删除文件 使用需谨慎

os.unlink(file_path) 必须存在且为文件路径，而不是文件夹路径

os.rmdir(dir_path) 必须存在且为文件夹路径，而且必须为空文件夹

shutil.rmtree(dir_path) 必须存在且为文件夹路径，不必为空文件夹

使用时最好在此代码前面加入print将即将被删除的文件打印出来，提供给用户最终决定权

zipfile模块相关 处理zip压缩文件

基本框架

import os

import zipfile

zipObj = zipfile.ZipFile(file_path) -> '生成一个ZipFile对象'

zipObj.namelist() -> '打印zip文件目录树，返回一个list' -> '子方法，返回zip文件的目录list'

spamInfo = zipObj.getinfo('spam.txt') -> 将list中的一个文件名传入子方法getinfo中，可以返回一个ZipInfo对象

ZinInfo对象则保存zip文件每个子文件的有用的信息，如：

spamInfo.file_size

spamInfo.compress_size

解压缩 （ZipFile对象的子方法）

extract()：第一个参数为zip文件中的子文件名，如namelist()中的一个文件名，第二个参数为目标文件夹路径

extractall()：无参数则解压全部文件到本地，也可以传递一个文件夹路径，在指定文件夹下解压缩

如果指定文件夹不存在，则创建

创建和添加ZIP文件

import zipfile

newZip = zipfile.ZipFile('new.zip','w') '此处可以用追加方式，只需将w改为a'

newZip.write(file_path, compress_type=zipfile.ZIP_DEFLATED) '指定压缩算法'

newZip.close()