purge.py
138 lines
| 5.0 KiB
| text/x-python
|
PythonLexer
/ hgext / purge.py
|
r4311 | # Copyright (C) 2006 - Marco Barisione <marco@barisione.org> | ||
# | ||||
# This is a small extension for Mercurial (http://www.selenic.com/mercurial) | ||||
# that removes files not known to mercurial | ||||
# | ||||
# This program was inspired by the "cvspurge" script contained in CVS utilities | ||||
# (http://www.red-bean.com/cvsutils/). | ||||
# | ||||
# To enable the "purge" extension put these lines in your ~/.hgrc: | ||||
# [extensions] | ||||
# hgext.purge = | ||||
# | ||||
# For help on the usage of "hg purge" use: | ||||
# hg help purge | ||||
# | ||||
# This program is free software; you can redistribute it and/or modify | ||||
# it under the terms of the GNU General Public License as published by | ||||
# the Free Software Foundation; either version 2 of the License, or | ||||
# (at your option) any later version. | ||||
# | ||||
# This program is distributed in the hope that it will be useful, | ||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||||
# GNU General Public License for more details. | ||||
# | ||||
# You should have received a copy of the GNU General Public License | ||||
# along with this program; if not, write to the Free Software | ||||
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. | ||||
|
r6574 | from mercurial import util, commands, cmdutil | ||
|
r4311 | from mercurial.i18n import _ | ||
import os | ||||
|
r6573 | def purge(ui, repo, *dirs, **opts): | ||
'''removes files not tracked by mercurial | ||||
Delete files not known to mercurial, this is useful to test local and | ||||
uncommitted changes in the otherwise clean source tree. | ||||
This means that purge will delete: | ||||
- Unknown files: files marked with "?" by "hg status" | ||||
- Ignored files: files usually ignored by Mercurial because they match | ||||
a pattern in a ".hgignore" file | ||||
- Empty directories: in fact Mercurial ignores directories unless they | ||||
contain files under source control managment | ||||
But it will leave untouched: | ||||
- Unmodified tracked files | ||||
- Modified tracked files | ||||
- New files added to the repository (with "hg add") | ||||
If directories are given on the command line, only files in these | ||||
directories are considered. | ||||
Be careful with purge, you could irreversibly delete some files you | ||||
forgot to add to the repository. If you only want to print the list of | ||||
files that this program would delete use the --print option. | ||||
''' | ||||
act = not opts['print'] | ||||
ignored = bool(opts['all']) | ||||
abort_on_err = bool(opts['abort_on_err']) | ||||
eol = opts['print0'] and '\0' or '\n' | ||||
if eol == '\0': | ||||
# --print0 implies --print | ||||
act = False | ||||
force = bool(opts['force']) | ||||
|
r4311 | def error(msg): | ||
if abort_on_err: | ||||
raise util.Abort(msg) | ||||
else: | ||||
ui.warn(_('warning: %s\n') % msg) | ||||
def remove(remove_func, name): | ||||
if act: | ||||
try: | ||||
remove_func(os.path.join(repo.root, name)) | ||||
except OSError, e: | ||||
error(_('%s cannot be removed') % name) | ||||
else: | ||||
ui.write('%s%s' % (name, eol)) | ||||
|
r5517 | if not force: | ||
_check_fs(ui, repo) | ||||
|
r4311 | directories = [] | ||
files = [] | ||||
|
r6582 | match = cmdutil.match(repo, dirs, opts) | ||
|
r6588 | match.dir = directories.append | ||
|
r6582 | for src, f, st in repo.dirstate.statwalk(match.files(), match, | ||
|
r6588 | ignored=ignored): | ||
|
r6592 | if src == 'f' and f not in repo.dirstate: | ||
|
r4311 | files.append(f) | ||
directories.sort() | ||||
for f in files: | ||||
if f not in repo.dirstate: | ||||
ui.note(_('Removing file %s\n') % f) | ||||
remove(os.remove, f) | ||||
for f in directories[::-1]: | ||||
|
r4463 | if match(f) and not os.listdir(repo.wjoin(f)): | ||
|
r4311 | ui.note(_('Removing directory %s\n') % f) | ||
remove(os.rmdir, f) | ||||
|
r5517 | def _check_fs(ui, repo): | ||
|
r4311 | """Abort if there is the chance of having problems with name-mangling fs | ||
In a name mangling filesystem (e.g. a case insensitive one) | ||||
dirstate.walk() can yield filenames different from the ones | ||||
stored in the dirstate. This already confuses the status and | ||||
add commands, but with purge this may cause data loss. | ||||
|
r4516 | |||
|
r5517 | To prevent this, this function will abort if there are uncommitted | ||
changes. | ||||
""" | ||||
|
r4311 | |||
|
r5517 | # We can't use (files, match) to do a partial walk here - we wouldn't | ||
# notice a modified README file if the user ran "hg purge readme" | ||||
modified, added, removed, deleted = repo.status()[:4] | ||||
if modified or added or removed or deleted: | ||||
if not util.checkfolding(repo.path) and not ui.quiet: | ||||
ui.warn(_("Purging on name mangling filesystems is not " | ||||
"fully supported.\n")) | ||||
raise util.Abort(_("outstanding uncommitted changes")) | ||||
|
r4311 | |||
cmdtable = { | ||||
|
r4695 | 'purge|clean': | ||
|
r4311 | (purge, | ||
[('a', 'abort-on-err', None, _('abort if an error occurs')), | ||||
|
r4691 | ('', 'all', None, _('purge ignored files too')), | ||
|
r5517 | ('f', 'force', None, _('purge even when there are uncommitted changes')), | ||
|
r4311 | ('p', 'print', None, _('print the file names instead of deleting them')), | ||
('0', 'print0', None, _('end filenames with NUL, for use with xargs' | ||||
|
r4463 | ' (implies -p)')), | ||
|
r5147 | ] + commands.walkopts, | ||
|
r4311 | _('hg purge [OPTION]... [DIR]...')) | ||
} | ||||