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