mercurial-scm/hg: mercurial/archival.py annotate

annotate mercurial/archival.py @ 13047:6c375e07d673

branch: operate on branch names in local string space where possible Previously, branch names were ideally manipulated as UTF-8 strings, because they were stored as UTF-8 in the dirstate and the changelog and could not be safely converted to the local encoding and back. However, only about 80% of branch name code was actually using the right encoding conventions. This patch uses the localstr addition to allow working on branch names as local strings, which simplifies handling so that the previously incorrect code becomes correct.

author	Matt Mackall <mpm@selenic.com>
date	Wed, 24 Nov 2010 15:56:32 -0600
parents	f00953d9533c
children	5dac0d04b838

rev	line source
2112 2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	1 # archival.py - revision archival for mercurial
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	2 #
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	3 # Copyright 2006 Vadim Gelfer <vadim.gelfer@gmail.com>
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	4 #
8225 46293a0c7e9f updated license to be explicit about GPL version 2 Martin Geisler <mg@lazybytes.net> parents: 7770 diff changeset	5 # This software may be used and distributed according to the terms of the
10263 25e572394f5c Update license to GPLv2+ Matt Mackall <mpm@selenic.com> parents: 9614 diff changeset	6 # GNU General Public License version 2 or any later version.
2112 2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	7
3891 6b4127c7d52a Simplify i18n imports Matt Mackall <mpm@selenic.com> parents: 3877 diff changeset	8 from i18n import _
6211 f89fd07fc51d Expand import * to allow Pyflakes to find problems Joel Rosdahl <joel@rosdahl.net> parents: 6183 diff changeset	9 from node import hex
9614 58edd448da4f archive: add branch and tag informations to the .hg_archival.txt file Gilles Moris <gilles.moris@free.fr> parents: 8778 diff changeset	10 import cmdutil
13047 6c375e07d673 branch: operate on branch names in local string space where possible Matt Mackall <mpm@selenic.com> parents: 12323 diff changeset	11 import util, encoding
8312 b87a50b7125c separate import lines from mercurial and general python modules Simon Heimberg <simohe@besonet.ch> parents: 8225 diff changeset	12 import cStringIO, os, stat, tarfile, time, zipfile
4652 06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	13 import zlib, gzip
2112 2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	14
11558 d8f6458434ec archival: remove prefix argument from archivers Martin Geisler <mg@lazybytes.net> parents: 11557 diff changeset	15 def tidyprefix(dest, kind, prefix):
2112 2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	16 '''choose prefix to use for names in archive. make sure prefix is
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	17 safe for consumers.'''
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	18
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	19 if prefix:
5842 111ed8c871bf Use util.normpath() instead of direct path string operation. Shun-ichi GOTO <shunichi.goto@gmail.com> parents: 4951 diff changeset	20 prefix = util.normpath(prefix)
2112 2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	21 else:
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	22 if not isinstance(dest, str):
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	23 raise ValueError('dest must be string if no prefix')
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	24 prefix = os.path.basename(dest)
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	25 lower = prefix.lower()
11558 d8f6458434ec archival: remove prefix argument from archivers Martin Geisler <mg@lazybytes.net> parents: 11557 diff changeset	26 for sfx in exts.get(kind, []):
2112 2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	27 if lower.endswith(sfx):
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	28 prefix = prefix[:-len(sfx)]
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	29 break
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	30 lpfx = os.path.normpath(util.localpath(prefix))
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	31 prefix = util.pconvert(lpfx)
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	32 if not prefix.endswith('/'):
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	33 prefix += '/'
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	34 if prefix.startswith('../') or os.path.isabs(lpfx) or '/../' in prefix:
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	35 raise util.Abort(_('archive prefix contains illegal components'))
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	36 return prefix
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	37
11557 57bdc2239535 archival: move commands.archive.guess_type to archival.guesskind Martin Geisler <mg@lazybytes.net> parents: 10282 diff changeset	38 exts = {
57bdc2239535 archival: move commands.archive.guess_type to archival.guesskind Martin Geisler <mg@lazybytes.net> parents: 10282 diff changeset	39 'tar': ['.tar'],
57bdc2239535 archival: move commands.archive.guess_type to archival.guesskind Martin Geisler <mg@lazybytes.net> parents: 10282 diff changeset	40 'tbz2': ['.tbz2', '.tar.bz2'],
57bdc2239535 archival: move commands.archive.guess_type to archival.guesskind Martin Geisler <mg@lazybytes.net> parents: 10282 diff changeset	41 'tgz': ['.tgz', '.tar.gz'],
57bdc2239535 archival: move commands.archive.guess_type to archival.guesskind Martin Geisler <mg@lazybytes.net> parents: 10282 diff changeset	42 'zip': ['.zip'],
57bdc2239535 archival: move commands.archive.guess_type to archival.guesskind Martin Geisler <mg@lazybytes.net> parents: 10282 diff changeset	43 }
57bdc2239535 archival: move commands.archive.guess_type to archival.guesskind Martin Geisler <mg@lazybytes.net> parents: 10282 diff changeset	44
57bdc2239535 archival: move commands.archive.guess_type to archival.guesskind Martin Geisler <mg@lazybytes.net> parents: 10282 diff changeset	45 def guesskind(dest):
57bdc2239535 archival: move commands.archive.guess_type to archival.guesskind Martin Geisler <mg@lazybytes.net> parents: 10282 diff changeset	46 for kind, extensions in exts.iteritems():
57bdc2239535 archival: move commands.archive.guess_type to archival.guesskind Martin Geisler <mg@lazybytes.net> parents: 10282 diff changeset	47 if util.any(dest.endswith(ext) for ext in extensions):
57bdc2239535 archival: move commands.archive.guess_type to archival.guesskind Martin Geisler <mg@lazybytes.net> parents: 10282 diff changeset	48 return kind
57bdc2239535 archival: move commands.archive.guess_type to archival.guesskind Martin Geisler <mg@lazybytes.net> parents: 10282 diff changeset	49 return None
57bdc2239535 archival: move commands.archive.guess_type to archival.guesskind Martin Geisler <mg@lazybytes.net> parents: 10282 diff changeset	50
57bdc2239535 archival: move commands.archive.guess_type to archival.guesskind Martin Geisler <mg@lazybytes.net> parents: 10282 diff changeset	51
8778 c5f36402daad use new style classes Benoit Boissinot <benoit.boissinot@ens-lyon.org> parents: 8527 diff changeset	52 class tarit(object):
2112 2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	53 '''write archive to tar file or stream. can write uncompressed,
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	54 or compress with gzip or bzip2.'''
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	55
4652 06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	56 class GzipFileWithTime(gzip.GzipFile):
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	57
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	58 def __init__(self, args, *kw):
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	59 timestamp = None
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	60 if 'timestamp' in kw:
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	61 timestamp = kw.pop('timestamp')
8527 f9a80054dd3c use 'x is None' instead of 'x == None' Martin Geisler <mg@lazybytes.net> parents: 8312 diff changeset	62 if timestamp is None:
4652 06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	63 self.timestamp = time.time()
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	64 else:
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	65 self.timestamp = timestamp
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	66 gzip.GzipFile.__init__(self, args, *kw)
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	67
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	68 def _write_gzip_header(self):
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	69 self.fileobj.write('\037\213') # magic header
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	70 self.fileobj.write('\010') # compression method
6495 3130c9ded04e python-2.6: deprecation of GzipFile.filename Dirkjan Ochtman <dirkjan@ochtman.nl> parents: 6211 diff changeset	71 # Python 2.6 deprecates self.filename
6498 315548fcc76b Remove trailing space Thomas Arendsen Hein <thomas@intevation.de> parents: 6495 diff changeset	72 fname = getattr(self, 'name', None) or self.filename
4652 06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	73 flags = 0
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	74 if fname:
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	75 flags = gzip.FNAME
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	76 self.fileobj.write(chr(flags))
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	77 gzip.write32u(self.fileobj, long(self.timestamp))
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	78 self.fileobj.write('\002')
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	79 self.fileobj.write('\377')
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	80 if fname:
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	81 self.fileobj.write(fname + '\000')
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	82
11558 d8f6458434ec archival: remove prefix argument from archivers Martin Geisler <mg@lazybytes.net> parents: 11557 diff changeset	83 def __init__(self, dest, mtime, kind=''):
2477 857591c586e0 use commit time as mtime for file archives. Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2353 diff changeset	84 self.mtime = mtime
4652 06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	85
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	86 def taropen(name, mode, fileobj=None):
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	87 if kind == 'gz':
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	88 mode = mode[0]
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	89 if not fileobj:
4731 1d5a2ee683b0 Fix tgz archival on Windows. csaba.henk@creo.hu parents: 4685 diff changeset	90 fileobj = open(name, mode + 'b')
4652 06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	91 gzfileobj = self.GzipFileWithTime(name, mode + 'b',
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	92 zlib.Z_BEST_COMPRESSION,
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	93 fileobj, timestamp=mtime)
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	94 return tarfile.TarFile.taropen(name, mode, gzfileobj)
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	95 else:
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	96 return tarfile.open(name, mode + kind, fileobj)
06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	97
2112 2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	98 if isinstance(dest, str):
4652 06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	99 self.z = taropen(dest, mode='w:')
2112 2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	100 else:
4357 3f1b0c0fb4fd Work around python 2.5.1 tarfile regression Brendan Cully <brendan@kublai.com> parents: 3615 diff changeset	101 # Python 2.5-2.5.1 have a regression that requires a name arg
4652 06de65673ec2 timestamp of gzip archives taken from changeset context csaba.henk@creo.hu parents: 4370 diff changeset	102 self.z = taropen(name='', mode='w\|', fileobj=dest)
2112 2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	103
4831 6f08bc1bd00b archive: add symlink support Alexis S. L. Carvalho <alexis@cecm.usp.br> parents: 4830 diff changeset	104 def addfile(self, name, mode, islink, data):
11558 d8f6458434ec archival: remove prefix argument from archivers Martin Geisler <mg@lazybytes.net> parents: 11557 diff changeset	105 i = tarfile.TarInfo(name)
2112 2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	106 i.mtime = self.mtime
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	107 i.size = len(data)
4831 6f08bc1bd00b archive: add symlink support Alexis S. L. Carvalho <alexis@cecm.usp.br> parents: 4830 diff changeset	108 if islink:
6f08bc1bd00b archive: add symlink support Alexis S. L. Carvalho <alexis@cecm.usp.br> parents: 4830 diff changeset	109 i.type = tarfile.SYMTYPE
6f08bc1bd00b archive: add symlink support Alexis S. L. Carvalho <alexis@cecm.usp.br> parents: 4830 diff changeset	110 i.mode = 0777
6f08bc1bd00b archive: add symlink support Alexis S. L. Carvalho <alexis@cecm.usp.br> parents: 4830 diff changeset	111 i.linkname = data
6f08bc1bd00b archive: add symlink support Alexis S. L. Carvalho <alexis@cecm.usp.br> parents: 4830 diff changeset	112 data = None
7770 fd3e5ff53a31 fix disappearing symlinks [issue1509] Peter van Dijk <mercurial-bugs@selenic.com> parents: 6913 diff changeset	113 i.size = 0
4831 6f08bc1bd00b archive: add symlink support Alexis S. L. Carvalho <alexis@cecm.usp.br> parents: 4830 diff changeset	114 else:
6f08bc1bd00b archive: add symlink support Alexis S. L. Carvalho <alexis@cecm.usp.br> parents: 4830 diff changeset	115 i.mode = mode
6f08bc1bd00b archive: add symlink support Alexis S. L. Carvalho <alexis@cecm.usp.br> parents: 4830 diff changeset	116 data = cStringIO.StringIO(data)
6f08bc1bd00b archive: add symlink support Alexis S. L. Carvalho <alexis@cecm.usp.br> parents: 4830 diff changeset	117 self.z.addfile(i, data)
2112 2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	118
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	119 def done(self):
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	120 self.z.close()
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	121
8778 c5f36402daad use new style classes Benoit Boissinot <benoit.boissinot@ens-lyon.org> parents: 8527 diff changeset	122 class tellable(object):
2112 2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	123 '''provide tell method for zipfile.ZipFile when writing to http
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	124 response file object.'''
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	125
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	126 def __init__(self, fp):
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	127 self.fp = fp
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	128 self.offset = 0
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	129
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	130 def __getattr__(self, key):
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	131 return getattr(self.fp, key)
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	132
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	133 def write(self, s):
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	134 self.fp.write(s)
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	135 self.offset += len(s)
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	136
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	137 def tell(self):
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	138 return self.offset
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	139
8778 c5f36402daad use new style classes Benoit Boissinot <benoit.boissinot@ens-lyon.org> parents: 8527 diff changeset	140 class zipit(object):
2112 2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	141 '''write archive to zip file or stream. can write uncompressed,
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	142 or compressed with deflate.'''
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	143
11558 d8f6458434ec archival: remove prefix argument from archivers Martin Geisler <mg@lazybytes.net> parents: 11557 diff changeset	144 def __init__(self, dest, mtime, compress=True):
2168 dd4ec4576cc8 Proper check to see if zip dest needs to be wrapped in tellable Colin McMillen <mcmillen@cs.cmu.edu> parents: 2112 diff changeset	145 if not isinstance(dest, str):
dd4ec4576cc8 Proper check to see if zip dest needs to be wrapped in tellable Colin McMillen <mcmillen@cs.cmu.edu> parents: 2112 diff changeset	146 try:
dd4ec4576cc8 Proper check to see if zip dest needs to be wrapped in tellable Colin McMillen <mcmillen@cs.cmu.edu> parents: 2112 diff changeset	147 dest.tell()
2169 4564794b6f55 Combine catching exceptions added in dd4ec4576cc8 in one except statement. Thomas Arendsen Hein <thomas@intevation.de> parents: 2168 diff changeset	148 except (AttributeError, IOError):
2168 dd4ec4576cc8 Proper check to see if zip dest needs to be wrapped in tellable Colin McMillen <mcmillen@cs.cmu.edu> parents: 2112 diff changeset	149 dest = tellable(dest)
2112 2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	150 self.z = zipfile.ZipFile(dest, 'w',
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	151 compress and zipfile.ZIP_DEFLATED or
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	152 zipfile.ZIP_STORED)
12319 381f131220ad archive: set date to 1980 for very old zip files Martin Geisler <mg@aragost.com> parents: 10282 diff changeset	153
381f131220ad archive: set date to 1980 for very old zip files Martin Geisler <mg@aragost.com> parents: 10282 diff changeset	154 # Python's zipfile module emits deprecation warnings if we try
381f131220ad archive: set date to 1980 for very old zip files Martin Geisler <mg@aragost.com> parents: 10282 diff changeset	155 # to store files with a date before 1980.
381f131220ad archive: set date to 1980 for very old zip files Martin Geisler <mg@aragost.com> parents: 10282 diff changeset	156 epoch = 315532800 # calendar.timegm((1980, 1, 1, 0, 0, 0, 1, 1, 0))
381f131220ad archive: set date to 1980 for very old zip files Martin Geisler <mg@aragost.com> parents: 10282 diff changeset	157 if mtime < epoch:
381f131220ad archive: set date to 1980 for very old zip files Martin Geisler <mg@aragost.com> parents: 10282 diff changeset	158 mtime = epoch
381f131220ad archive: set date to 1980 for very old zip files Martin Geisler <mg@aragost.com> parents: 10282 diff changeset	159
2477 857591c586e0 use commit time as mtime for file archives. Vadim Gelfer <vadim.gelfer@gmail.com> parents: 2353 diff changeset	160 self.date_time = time.gmtime(mtime)[:6]
2112 2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	161
4831 6f08bc1bd00b archive: add symlink support Alexis S. L. Carvalho <alexis@cecm.usp.br> parents: 4830 diff changeset	162 def addfile(self, name, mode, islink, data):
11558 d8f6458434ec archival: remove prefix argument from archivers Martin Geisler <mg@lazybytes.net> parents: 11557 diff changeset	163 i = zipfile.ZipInfo(name, self.date_time)
2112 2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	164 i.compress_type = self.z.compression
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	165 # unzip will not honor unix file modes unless file creator is
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	166 # set to unix (id 3).
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	167 i.create_system = 3
4831 6f08bc1bd00b archive: add symlink support Alexis S. L. Carvalho <alexis@cecm.usp.br> parents: 4830 diff changeset	168 ftype = stat.S_IFREG
6f08bc1bd00b archive: add symlink support Alexis S. L. Carvalho <alexis@cecm.usp.br> parents: 4830 diff changeset	169 if islink:
6f08bc1bd00b archive: add symlink support Alexis S. L. Carvalho <alexis@cecm.usp.br> parents: 4830 diff changeset	170 mode = 0777
6f08bc1bd00b archive: add symlink support Alexis S. L. Carvalho <alexis@cecm.usp.br> parents: 4830 diff changeset	171 ftype = stat.S_IFLNK
6f08bc1bd00b archive: add symlink support Alexis S. L. Carvalho <alexis@cecm.usp.br> parents: 4830 diff changeset	172 i.external_attr = (mode \| ftype) << 16L
2112 2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	173 self.z.writestr(i, data)
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	174
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	175 def done(self):
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	176 self.z.close()
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	177
8778 c5f36402daad use new style classes Benoit Boissinot <benoit.boissinot@ens-lyon.org> parents: 8527 diff changeset	178 class fileit(object):
2112 2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	179 '''write archive as files in directory.'''
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	180
11558 d8f6458434ec archival: remove prefix argument from archivers Martin Geisler <mg@lazybytes.net> parents: 11557 diff changeset	181 def __init__(self, name, mtime):
2112 2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	182 self.basedir = name
4830 74f36b1027f4 archive: use util.opener when archiving files. Alexis S. L. Carvalho <alexis@cecm.usp.br> parents: 4731 diff changeset	183 self.opener = util.opener(self.basedir)
2112 2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	184
4831 6f08bc1bd00b archive: add symlink support Alexis S. L. Carvalho <alexis@cecm.usp.br> parents: 4830 diff changeset	185 def addfile(self, name, mode, islink, data):
6f08bc1bd00b archive: add symlink support Alexis S. L. Carvalho <alexis@cecm.usp.br> parents: 4830 diff changeset	186 if islink:
6f08bc1bd00b archive: add symlink support Alexis S. L. Carvalho <alexis@cecm.usp.br> parents: 4830 diff changeset	187 self.opener.symlink(data, name)
6f08bc1bd00b archive: add symlink support Alexis S. L. Carvalho <alexis@cecm.usp.br> parents: 4830 diff changeset	188 return
4830 74f36b1027f4 archive: use util.opener when archiving files. Alexis S. L. Carvalho <alexis@cecm.usp.br> parents: 4731 diff changeset	189 f = self.opener(name, "w", atomictemp=True)
74f36b1027f4 archive: use util.opener when archiving files. Alexis S. L. Carvalho <alexis@cecm.usp.br> parents: 4731 diff changeset	190 f.write(data)
74f36b1027f4 archive: use util.opener when archiving files. Alexis S. L. Carvalho <alexis@cecm.usp.br> parents: 4731 diff changeset	191 f.rename()
2112 2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	192 destfile = os.path.join(self.basedir, name)
4830 74f36b1027f4 archive: use util.opener when archiving files. Alexis S. L. Carvalho <alexis@cecm.usp.br> parents: 4731 diff changeset	193 os.chmod(destfile, mode)
2112 2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	194
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	195 def done(self):
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	196 pass
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	197
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	198 archivers = {
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	199 'files': fileit,
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	200 'tar': tarit,
11558 d8f6458434ec archival: remove prefix argument from archivers Martin Geisler <mg@lazybytes.net> parents: 11557 diff changeset	201 'tbz2': lambda name, mtime: tarit(name, mtime, 'bz2'),
d8f6458434ec archival: remove prefix argument from archivers Martin Geisler <mg@lazybytes.net> parents: 11557 diff changeset	202 'tgz': lambda name, mtime: tarit(name, mtime, 'gz'),
d8f6458434ec archival: remove prefix argument from archivers Martin Geisler <mg@lazybytes.net> parents: 11557 diff changeset	203 'uzip': lambda name, mtime: zipit(name, mtime, False),
2112 2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	204 'zip': zipit,
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	205 }
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	206
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	207 def archive(repo, dest, node, kind, decode=True, matchfn=None,
12323 f00953d9533c subrepo: add support for 'hg archive' Martin Geisler <mg@aragost.com> parents: 12321 diff changeset	208 prefix=None, mtime=None, subrepos=False):
2112 2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	209 '''create archive of repo as it was at node.
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	210
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	211 dest can be name of directory, name of archive file, or file
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	212 object to write archive to.
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	213
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	214 kind is type of archive to create.
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	215
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	216 decode tells whether to put files through decode filters from
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	217 hgrc.
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	218
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	219 matchfn is function to filter names of files to write to archive.
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	220
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	221 prefix is name of path to put before every archive member.'''
2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	222
11558 d8f6458434ec archival: remove prefix argument from archivers Martin Geisler <mg@lazybytes.net> parents: 11557 diff changeset	223 if kind == 'files':
d8f6458434ec archival: remove prefix argument from archivers Martin Geisler <mg@lazybytes.net> parents: 11557 diff changeset	224 if prefix:
d8f6458434ec archival: remove prefix argument from archivers Martin Geisler <mg@lazybytes.net> parents: 11557 diff changeset	225 raise util.Abort(_('cannot give prefix when archiving to files'))
d8f6458434ec archival: remove prefix argument from archivers Martin Geisler <mg@lazybytes.net> parents: 11557 diff changeset	226 else:
d8f6458434ec archival: remove prefix argument from archivers Martin Geisler <mg@lazybytes.net> parents: 11557 diff changeset	227 prefix = tidyprefix(dest, kind, prefix)
d8f6458434ec archival: remove prefix argument from archivers Martin Geisler <mg@lazybytes.net> parents: 11557 diff changeset	228
4951 667290b6c95e archive: delay extraction of file revisions Alexis S. L. Carvalho <alexis@cecm.usp.br> parents: 4831 diff changeset	229 def write(name, mode, islink, getdata):
10282 08a0f04b56bd many, many trivial check-code fixups Matt Mackall <mpm@selenic.com> parents: 10263 diff changeset	230 if matchfn and not matchfn(name):
08a0f04b56bd many, many trivial check-code fixups Matt Mackall <mpm@selenic.com> parents: 10263 diff changeset	231 return
4951 667290b6c95e archive: delay extraction of file revisions Alexis S. L. Carvalho <alexis@cecm.usp.br> parents: 4831 diff changeset	232 data = getdata()
2112 2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	233 if decode:
4005 656e06eebda7 replace filehandle version of wwrite with wwritedata Matt Mackall <mpm@selenic.com> parents: 3968 diff changeset	234 data = repo.wwritedata(name, data)
11558 d8f6458434ec archival: remove prefix argument from archivers Martin Geisler <mg@lazybytes.net> parents: 11557 diff changeset	235 archiver.addfile(prefix + name, mode, islink, data)
2112 2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	236
6019 b70a530bdb93 cleanly abort on unknown archive type (issue966) Dirkjan Ochtman <dirkjan@ochtman.nl> parents: 4951 diff changeset	237 if kind not in archivers:
6913 580d5e6bfc1f move % out of translatable strings Martin Geisler <mg@daimi.au.dk> parents: 6749 diff changeset	238 raise util.Abort(_("unknown archive type '%s'") % kind)
6749 51b0e799352f manifest: remove execf/linkf methods Matt Mackall <mpm@selenic.com> parents: 6747 diff changeset	239
51b0e799352f manifest: remove execf/linkf methods Matt Mackall <mpm@selenic.com> parents: 6747 diff changeset	240 ctx = repo[node]
11558 d8f6458434ec archival: remove prefix argument from archivers Martin Geisler <mg@lazybytes.net> parents: 11557 diff changeset	241 archiver = archivers[kind](dest, mtime or ctx.date()[0])
6749 51b0e799352f manifest: remove execf/linkf methods Matt Mackall <mpm@selenic.com> parents: 6747 diff changeset	242
6183 0750e4ba9d3d Add config option to disable putting .hg_archival.txt inside archives. Thomas Arendsen Hein <thomas@intevation.de> parents: 6020 diff changeset	243 if repo.ui.configbool("ui", "archivemeta", True):
9614 58edd448da4f archive: add branch and tag informations to the .hg_archival.txt file Gilles Moris <gilles.moris@free.fr> parents: 8778 diff changeset	244 def metadata():
58edd448da4f archive: add branch and tag informations to the .hg_archival.txt file Gilles Moris <gilles.moris@free.fr> parents: 8778 diff changeset	245 base = 'repo: %s\nnode: %s\nbranch: %s\n' % (
13047 6c375e07d673 branch: operate on branch names in local string space where possible Matt Mackall <mpm@selenic.com> parents: 12323 diff changeset	246 repo[0].hex(), hex(node), encoding.fromlocal(ctx.branch()))
9614 58edd448da4f archive: add branch and tag informations to the .hg_archival.txt file Gilles Moris <gilles.moris@free.fr> parents: 8778 diff changeset	247
58edd448da4f archive: add branch and tag informations to the .hg_archival.txt file Gilles Moris <gilles.moris@free.fr> parents: 8778 diff changeset	248 tags = ''.join('tag: %s\n' % t for t in ctx.tags()
58edd448da4f archive: add branch and tag informations to the .hg_archival.txt file Gilles Moris <gilles.moris@free.fr> parents: 8778 diff changeset	249 if repo.tagtype(t) == 'global')
58edd448da4f archive: add branch and tag informations to the .hg_archival.txt file Gilles Moris <gilles.moris@free.fr> parents: 8778 diff changeset	250 if not tags:
58edd448da4f archive: add branch and tag informations to the .hg_archival.txt file Gilles Moris <gilles.moris@free.fr> parents: 8778 diff changeset	251 repo.ui.pushbuffer()
58edd448da4f archive: add branch and tag informations to the .hg_archival.txt file Gilles Moris <gilles.moris@free.fr> parents: 8778 diff changeset	252 opts = {'template': '{latesttag}\n{latesttagdistance}',
58edd448da4f archive: add branch and tag informations to the .hg_archival.txt file Gilles Moris <gilles.moris@free.fr> parents: 8778 diff changeset	253 'style': '', 'patch': None, 'git': None}
58edd448da4f archive: add branch and tag informations to the .hg_archival.txt file Gilles Moris <gilles.moris@free.fr> parents: 8778 diff changeset	254 cmdutil.show_changeset(repo.ui, repo, opts).show(ctx)
58edd448da4f archive: add branch and tag informations to the .hg_archival.txt file Gilles Moris <gilles.moris@free.fr> parents: 8778 diff changeset	255 ltags, dist = repo.ui.popbuffer().split('\n')
58edd448da4f archive: add branch and tag informations to the .hg_archival.txt file Gilles Moris <gilles.moris@free.fr> parents: 8778 diff changeset	256 tags = ''.join('latesttag: %s\n' % t for t in ltags.split(':'))
58edd448da4f archive: add branch and tag informations to the .hg_archival.txt file Gilles Moris <gilles.moris@free.fr> parents: 8778 diff changeset	257 tags += 'latesttagdistance: %s\n' % dist
58edd448da4f archive: add branch and tag informations to the .hg_archival.txt file Gilles Moris <gilles.moris@free.fr> parents: 8778 diff changeset	258
58edd448da4f archive: add branch and tag informations to the .hg_archival.txt file Gilles Moris <gilles.moris@free.fr> parents: 8778 diff changeset	259 return base + tags
58edd448da4f archive: add branch and tag informations to the .hg_archival.txt file Gilles Moris <gilles.moris@free.fr> parents: 8778 diff changeset	260
58edd448da4f archive: add branch and tag informations to the .hg_archival.txt file Gilles Moris <gilles.moris@free.fr> parents: 8778 diff changeset	261 write('.hg_archival.txt', 0644, False, metadata)
58edd448da4f archive: add branch and tag informations to the .hg_archival.txt file Gilles Moris <gilles.moris@free.fr> parents: 8778 diff changeset	262
6749 51b0e799352f manifest: remove execf/linkf methods Matt Mackall <mpm@selenic.com> parents: 6747 diff changeset	263 for f in ctx:
51b0e799352f manifest: remove execf/linkf methods Matt Mackall <mpm@selenic.com> parents: 6747 diff changeset	264 ff = ctx.flags(f)
51b0e799352f manifest: remove execf/linkf methods Matt Mackall <mpm@selenic.com> parents: 6747 diff changeset	265 write(f, 'x' in ff and 0755 or 0644, 'l' in ff, ctx[f].data)
12323 f00953d9533c subrepo: add support for 'hg archive' Martin Geisler <mg@aragost.com> parents: 12321 diff changeset	266
f00953d9533c subrepo: add support for 'hg archive' Martin Geisler <mg@aragost.com> parents: 12321 diff changeset	267 if subrepos:
f00953d9533c subrepo: add support for 'hg archive' Martin Geisler <mg@aragost.com> parents: 12321 diff changeset	268 for subpath in ctx.substate:
f00953d9533c subrepo: add support for 'hg archive' Martin Geisler <mg@aragost.com> parents: 12321 diff changeset	269 sub = ctx.sub(subpath)
f00953d9533c subrepo: add support for 'hg archive' Martin Geisler <mg@aragost.com> parents: 12321 diff changeset	270 sub.archive(archiver, prefix)
f00953d9533c subrepo: add support for 'hg archive' Martin Geisler <mg@aragost.com> parents: 12321 diff changeset	271
2112 2b03c6733efa add "archive" command, like "cvs export" only better. Vadim Gelfer <vadim.gelfer@gmail.com> parents: diff changeset	272 archiver.done()

Mercurial > public > mercurial-scm > hg

annotate mercurial/archival.py @ 13047:6c375e07d673