purge.py
109 lines
| 3.9 KiB
| text/x-python
|
PythonLexer
/ hgext / purge.py
Emanuele Aina
|
r4311 | # Copyright (C) 2006 - Marco Barisione <marco@barisione.org> | ||
# | ||||
Dirkjan Ochtman
|
r8936 | # This is a small extension for Mercurial (http://mercurial.selenic.com/) | ||
Emanuele Aina
|
r4311 | # that removes files not known to mercurial | ||
# | ||||
Martin Geisler
|
r9270 | # This program was inspired by the "cvspurge" script contained in CVS | ||
# utilities (http://www.red-bean.com/cvsutils/). | ||||
Emanuele Aina
|
r4311 | # | ||
# 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 | ||||
Martin Geisler
|
r15782 | # along with this program; if not, see <http://www.gnu.org/licenses/>. | ||
Emanuele Aina
|
r4311 | |||
Dirkjan Ochtman
|
r8934 | '''command to delete untracked files from the working directory''' | ||
Dirkjan Ochtman
|
r8873 | |||
Matt Mackall
|
r14322 | from mercurial import util, commands, cmdutil, scmutil | ||
Emanuele Aina
|
r4311 | from mercurial.i18n import _ | ||
Patrick Mezard
|
r8043 | import os, stat | ||
Emanuele Aina
|
r4311 | |||
Adrian Buehlmann
|
r14310 | cmdtable = {} | ||
command = cmdutil.command(cmdtable) | ||||
@command('purge|clean', | ||||
[('a', 'abort-on-err', None, _('abort if an error occurs')), | ||||
('', 'all', None, _('purge ignored files too')), | ||||
('p', 'print', None, _('print filenames instead of deleting them')), | ||||
('0', 'print0', None, _('end filenames with NUL, for use with xargs' | ||||
' (implies -p/--print)')), | ||||
] + commands.walkopts, | ||||
_('hg purge [OPTION]... [DIR]...')) | ||||
Matt Mackall
|
r6573 | def purge(ui, repo, *dirs, **opts): | ||
Benjamin Pollack
|
r7605 | '''removes files not tracked by Mercurial | ||
Matt Mackall
|
r6573 | |||
Martin Geisler
|
r9270 | Delete files not known to Mercurial. This is useful to test local | ||
and uncommitted changes in an otherwise-clean source tree. | ||||
Matt Mackall
|
r6573 | |||
This means that purge will delete: | ||||
Martin Geisler
|
r9215 | |||
Martin Geisler
|
r10973 | - Unknown files: files marked with "?" by :hg:`status` | ||
Martin Geisler
|
r9270 | - Empty directories: in fact Mercurial ignores directories unless | ||
they contain files under source control management | ||||
Martin Geisler
|
r9215 | |||
Matt Mackall
|
r6573 | But it will leave untouched: | ||
Martin Geisler
|
r9215 | |||
- Modified and unmodified tracked files | ||||
- Ignored files (unless --all is specified) | ||||
Martin Geisler
|
r10973 | - New files added to the repository (with :hg:`add`) | ||
Matt Mackall
|
r6573 | |||
If directories are given on the command line, only files in these | ||||
directories are considered. | ||||
Martin Geisler
|
r9270 | 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. | ||||
Matt Mackall
|
r6573 | ''' | ||
act = not opts['print'] | ||||
Matt Mackall
|
r6757 | eol = '\n' | ||
if opts['print0']: | ||||
eol = '\0' | ||||
act = False # --print0 implies --print | ||||
Emanuele Aina
|
r4311 | |||
def remove(remove_func, name): | ||||
if act: | ||||
try: | ||||
Martin Geisler
|
r7570 | remove_func(repo.wjoin(name)) | ||
Benoit Boissinot
|
r7280 | except OSError: | ||
Matt Mackall
|
r6757 | m = _('%s cannot be removed') % name | ||
if opts['abort_on_err']: | ||||
raise util.Abort(m) | ||||
ui.warn(_('warning: %s\n') % m) | ||||
Emanuele Aina
|
r4311 | else: | ||
ui.write('%s%s' % (name, eol)) | ||||
Patrick Mezard
|
r8043 | def removefile(path): | ||
Patrick Mezard
|
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) | ||||
Patrick Mezard
|
r8043 | |||
Emanuele Aina
|
r4311 | directories = [] | ||
Matt Mackall
|
r14671 | match = scmutil.match(repo[None], dirs, opts) | ||
Matt Mackall
|
r6588 | match.dir = directories.append | ||
Matt Mackall
|
r6754 | status = repo.status(match=match, ignored=opts['all'], unknown=True) | ||
Emanuele Aina
|
r4311 | |||
Matt Mackall
|
r8209 | for f in sorted(status[4] + status[5]): | ||
Matt Mackall
|
r6754 | ui.note(_('Removing file %s\n') % f) | ||
Patrick Mezard
|
r8043 | remove(removefile, f) | ||
Emanuele Aina
|
r4311 | |||
Matt Mackall
|
r8209 | for f in sorted(directories, reverse=True): | ||
Emanuele Aina
|
r4463 | if match(f) and not os.listdir(repo.wjoin(f)): | ||
Emanuele Aina
|
r4311 | ui.note(_('Removing directory %s\n') % f) | ||
remove(os.rmdir, f) | ||||