saltstack在解压zip文件时保留原始修改时间的方法
def unzip(zip_file, dest,excludes
=None, options
=None, template
=None, runas
=None, trim_output
=False, password
=None, extract_perms
=True):
if not excludes: excludes
= []if runas: euid
= os.geteuid() egid
= os.getegid() uinfo
= __salt__['user.info'](runas)if not uinfo:raise SaltInvocationError("User '{0}' does not exist".format(runas) )
zip_file, dest
= _render_filenames(zip_file, dest, None, template)
if runas and (euid != uinfo['uid'] or egid != uinfo['gid']):# Change the egid first, as changing it after the euid will fail # if the runas user is non-privileged.
os.setegid(uinfo['gid'])
os.seteuid(uinfo['uid'])
try:
exc = None
# Define cleaned_files here so that an exception will not prevent this
# variable from being defined and cause a NameError in the return
# statement at the end of the function.
cleaned_files = []
with contextlib.closing(zipfile.ZipFile(zip_file, "r")) as zfile:
# 旧方法修改文件的修改时间
if True:
files = zfile.namelist()
cleaned_files.extend()
for file in zfile.infolist():
d = file.date_time#修改时间元信息
gettime = "%s/%s/%s %s:%s" % (d, d, d, d, d)
zfile.extract(file, dest, password)#解压
filep = os.path.join(dest, file.filename)
print "recover:%s modify time" % filep
timearry = time.mktime(time.strptime(gettime, '%Y/%m/%d %H:%M'))
os.utime(filep, (timearry, timearry))#替换时间
else:
files = zfile.namelist()
if isinstance(excludes, string_types):
excludes =
elif isinstance(excludes, (float, integer_types)):
excludes =
cleaned_files.extend()
for target in cleaned_files:
if target not in excludes:
if salt.utils.is_windows() is False:
info = zfile.getinfo(target)
# Check if zipped file is a symbolic link
if stat.S_ISLNK(info.external_attr >> 16):
source = zfile.read(target)
os.symlink(source, os.path.join(dest, target))
continue
zfile.extract(target, dest, password)
modify_date = zfile.getinfo(target).date_time
gettime = "%s/%s/%s %s:%s" % (modify_date, modify_date, modify_date, modify_date, modify_date)
file_path = os.path.join(dest, target)
timearry = time.mktime(time.strptime(gettime, '%Y/%m/%d %H:%M'))
os.utime(file_path, (timearry, timearry))
if extract_perms:
perm = zfile.getinfo(target).external_attr >> 16
if perm == 0:
umask_ = os.umask(0)
os.umask(umask_)
if target.endswith('/'):
perm = 0o777 & ~umask_
else:
perm = 0o666 & ~umask_
os.chmod(os.path.join(dest, target), perm)
except Exception as exc:
pass
finally:
# Restore the euid/egid
if runas:
os.seteuid(euid)
os.setegid(egid)
if exc is not None:
# Wait to raise the exception until euid/egid are restored to avoid
# permission errors in writing to minion log.
raise CommandExecutionError(
'Exception encountered unpacking zipfile: {0}'.format(exc)
)
return _trim_files(cleaned_files, trim_output)
页:
[1]