purge.py
119 lines
| 4.4 KiB
| text/x-python
|
PythonLexer
/ hgext / purge.py
|
r4311 | # Copyright (C) 2006 - Marco Barisione <marco@barisione.org> | ||
# | ||||
|
r26421 | # This is a small extension for Mercurial (https://mercurial-scm.org/) | ||
|
r4311 | # that removes files not known to mercurial | ||
# | ||||
|
r9270 | # This program was inspired by the "cvspurge" script contained in CVS | ||
# utilities (http://www.red-bean.com/cvsutils/). | ||||
|
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 | ||||
|
r15782 | # along with this program; if not, see <http://www.gnu.org/licenses/>. | ||
|
r4311 | |||
|
r8934 | '''command to delete untracked files from the working directory''' | ||
|
r8873 | |||
|
r14322 | from mercurial import util, commands, cmdutil, scmutil | ||
|
r4311 | from mercurial.i18n import _ | ||
|
r21994 | import os | ||
|
r4311 | |||
|
r14310 | cmdtable = {} | ||
command = cmdutil.command(cmdtable) | ||||
|
r25186 | # Note for extension authors: ONLY specify testedwith = 'internal' for | ||
# extensions which SHIP WITH MERCURIAL. Non-mainline extensions should | ||||
# be specifying the version(s) of Mercurial they are tested with, or | ||||
# leave the attribute unspecified. | ||||
|
r16743 | testedwith = 'internal' | ||
|
r14310 | |||
@command('purge|clean', | ||||
[('a', 'abort-on-err', None, _('abort if an error occurs')), | ||||
('', 'all', None, _('purge ignored files too')), | ||||
|
r21853 | ('', 'dirs', None, _('purge empty directories')), | ||
('', 'files', None, _('purge files')), | ||||
|
r14310 | ('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]...')) | ||||
|
r6573 | def purge(ui, repo, *dirs, **opts): | ||
|
r7605 | '''removes files not tracked by Mercurial | ||
|
r6573 | |||
|
r9270 | Delete files not known to Mercurial. This is useful to test local | ||
and uncommitted changes in an otherwise-clean source tree. | ||||
|
r6573 | |||
|
r21853 | This means that purge will delete the following by default: | ||
|
r9215 | |||
|
r10973 | - Unknown files: files marked with "?" by :hg:`status` | ||
|
r9270 | - Empty directories: in fact Mercurial ignores directories unless | ||
they contain files under source control management | ||||
|
r9215 | |||
|
r6573 | But it will leave untouched: | ||
|
r9215 | |||
- Modified and unmodified tracked files | ||||
- Ignored files (unless --all is specified) | ||||
|
r10973 | - New files added to the repository (with :hg:`add`) | ||
|
r6573 | |||
|
r21853 | The --files and --dirs options can be used to direct purge to delete | ||
only files, only directories, or both. If neither option is given, | ||||
both will be deleted. | ||||
|
r6573 | If directories are given on the command line, only files in these | ||
directories are considered. | ||||
|
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. | ||||
|
r6573 | ''' | ||
act = not opts['print'] | ||||
|
r6757 | eol = '\n' | ||
if opts['print0']: | ||||
eol = '\0' | ||||
act = False # --print0 implies --print | ||||
|
r21853 | removefiles = opts['files'] | ||
removedirs = opts['dirs'] | ||||
if not removefiles and not removedirs: | ||||
removefiles = True | ||||
removedirs = True | ||||
|
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)) | ||||
|
r14671 | match = scmutil.match(repo[None], dirs, opts) | ||
|
r22265 | if removedirs: | ||
directories = [] | ||||
match.explicitdir = match.traversedir = directories.append | ||||
|
r6754 | status = repo.status(match=match, ignored=opts['all'], unknown=True) | ||
|
r4311 | |||
|
r21853 | if removefiles: | ||
|
r22920 | for f in sorted(status.unknown + status.ignored): | ||
|
r21853 | if act: | ||
ui.note(_('removing file %s\n') % f) | ||||
|
r21983 | remove(util.unlink, f) | ||
|
r4311 | |||
|
r21853 | if removedirs: | ||
for f in sorted(directories, reverse=True): | ||||
if match(f) and not os.listdir(repo.wjoin(f)): | ||||
if act: | ||||
ui.note(_('removing directory %s\n') % f) | ||||
remove(os.rmdir, f) | ||||