Login

Modified ZipFile() to support writedir() (with auto pyc/py selection)

Author:
sleepycal
Posted:
August 6, 2011
Language:
Python
Version:
Not specified
Tags:
zipfile auto writedir py pyc
Score:
0 (after 0 ratings)

See code

  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
class _ZipFile ( zipfile.ZipFile ):
    # Inspired by:
    # http://www.calazan.com/how-to-zip-an-entire-directory-with-python/comment-page-1/#comment-45304
    #
    # Heavily modified to support:
    #  - automatic choice between .pyc and .py
    #  - partially integrated with ZipFile class
    #
    # TODO:
    #  - Allow auto compile to /tmp when .pyc is not present.
    
    def writedir(self, pathname, basename = ""):
        """Zip the contents of an entire folder (with that folder included
        in the archive). Empty subfolders will be included in the archive
        as well.
        """
        
        _paths = {}
        
        def debug_log(msg):
            if self.debug:
                print msg
        
        def AddToZIP(absolute_path, relative_path):
            # check the ext of the file
            _path, _ext = os.path.splitext(absolute_path)
            
            # check if it is a file
            _is_file = os.path.isfile(absolute_path)
            
            # generate absolute pyc/py paths
            _pyc = "%s.pyc" % ( _path )
            _py = "%s.py" % ( _path )
            
            # generate relative pyc/py paths
            _pyc_rel = "%s.pyc" % ( _path )
            _py_rel = "%s.py" % ( _path )
            
            if _is_file:
                # if .py, then check if pyc is there
                if _ext == '.py':
                    # if its there then dont add
                    if _paths.has_key(_pyc):
                        debug_log("[py/skip] '%s'" % ( _py ) )
                    else:
                        _paths[_py] = _py_rel
                
                # if .pyc, then check if we have a py
                elif _ext == '.pyc':
                    # if we have a py, then replace it with the pyc
                    if _paths.has_key(_py):
                        debug_log("[pyc/replace] '%s'" % ( _pyc) )
                        _paths[_pyc] = _pyc_rel
                    else:
                        debug_log("[pyc/add] '%s'" % ( _pyc ) )
                        _paths[_pyc] = _pyc_rel
                        
            else:
                debug_log("[dir/add] '%s'" % ( absolute_path ) )
                _paths[absolute_path] = relative_path
                
        def GenerateZIP():
            for _abs, _rel in _paths.iteritems():
                print "[zip/add] %s" % ( _abs )
                self.write(_abs, _rel)
            
        parent_folder = os.path.dirname(pathname)
        # Retrieve the paths of the folder contents.
        contents = os.walk(pathname)
        
        try:            
            for root, folders, files in contents:
                # Include all subfolders, including empty ones.
                for folder_name in folders:
                    absolute_path = os.path.join(root, folder_name)
                    relative_path = absolute_path.replace(parent_folder + '\\','')
                    AddToZIP(absolute_path, relative_path)
                    
                for file_name in files:
                    absolute_path = os.path.join(root, file_name)
                    relative_path = absolute_path.replace(parent_folder + '\\','')
                    AddToZIP(absolute_path, relative_path)
                    
            GenerateZIP()
            
            debug_log("ZIP generated successfully.")
            
        except IOError, message:
            print message
            raise
        
        except OSError, message:
            print message
            raise
        
        except zipfile.BadZipfile, message:
            print message
            raise
        
        finally:
            self.close()
            
# usage
# create zip object
_zip_obj = _ZipFile("filenamehere.txt", 'w')
        
# enable debug mode
_zip_obj.debug = debug
        
# attempt to zip the dir
_zip_obj.writedir(
    pathname = dirpath
)

More like this

  1. Auto slug field by GaretJax 7 years, 4 months ago
  2. Arbitrary auto-generated primary keys by gsakkis 5 years, 1 month ago
  3. automatic urlpattern creator by sleytr 6 years ago
  4. send_html_mail by amitu 6 years, 10 months ago
  5. Deep package test runner by eternicode 4 years, 5 months ago

Comments

Please login first before commenting.