压缩某个文件夹中的某种类型的文件

  1 #!/usr/bin/env python

  2 # -*- coding:utf-8 -*-

  3

  4 import os

  5 import gzip

  6

  7

  8 __doc__ ='''

  9         zip files.

 10         usage:

 11                 zipfile /home/tmp/ .pdf

 12 '''

 13

 14 def zipfiles(path, ftype):

 15

 16         try:

 17                 all_fd = os.listdir(path)

 18         except OSError:

 19                 print 'No such file or directory: %s' % (path, )

 20

 21                 return

 22

 23         for fd in all_fd:

 24                 fd_full = os.path.join(path, fd)

 25

 26                 if os.path.isfile(fd_full) and fd_full.endswith(ftype):

 27

 28                         try:

 29                                 fsrc = open(fd_full, 'rb')

 30                                 fzip = gzip.open(fd_full+'.gz', 'wb')

 31

 32                                 fzip.writelines(fsrc)

 33

 34                                 fsrc.close()

 35                                 fzip.close()

 36                         except IOError:

 37                                 print 'zip failed: %s' % (fd, )

 38

 39         return

 40

 41 #-------------------------------------------------------------------------------

 42

 43 import sys

 44

 45 if __name__ == '__main__':

 46         print sys.argv

 47

 48         if len(sys.argv) != 3:

 49                 print __doc__

 50

 51         else:

 52                 zipfiles(sys.argv[1], sys.argv[2])

#===================================================

  比较粗糙，可以用，但没有递归文件夹，自己再试试看。