- Author:
- sleepycal
- Posted:
- August 6, 2011
- Language:
- Python
- Version:
- Not specified
- 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
- Template tag - list punctuation for a list of items by shapiromatron 9 months ago
- JSONRequestMiddleware adds a .json() method to your HttpRequests by cdcarter 9 months, 1 week ago
- Serializer factory with Django Rest Framework by julio 1 year, 4 months ago
- Image compression before saving the new model / work with JPG, PNG by Schleidens 1 year, 4 months ago
- Help text hyperlinks by sa2812 1 year, 5 months ago
Comments
Please login first before commenting.