精悍的Python代码段

精悍的Python代码段 - codingcool - 博客园

精悍的Python代码段

1 文件处理

readlines() 和 writelines()方法可以讲列表元素依次写到文件中；

file类本身没有提供复制方法，可以使用read()和write()方法模拟实现文件的拷贝，也可以使用shutil模块：

shutil.copyfile('hello.txt','hello2.txt')
shutil.move('hello.txt','../')
shutil.copyfile('hello2.txt','hello3.txt')

批量改变当前目录中文件的名字:

方式一：

files = os.listdir(".")
for filename in files:
    pos = filename.find(".");
    if filename[pos+1:] == 'html':
        newname = filename[:pos+1] + 'htm'
        os.rename(filename,newname)

方式二：

files = os.listdir(".")
for filename in files:
    li = os.path.splitext(filename)
    if li[1] == 'html':
        newname = li[0] + '.htm'
        os.rename(filename,newname)

方式三：

可以使用glob模块中的glob方法获取指定条件的文件列表：

例如使用l = glob.glob('*.html')获取html为后缀的文件

l = glob.glob('c:\\w*\\*.txt')获取C盘中w开头目录中所有的文本文件。

统计文件中指定字符串的数量（是统计字符串而不是单词，字符串可以不是整个单词）：

import re
def str_count(filename,s):
    count = 0
    f = file(filename,'r+')
    for line in f.readlines():
        li = re.findall(s,line);
        if li.count(s) > 0:
            count += li.count(s);
    f.close()
    return count

统计文件中指定单词的数量：

def word_count(filename,word):
    count = 0
    f = file(filename,'r+')
    for line in f.readlines():
        p = r'\b'+word+r'\b'
        print p
        li = re.findall(p,line);
        print li
        if li.count(word) > 0:
            count += li.count(word);
    f.close()
    return count

修改文件中指定字符串的代码，这里面有两种修改方式，即是否进行全词匹配：

#f1源文件，f2修改后文件，s待修改的字符串，_s为改后字符串，whole是否全词匹配，默认值0代表全词匹配
def replace_str (file1,file2,s,_s,whole = 0):
    f1 = file(file1,'r+')
    f2 = file(file2,'w+')
    p = r'\b' + s + r'\b'
    if whole == 1:
        p = s
    for line in f1.readlines():   
        t = re.sub(p,_s,line)
        f2.write(t)
    f1.close()
    f2.close()

difflib是python提供的比较序列(string list)差异的模块。实现了三个类：

1>SequenceMatcher 任意类型序列的比较 (可以比较字符串)
2>Differ 对字符串进行比较
3>HtmlDiff 将比较结果输出为html格式.

 

使用difflib进行序列的比较与结果输出：

import difflib
from pprint import pprint
a = 'pythonclub.org is wonderful'
b = 'Pythonclub.org also wonderful'
s = difflib.SequenceMatcher(None, a, b)
print "s.get_matching_blocks():"
pprint(s.get_matching_blocks())
print
print "s.get_opcodes():"
for tag, i1, i2, j1, j2 in s.get_opcodes():
    print ("%7s a[%d:%d] (%s) b[%d:%d] (%s)" %  (tag, i1, i2, a[i1:i2], j1, j2, b[j1:j2]))

来看另一个例子：

def reportSingleFile(srcfile, basefile, rpt):
    src = file(srcfile).read().split(' ')
    base = file(basefile).read().split(' ')
    import difflib
    s = difflib.SequenceMatcher( lambda x: len(x.strip()) == 0, # ignore blank lines
        base, src) 
    lstres = []
    for tag, i1, i2, j1, j2 in s.get_opcodes():
        print (tag, i1, i2, j1, j2)
        #print lstres
        if tag == 'equal':
            pass
        elif  tag == 'delete' :
            lstres.append('DELETE (line: %d)' % i1)
            lstres += base[i1:i2]
            lstres.append(' ')
        elif tag == 'insert' :   
            lstres.append('INSERT (line: %d)' % j1)
            lstres += src[j1:j2]
            lstres.append(' ')
        elif tag == 'replace' :   
            lstres.append('REPLACE:')
            lstres.append('Before (line: %d) ' % i1)
            lstres += base[i1:i2]
            lstres.append('After (line: %d) ' % j1)
            lstres += src[j1:j2]
            lstres.append(' ')
        else:
            pass
    print ' '.join(lstres)

使用Differ进行字符串比较

import difflib
diff = difflib.Differ().compare("start up","starT u4p")    
print "\n".join(list(diff))

使用HtmlDiff进行统计：

from difflib import *
s = HtmlDiff.make_file(HtmlDiff(),"start up","storT up")
f=open(r"C:/dong/result.html","w")
f.write(s)
f.close()

越来越发现python非常适合做一些日常开发的工具。平时，我们经常用一些比较工具，比较目录、比较两个文本文件的变化。最近发现，python的标准库里居然带了这些功能的算法。自己处理一下，就可以写出一个很实用的比较工具了。文件和目录比较Module叫做filecmp。最酷的是他提供了一个叫dircmp的类，可以直接比较两个目录。

目录遍历的三种方式：

1 使用递归函数：
def visitDir (path):
    li = os.listdir(path)
    for p in li:
        pathname = os.path.join(path,p)
        if not os.path.isfile(pathname):
            visitDir(pathname)
        else:
            print pathname

2 使用os.path.walk()

import os,os.path

def visitDir (arg,dirname,names):
    for filepath in names:
        print os.path.join(dirname,filepath)

path1 = 'C:\\dong'     
os.path.walk(path1,visitDir,())

3 使用os.walk()

def visitDir (path):
    for root,dirs,files in os.walk(path):
        for filepath in files:
            print os.path.join(root,filepath)

path1 = 'C:\\dong'     
visitDir(path1)

自己制作的简单日志记录文件：

#coding=gbk
import sys,time

sys.stderr = open("record.log","a")
f = open(r"hello.txt",'r')
t = time.strftime("%Y-%m_%d%X",time.localtime())
context = f.read()
if context:  #不为空
    sys.stderr.write(t + " " + context)
else:
    raise Exception,t+"异常信息"

使用Python模拟Java中的输入和输出流：

#coding=gbk
def FileInputStream (filename):
    try:
        f = open(filename)
        for line in f:
            for byte in line:
                yield byte
    except StopIteration,e:
        f.close()
        return
   
def FileOutputStream (inputStream,filename):
    try:
        f = open(filename,'w')
        while True:
            byte = inputStream.next()
            f.write(byte)
    except StopIteration,e:
        f.close()
        return