##// END OF EJS Templates
automv: improve function docstrings
Martijn Pieters -
r28149:d356d525 default
parent child Browse files
Show More
@@ -1,81 +1,92
1 1 # automv.py
2 2 #
3 3 # Copyright 2013-2016 Facebook, Inc.
4 4 #
5 5 # This software may be used and distributed according to the terms of the
6 6 # GNU General Public License version 2 or any later version.
7 7 """Check for unrecorded moves at commit time (EXPERIMENTAL)
8 8
9 9 This extension checks at commit/amend time if any of the committed files
10 10 comes from an unrecorded mv.
11 11
12 12 The threshold at which a file is considered a move can be set with the
13 13 ``automv.similarity`` config option; the default value is 1.00.
14 14
15 15 """
16 16 from __future__ import absolute_import
17 17
18 18 from mercurial import (
19 19 commands,
20 20 copies,
21 21 extensions,
22 22 scmutil,
23 23 similar
24 24 )
25 25 from mercurial.i18n import _
26 26
27 27 def extsetup(ui):
28 28 entry = extensions.wrapcommand(
29 29 commands.table, 'commit', mvcheck)
30 30 entry[1].append(
31 31 ('', 'no-automv', None,
32 32 _('disable automatic file move detection')))
33 33
34 34 def mvcheck(orig, ui, repo, *pats, **opts):
35 """Hook to check for moves at commit time"""
35 36 disabled = opts.pop('no_automv', False)
36 37 if not disabled:
37 38 threshold = float(ui.config('automv', 'similarity', '1.00'))
38 39 if threshold > 0:
39 40 match = scmutil.match(repo[None], pats, opts)
40 41 added, removed = _interestingfiles(repo, match)
41 42 renames = _findrenames(repo, match, added, removed, threshold)
42 43 _markchanges(repo, renames)
43 44
44 45 return orig(ui, repo, *pats, **opts)
45 46
46 47 def _interestingfiles(repo, matcher):
48 """Find what files were added or removed in this commit.
49
50 Returns a tuple of two lists: (added, removed). Only files not *already*
51 marked as moved are included in the added list.
52
53 """
47 54 stat = repo.status(match=matcher)
48 55 added = stat[1]
49 56 removed = stat[2]
50 57
51 58 copy = copies._forwardcopies(repo['.'], repo[None], matcher)
52 59 # remove the copy files for which we already have copy info
53 60 added = [f for f in added if f not in copy]
54 61
55 62 return added, removed
56 63
57 64 def _findrenames(repo, matcher, added, removed, similarity):
58 """Find renames from removed files of the current commit/amend files
59 to the added ones"""
65 """Find what files in added are really moved files.
66
67 Any file named in removed that is at least similarity% similar to a file
68 in added is seen as a rename.
69
70 """
60 71 renames = {}
61 72 if similarity > 0:
62 73 for src, dst, score in similar.findrenames(
63 74 repo, added, removed, similarity):
64 75 if repo.ui.verbose:
65 76 repo.ui.status(
66 77 _('detected move of %s as %s (%d%% similar)\n') % (
67 78 matcher.rel(src), matcher.rel(dst), score * 100))
68 79 renames[dst] = src
69 80 if renames:
70 81 repo.ui.status(_('detected move of %d files\n') % len(renames))
71 82 return renames
72 83
73 84 def _markchanges(repo, renames):
74 85 """Marks the files in renames as copied."""
75 86 wctx = repo[None]
76 87 wlock = repo.wlock()
77 88 try:
78 89 for dst, src in renames.iteritems():
79 90 wctx.copy(src, dst)
80 91 finally:
81 92 wlock.release()
General Comments 0
You need to be logged in to leave comments. Login now