Welcome, guest | Sign In | My Account | Store | Cart

Poor man's Winzip (Python recipe) by Dirk Krause
ActiveState Code (http://code.activestate.com/recipes/104733/)

Maybe not too sophisticated, but a handy little tool I use for myself.

This program does the three essential tasks that (at least) I need: 1. Unzip an archive to a folder 2. Zip a folder to an archive 3. Zip a (big) file to an archive

Freeze it, drop it into the 'sendto' folder, and you have a very convenient way to use this program via right mouseclick/sendto->zip2folder.

Python, 169 lines
Download
  1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
"""
zip2folder.py by Dirk Krause, d_krause@pixelpark.com, 12/27/2001

Abstract: A poor man's WinZip

This program does the three essential tasks that (at least) I need:
1. Unzip an archive to a folder
2. Zip a folder to an archive
3. Zip a (big) file to an archive

For convenient usage, get Gordon McMillan's Installer Package, and freeze this application with
> python freeze.py zip2folder.py
After that, you will have a self containing EXE that works even without having python installed.
Drop this thing into the 'sendto' folder, and you have a very convenient way to use this program
via right mouseclick/sendto zip2folder.
"""



import zipfile
import sys, os, shutil, string

argument = ''
if len(sys.argv) > 1:
    argument = sys.argv[1]
else:
    print 'No argument!'

# argument = 'C:/installer4/test.zip'


MODE_IS_UNKNOWN   = 0
MODE_IS_FILE      = 1
MODE_IS_ZIPFILE   = 2
MODE_IS_DIRECTORY = 3




class RecursiveFileIterator:
    # Great script from Daniel Dittmar,
    # http://www.faqts.com/knowledge-base/view.phtml/aid/6000
    def __init__ (self, *rootDirs):
        self.dirQueue = list (rootDirs)
        self.includeDirs = None
        self.fileQueue = []

    def __getitem__ (self, index):
        while len (self.fileQueue) == 0:
            self.nextDir ()
        result = self.fileQueue [0]
        del self.fileQueue [0]
        return result

    def nextDir (self):
        dir = self.dirQueue [0]   # fails with IndexError, which is fine
                                  # for iterator interface
        del self.dirQueue [0]
        list = os.listdir (dir)
        join = os.path.join
        isdir = os.path.isdir
        for basename in list:
            fullPath = join (dir, basename)
            if isdir (fullPath):
                self.dirQueue.append (fullPath)
                if self.includeDirs:
                    self.fileQueue.append (fullPath)
            else:
                self.fileQueue.append (fullPath)


doit = 0
do_mode = MODE_IS_UNKNOWN


if os.path.isfile(argument):
    if string.lower(os.path.splitext(argument)[1]) == '.zip':
        do_mode = MODE_IS_ZIPFILE
    else:
        do_mode = MODE_IS_FILE
else:
    do_mode = MODE_IS_UNKNOWN


if os.path.isdir(argument):
    do_mode = MODE_IS_DIRECTORY



if do_mode == MODE_IS_FILE:
    print 'converting file %s to zip?' % argument

if do_mode == MODE_IS_ZIPFILE:
    print 'converting zipfile %s to folder?' % argument

if do_mode == MODE_IS_DIRECTORY:
    print 'converting directory %s to zipfile?' % argument

if do_mode in (MODE_IS_FILE, MODE_IS_ZIPFILE, MODE_IS_DIRECTORY):
    # comment the following line to not have to confirm
    if raw_input('[y/n]')[0] in 'yY':
        doit = -1


if do_mode == MODE_IS_UNKNOWN:
    print '%s is not a file.' % argument


if doit == -1:
    print 'ok, doing it.'
else:
    print 'exiting.'
    

if doit == -1 and do_mode == MODE_IS_ZIPFILE:
    zipfilename = argument

    zipdir = os.path.splitext(zipfilename)[0] + '/'

    # if you want the folder to be cleared before unzipping,
    # uncomment the following lines:

    #try:
    #    shutil.rmtree(zipdir)
    #    print 'Deleted',zipdir
    #except: pass


    zipdir = os.path.splitext(zipfilename)[0] + '/'

    file = zipfile.ZipFile(zipfilename, "r")
    print 'Number of files', len(file.infolist())
    for zfile in file.infolist():
        #print '--------------File:',
        #print zfile.filename, zfile.date_time, zfile.file_size, zfile.compress_size
        dirname = zipdir + os.path.dirname(zfile.filename)

        try: os.makedirs( dirname )
        except: pass
        
        if zfile.file_size > 0:
            temp = file.read(zfile.filename)
            # print temp
            fname =  zipdir + zfile.filename
            print fname
            f = open(fname, "wb").write(temp)


if doit == -1 and do_mode == MODE_IS_DIRECTORY:
    directory = argument

    print 'directory', os.path.abspath(directory)
    
    file = zipfile.ZipFile(os.path.abspath(directory)+".zip", "w")
    
    for name in RecursiveFileIterator(directory):
        print 'try', name
        if os.path.isfile(name):
            file.write(name, name, zipfile.ZIP_DEFLATED)
    
    file.close()

if doit == -1 and do_mode == MODE_IS_FILE:
    file = zipfile.ZipFile(os.path.abspath(argument)+".zip", "w")
    file.write(argument, argument, zipfile.ZIP_DEFLATED)
    file.close()

# comment the following line to have the dos box disappear immediately
inp = raw_input('<... press the anykey key.>')