python一些函数记录
这次呢,主要是记录一下再实验用的一些模块或函数。
os模块
os.listdir()函数可以显示指定目录下的所有信息。 eg: os.listdir(“./") 当然还有使用walk() os(.stat;path) os.path() 对文件路径进行处理
os-file-dir 文件目录信息
#coding=utf-8 import os for root,dirs,files in os.walk('./'); print root,dirs,files
将得到的文件、文件名存放到指定文件内
#coding=utf-8 import os for root,dirs,files in os.walk('./'): open('test.txt','a').write("{}{}{}".format(root,dirs,files))
#下面另外两种写法 #coding=utf-8 import os for root,dirs,files in os.walk('./'): export +="\n {}{}{}".format(root,dirs,files) open('demo01.py','w').write(export) #coding=utf-8 import os export = [] for root,dirs,files in os.walk('./'): export.append("\n {}{}{}".format(root,dirs,files)) open('demo01.py','w').write(''.join(export))
文件搜索预期效果
python pysoso.py -e mysoso.txt #将目录内容记录为 mysoso.txt python pysoso.py -e sos/mysoso.txt #将目录内容保存到sos目录下的mysoso.txt python pysoso.py -d sos -k 中国加油 #搜索sos目录下的信息,寻找有中国加油字样的文件或目录。
想要达到这种效果就需要将文件内容写为函数式。
#coding=utf-8 import os def cdWalker(cdm,cdf): export = "" for root,dirs,file in os.walk(cdm): export += "\n{};{};{}".format(root,dirs,files) open(cdf,'w').write(export) cdWalker('./','cd1.txt') cdWalker('./','cd2.txt')
加入提示信息
#coding=utf-8 import os,sys print sys.argv def cdWalker(cdm,cdf): export = "" for root,dirs,files in os.walk(cdm): export += "\n{};{};{}".format(root,dirs,files) open(cdf,'w').write(export) #cdWalker('./','mysoso.txt')
加入if……else判断,
#coding=utf-8 import os,sys CDM = './' def cdWalker(cdm,cdf): export = '' for root,dirs,files in os.walk(cdm): export += '\n{};{};{}'.format(root,dirs,files) open(cdf,'w').write(export) #cdWalker('./','mysos.txt') if "-e" ==sys.argv[1]: cdWalker(CDM,sys.argv[2]) print "记录信息到{}".format(sys.argv[2]) else: print '''PyC使用方式: python pyc.py -e mysos.txt #将文件内容记录为mysos.txt '''
---------------------------------------------分割线休息休息----------------------------------------------------