python一些函数记录

Posted 2021-03-06 MTcx

这次呢，主要是记录一下再实验用的一些模块或函数。

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 +="
 {}{}{}".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("
 {}{}{}".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 += "
{};{};{}".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 += "
{};{};{}".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 += ‘
{};{};{}‘.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
    ‘‘‘

---------------------------------------------分割线休息休息----------------------------------------------------