web/ldt/utils/zipfileext.py
author wakimd
Fri, 22 Oct 2010 18:14:30 +0200
changeset 1 eb9188f2ee4f
permissions -rw-r--r--
first commit
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
1
eb9188f2ee4f first commit
wakimd
parents:
diff changeset
     1
import zipfile, os, os.path
eb9188f2ee4f first commit
wakimd
parents:
diff changeset
     2
eb9188f2ee4f first commit
wakimd
parents:
diff changeset
     3
class ZipFileExt(zipfile.ZipFile):
eb9188f2ee4f first commit
wakimd
parents:
diff changeset
     4
    def unzip_into_dir(self, dir):
eb9188f2ee4f first commit
wakimd
parents:
diff changeset
     5
        if not os.path.exists(dir):
eb9188f2ee4f first commit
wakimd
parents:
diff changeset
     6
            os.mkdir(dir, 0777)
eb9188f2ee4f first commit
wakimd
parents:
diff changeset
     7
        for name in self.namelist():
eb9188f2ee4f first commit
wakimd
parents:
diff changeset
     8
            if name.endswith('/'):
eb9188f2ee4f first commit
wakimd
parents:
diff changeset
     9
                os.mkdir(os.path.join(dir,name))
eb9188f2ee4f first commit
wakimd
parents:
diff changeset
    10
            else:
eb9188f2ee4f first commit
wakimd
parents:
diff changeset
    11
                outfile = open(os.path.join(dir,name), 'wb')
eb9188f2ee4f first commit
wakimd
parents:
diff changeset
    12
                outfile.write(self.read(name))
eb9188f2ee4f first commit
wakimd
parents:
diff changeset
    13
                outfile.close()
eb9188f2ee4f first commit
wakimd
parents:
diff changeset
    14