Mercurial > public > mercurial-scm > hg

annotate contrib/purge/purge.py @ 4150:cf26f4564000

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Merge _remove_file() and _remove_dir() in a single function
author Emanuele Aina <faina.mail@tiscali.it>
date Tue, 06 Mar 2007 17:45:44 -0300
parents e59286f15189
children 337010e50dcd
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
2364
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
1 # Copyright (C) 2006 - Marco Barisione <marco@barisione.org>
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
2 #
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
3 # This is a small extension for Mercurial (http://www.selenic.com/mercurial)
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
4 # that removes files not known to mercurial
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
5 #
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
6 # This program is free software; you can redistribute it and/or modify
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
7 # it under the terms of the GNU General Public License as published by
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
8 # the Free Software Foundation; either version 2 of the License, or
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
9 # (at your option) any later version.
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
10 #
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
11 # This program is distributed in the hope that it will be useful,
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
14 # GNU General Public License for more details.
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
15 #
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
16 # You should have received a copy of the GNU General Public License
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
17 # along with this program; if not, write to the Free Software
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
18 # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
19
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
20 from mercurial import hg, util
4121
d250076824e3 Use the mercurial i18n infrastructure in the purge extension
Emanuele Aina <em@nerd.ocracy.org>
parents: 4120
diff changeset
21 from mercurial.i18n import _
2364
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
22 import os
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
23
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
24 class Purge(object):
2378
6e5d40ec862d Removed --nothing, added --print and --print0
Marco Barisione <marco@barisione.org>
parents: 2377
diff changeset
25 def __init__(self, act=True, abort_on_err=False, eol='\n'):
2364
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
26 self._repo = None
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
27 self._ui = None
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
28 self._act = act
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
29 self._abort_on_err = abort_on_err
2378
6e5d40ec862d Removed --nothing, added --print and --print0
Marco Barisione <marco@barisione.org>
parents: 2377
diff changeset
30 self._eol = eol
2364
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
31
2377
626779aba9bb The meaning of the directories on the command line is now explained correctly
Marco Barisione <marco@barisione.org>
parents: 2376
diff changeset
32 def purge(self, ui, repo, dirs=None):
2364
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
33 self._repo = repo
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
34 self._ui = ui
4147
691f9168a815 Make the purge extension use the statwalk walker from the dirstate object
Emanuele Aina <faina.mail@tiscali.it>
parents: 4121
diff changeset
35
691f9168a815 Make the purge extension use the statwalk walker from the dirstate object
Emanuele Aina <faina.mail@tiscali.it>
parents: 4121
diff changeset
36 directories = []
691f9168a815 Make the purge extension use the statwalk walker from the dirstate object
Emanuele Aina <faina.mail@tiscali.it>
parents: 4121
diff changeset
37 files = []
691f9168a815 Make the purge extension use the statwalk walker from the dirstate object
Emanuele Aina <faina.mail@tiscali.it>
parents: 4121
diff changeset
38 for src, f, st in repo.dirstate.statwalk(files=dirs, ignored=True,
691f9168a815 Make the purge extension use the statwalk walker from the dirstate object
Emanuele Aina <faina.mail@tiscali.it>
parents: 4121
diff changeset
39 directories=True):
691f9168a815 Make the purge extension use the statwalk walker from the dirstate object
Emanuele Aina <faina.mail@tiscali.it>
parents: 4121
diff changeset
40 if src == 'd':
691f9168a815 Make the purge extension use the statwalk walker from the dirstate object
Emanuele Aina <faina.mail@tiscali.it>
parents: 4121
diff changeset
41 directories.append(f)
691f9168a815 Make the purge extension use the statwalk walker from the dirstate object
Emanuele Aina <faina.mail@tiscali.it>
parents: 4121
diff changeset
42 elif src == 'f' and f not in repo.dirstate:
691f9168a815 Make the purge extension use the statwalk walker from the dirstate object
Emanuele Aina <faina.mail@tiscali.it>
parents: 4121
diff changeset
43 files.append(f)
2364
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
44
4147
691f9168a815 Make the purge extension use the statwalk walker from the dirstate object
Emanuele Aina <faina.mail@tiscali.it>
parents: 4121
diff changeset
45 directories.sort()
691f9168a815 Make the purge extension use the statwalk walker from the dirstate object
Emanuele Aina <faina.mail@tiscali.it>
parents: 4121
diff changeset
46
691f9168a815 Make the purge extension use the statwalk walker from the dirstate object
Emanuele Aina <faina.mail@tiscali.it>
parents: 4121
diff changeset
47 for f in files:
4150
cf26f4564000 Merge _remove_file() and _remove_dir() in a single function
Emanuele Aina <faina.mail@tiscali.it>
parents: 4149
diff changeset
48 if f not in self._repo.dirstate:
cf26f4564000 Merge _remove_file() and _remove_dir() in a single function
Emanuele Aina <faina.mail@tiscali.it>
parents: 4149
diff changeset
49 self._ui.note(_('Removing file %s\n') % f)
cf26f4564000 Merge _remove_file() and _remove_dir() in a single function
Emanuele Aina <faina.mail@tiscali.it>
parents: 4149
diff changeset
50 self._remove(os.remove, f)
4147
691f9168a815 Make the purge extension use the statwalk walker from the dirstate object
Emanuele Aina <faina.mail@tiscali.it>
parents: 4121
diff changeset
51
691f9168a815 Make the purge extension use the statwalk walker from the dirstate object
Emanuele Aina <faina.mail@tiscali.it>
parents: 4121
diff changeset
52 for f in directories[::-1]:
4148
de85ff0aaac5 Reduce the absolute/relative path conversion in the purge extension
Emanuele Aina <faina.mail@tiscali.it>
parents: 4147
diff changeset
53 if not os.listdir(repo.wjoin(f)):
4150
cf26f4564000 Merge _remove_file() and _remove_dir() in a single function
Emanuele Aina <faina.mail@tiscali.it>
parents: 4149
diff changeset
54 self._ui.note(_('Removing directory %s\n') % f)
cf26f4564000 Merge _remove_file() and _remove_dir() in a single function
Emanuele Aina <faina.mail@tiscali.it>
parents: 4149
diff changeset
55 self._remove(os.rmdir, f)
2364
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
56
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
57 self._repo = None
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
58 self._ui = None
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
59
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
60 def _error(self, msg):
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
61 if self._abort_on_err:
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
62 raise util.Abort(msg)
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
63 else:
2376
52cfb9864257 Marked strings for translation
Marco Barisione <marco@barisione.org>
parents: 2375
diff changeset
64 self._ui.warn(_('warning: %s\n') % msg)
2364
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
65
4150
cf26f4564000 Merge _remove_file() and _remove_dir() in a single function
Emanuele Aina <faina.mail@tiscali.it>
parents: 4149
diff changeset
66 def _remove(self, remove_func, name):
2364
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
67 if self._act:
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
68 try:
4150
cf26f4564000 Merge _remove_file() and _remove_dir() in a single function
Emanuele Aina <faina.mail@tiscali.it>
parents: 4149
diff changeset
69 remove_func(self._repo.wjoin(name))
4148
de85ff0aaac5 Reduce the absolute/relative path conversion in the purge extension
Emanuele Aina <faina.mail@tiscali.it>
parents: 4147
diff changeset
70 except OSError, e:
de85ff0aaac5 Reduce the absolute/relative path conversion in the purge extension
Emanuele Aina <faina.mail@tiscali.it>
parents: 4147
diff changeset
71 self._error(_('%s cannot be removed') % name)
de85ff0aaac5 Reduce the absolute/relative path conversion in the purge extension
Emanuele Aina <faina.mail@tiscali.it>
parents: 4147
diff changeset
72 else:
de85ff0aaac5 Reduce the absolute/relative path conversion in the purge extension
Emanuele Aina <faina.mail@tiscali.it>
parents: 4147
diff changeset
73 self._ui.write('%s%s' % (name, self._eol))
2364
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
74
2369
9da3dd62c827 Purge.from_command is now a function called purge
demian@gaudron.lan
parents: 2364
diff changeset
75
2377
626779aba9bb The meaning of the directories on the command line is now explained correctly
Marco Barisione <marco@barisione.org>
parents: 2376
diff changeset
76 def purge(ui, repo, *dirs, **opts):
2369
9da3dd62c827 Purge.from_command is now a function called purge
demian@gaudron.lan
parents: 2364
diff changeset
77 '''removes files not tracked by mercurial
9da3dd62c827 Purge.from_command is now a function called purge
demian@gaudron.lan
parents: 2364
diff changeset
78
9da3dd62c827 Purge.from_command is now a function called purge
demian@gaudron.lan
parents: 2364
diff changeset
79 Delete files not known to mercurial, this is useful to test local and
9da3dd62c827 Purge.from_command is now a function called purge
demian@gaudron.lan
parents: 2364
diff changeset
80 uncommitted changes in the otherwise clean source tree.
9da3dd62c827 Purge.from_command is now a function called purge
demian@gaudron.lan
parents: 2364
diff changeset
81
9da3dd62c827 Purge.from_command is now a function called purge
demian@gaudron.lan
parents: 2364
diff changeset
82 This means that purge will delete:
9da3dd62c827 Purge.from_command is now a function called purge
demian@gaudron.lan
parents: 2364
diff changeset
83 - Unknown files: files marked with "?" by "hg status"
2377
626779aba9bb The meaning of the directories on the command line is now explained correctly
Marco Barisione <marco@barisione.org>
parents: 2376
diff changeset
84 - Ignored files: files usually ignored by Mercurial because they match
626779aba9bb The meaning of the directories on the command line is now explained correctly
Marco Barisione <marco@barisione.org>
parents: 2376
diff changeset
85 a pattern in a ".hgignore" file
2381
ab7a438294fc Rewritten install instructions for hg-purge to match new situation, fixed typos.
Thomas Arendsen Hein <thomas@intevation.de>
parents: 2379
diff changeset
86 - Empty directories: in fact Mercurial ignores directories unless they
2369
9da3dd62c827 Purge.from_command is now a function called purge
demian@gaudron.lan
parents: 2364
diff changeset
87 contain files under source control managment
9da3dd62c827 Purge.from_command is now a function called purge
demian@gaudron.lan
parents: 2364
diff changeset
88 But it will leave untouched:
9da3dd62c827 Purge.from_command is now a function called purge
demian@gaudron.lan
parents: 2364
diff changeset
89 - Unmodified tracked files
9da3dd62c827 Purge.from_command is now a function called purge
demian@gaudron.lan
parents: 2364
diff changeset
90 - Modified tracked files
9da3dd62c827 Purge.from_command is now a function called purge
demian@gaudron.lan
parents: 2364
diff changeset
91 - New files added to the repository (with "hg add")
2364
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
92
2377
626779aba9bb The meaning of the directories on the command line is now explained correctly
Marco Barisione <marco@barisione.org>
parents: 2376
diff changeset
93 If directories are given on the command line, only files in these
626779aba9bb The meaning of the directories on the command line is now explained correctly
Marco Barisione <marco@barisione.org>
parents: 2376
diff changeset
94 directories are considered.
2364
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
95
2369
9da3dd62c827 Purge.from_command is now a function called purge
demian@gaudron.lan
parents: 2364
diff changeset
96 Be careful with purge, you could irreversibly delete some files you
9da3dd62c827 Purge.from_command is now a function called purge
demian@gaudron.lan
parents: 2364
diff changeset
97 forgot to add to the repository. If you only want to print the list of
2381
ab7a438294fc Rewritten install instructions for hg-purge to match new situation, fixed typos.
Thomas Arendsen Hein <thomas@intevation.de>
parents: 2379
diff changeset
98 files that this program would delete use the --print option.
2369
9da3dd62c827 Purge.from_command is now a function called purge
demian@gaudron.lan
parents: 2364
diff changeset
99 '''
2378
6e5d40ec862d Removed --nothing, added --print and --print0
Marco Barisione <marco@barisione.org>
parents: 2377
diff changeset
100 act = not opts['print']
2370
de893ad6bd17 Command line options are read in a saner way
demian@gaudron.lan
parents: 2369
diff changeset
101 abort_on_err = bool(opts['abort_on_err'])
2378
6e5d40ec862d Removed --nothing, added --print and --print0
Marco Barisione <marco@barisione.org>
parents: 2377
diff changeset
102 eol = opts['print0'] and '\0' or '\n'
6e5d40ec862d Removed --nothing, added --print and --print0
Marco Barisione <marco@barisione.org>
parents: 2377
diff changeset
103 if eol == '\0':
6e5d40ec862d Removed --nothing, added --print and --print0
Marco Barisione <marco@barisione.org>
parents: 2377
diff changeset
104 # --print0 implies --print
6e5d40ec862d Removed --nothing, added --print and --print0
Marco Barisione <marco@barisione.org>
parents: 2377
diff changeset
105 act = False
6e5d40ec862d Removed --nothing, added --print and --print0
Marco Barisione <marco@barisione.org>
parents: 2377
diff changeset
106 p = Purge(act, abort_on_err, eol)
2377
626779aba9bb The meaning of the directories on the command line is now explained correctly
Marco Barisione <marco@barisione.org>
parents: 2376
diff changeset
107 p.purge(ui, repo, dirs)
2364
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
108
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
109
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
110 cmdtable = {
2382
b429566d1994 Make indentation of purge's cmdtable match to mercurial/commands.py
Thomas Arendsen Hein <thomas@intevation.de>
parents: 2381
diff changeset
111 'purge':
b429566d1994 Make indentation of purge's cmdtable match to mercurial/commands.py
Thomas Arendsen Hein <thomas@intevation.de>
parents: 2381
diff changeset
112 (purge,
b429566d1994 Make indentation of purge's cmdtable match to mercurial/commands.py
Thomas Arendsen Hein <thomas@intevation.de>
parents: 2381
diff changeset
113 [('a', 'abort-on-err', None, _('abort if an error occurs')),
b429566d1994 Make indentation of purge's cmdtable match to mercurial/commands.py
Thomas Arendsen Hein <thomas@intevation.de>
parents: 2381
diff changeset
114 ('p', 'print', None, _('print the file names instead of deleting them')),
b429566d1994 Make indentation of purge's cmdtable match to mercurial/commands.py
Thomas Arendsen Hein <thomas@intevation.de>
parents: 2381
diff changeset
115 ('0', 'print0', None, _('end filenames with NUL, for use with xargs'
b429566d1994 Make indentation of purge's cmdtable match to mercurial/commands.py
Thomas Arendsen Hein <thomas@intevation.de>
parents: 2381
diff changeset
116 ' (implies -p)'))],
b429566d1994 Make indentation of purge's cmdtable match to mercurial/commands.py
Thomas Arendsen Hein <thomas@intevation.de>
parents: 2381
diff changeset
117 _('hg purge [OPTION]... [DIR]...'))
2364
f368a1c302d5 Initial commit
demian@gaudron.lan
parents:
diff changeset
118 }