purge.py
110 lines
| 4.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 _ | ||
|
r8043 | import os, stat | ||
|
r4311 | |||
|
r6573 | def purge(ui, repo, *dirs, **opts): | ||
|
r7605 | '''removes files not tracked by Mercurial | ||
|
r6573 | |||
|
r7998 | Delete files not known to Mercurial. This is useful to test local | ||
and uncommitted changes in an otherwise-clean source tree. | ||||
|
r6573 | |||
This means that purge will delete: | ||||
- Unknown files: files marked with "?" by "hg status" | ||||
|
r7998 | - Empty directories: in fact Mercurial ignores directories unless | ||
they contain files under source control managment | ||||
|
r6573 | But it will leave untouched: | ||
|
r7605 | - Modified and unmodified tracked files | ||
- Ignored files (unless --all is specified) | ||||
|
r6573 | - New files added to the repository (with "hg add") | ||
If directories are given on the command line, only files in these | ||||
directories are considered. | ||||
|
r7998 | Be careful with purge, as 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. | ||||
|
r6573 | ''' | ||
act = not opts['print'] | ||||
|
r6757 | eol = '\n' | ||
if opts['print0']: | ||||
eol = '\0' | ||||
act = False # --print0 implies --print | ||||
|
r4311 | |||
def remove(remove_func, name): | ||||
if act: | ||||
try: | ||||
|
r7570 | remove_func(repo.wjoin(name)) | ||
|
r7280 | except OSError: | ||
|
r6757 | m = _('%s cannot be removed') % name | ||
if opts['abort_on_err']: | ||||
raise util.Abort(m) | ||||
ui.warn(_('warning: %s\n') % m) | ||||
|
r4311 | else: | ||
ui.write('%s%s' % (name, eol)) | ||||
|
r8043 | def removefile(path): | ||
|
r8044 | try: | ||
os.remove(path) | ||||
except OSError: | ||||
# read-only files cannot be unlinked under Windows | ||||
s = os.stat(path) | ||||
if (s.st_mode & stat.S_IWRITE) != 0: | ||||
raise | ||||
os.chmod(path, stat.S_IMODE(s.st_mode) | stat.S_IWRITE) | ||||
os.remove(path) | ||||
|
r8043 | |||
|
r4311 | directories = [] | ||
|
r6582 | match = cmdutil.match(repo, dirs, opts) | ||
|
r6588 | match.dir = directories.append | ||
|
r6754 | status = repo.status(match=match, ignored=opts['all'], unknown=True) | ||
|
r4311 | |||
|
r6762 | for f in util.sort(status[4] + status[5]): | ||
|
r6754 | ui.note(_('Removing file %s\n') % f) | ||
|
r8043 | remove(removefile, f) | ||
|
r4311 | |||
|
r6762 | for f in util.sort(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) | ||||
cmdtable = { | ||||
|
r4695 | 'purge|clean': | ||
|
r4311 | (purge, | ||
[('a', 'abort-on-err', None, _('abort if an error occurs')), | ||||
|
r4691 | ('', 'all', None, _('purge ignored files too')), | ||
|
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]...')) | ||
} | ||||