学Python脚本学了有一小阵了,可以写一些简单的脚本。前段时间帮同学写了一个方便他工作中使用。
需求如下:有一个文件mini.list,里面包含一些相对路径和空行。有一个文件夹里有很多文件,文件是绝对路径。想要在很多个文件中找mini.list里面的路径,最终输出要寻找的绝对路径+文件名。
脚本如下
# !python3
import os,sys
reference_fiename='mini.list' #要查找的文件名称
path=sys.path[0] #获取本文件所在目录,默认要查找的a文件和本文件在同一目录下
reference_file=open(os.path.join(path,reference_fiename)) #打开reference file
search_path=[] #定义需要查找的路径:如果在reference file中的行不为空,则定义为需要查找的路径
for line in reference_file.readlines():
if line != '\n' and line != '':
search_path.append(line)
workpath=os.path.join(path,'Data') #默认需要在其中查找的文件在本文件目录下的子文件夹‘Data’中
newfile=open(os.path.join(path,'newdata.txt'), 'a') #在本文件所在目录中新建名为'newdata.txt'的文件
for filename in os.listdir(workpath): #历遍需要查找的文件中的每一个文件
workfile=open(os.path.join(workpath, filename)) #打开需要查找的文件
workfile_lines=workfile.readlines() #读取需要查找的文件的每一行
for i in range(len(search_path)): #需要查找的文件的每一行和需要查找的路径相比对
for j in range(len(workfile_lines)):
if search_path[i] in workfile_lines[j]:
newfile.write(filename.split('.')[0]+' '+workfile_lines[j].split('\n')[0]+'\n')
print('Done')
newfile.close()
在自己电脑上测试是没有问题的,但是发给同学之后他在Lunix系统上运行是不行的。最后要了一份demo文件debug,发现在Lunix系统中的换行符是‘\r\n’。
去搜索了一下:
在ASCII码表中,\r 回车符 \n换行符
windows中dos窗口中
\r 回到本行最左边,并且会替换后面的文本
\n 换行
\r\n 和\n\r换行
按照标准来,用\r\n。
作者:宋承亮
链接:https://www.jianshu.com/p/4d1853562e67
来源:简书
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
最后把这段
for line in reference_file.readlines():
if line != '\n' and line != '':
search_path.append(line)
改成了
for line in reference_file.readlines():
if line.strip() != '':
search_path.append(line)
用strip()感觉简单很多,也不容易出错。