web/ldt/utils/zipfileext.py
author ymh <ymh.work@gmail.com>
Sat, 12 Jun 2010 04:25:05 +0200
changeset 0 cc4a51750724
permissions -rw-r--r--
first commit
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
0
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
     1
import zipfile, os, os.path
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
     2
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
     3
class ZipFileExt(zipfile.ZipFile):
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
     4
    def unzip_into_dir(self, dir):
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
     5
        if not os.path.exists(dir):
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
     6
            os.mkdir(dir, 0777)
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
     7
        for name in self.namelist():
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
     8
            if name.endswith('/'):
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
     9
                os.mkdir(os.path.join(dir,name))
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    10
            else:
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    11
                outfile = open(os.path.join(dir,name), 'wb')
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    12
                outfile.write(self.read(name))
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    13
                outfile.close()
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    14