Kiln » TortoiseHg » TortoiseHg
Clone URL:  
cachethg.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
import os from mercurial import hg, cmdutil, util, ui, node, merge import thgutil import sys try: from mercurial.error import RepoError except ImportError: from mercurial.repo import RepoError try: from win32api import GetTickCount CACHE_TIMEOUT = 5000 except ImportError: from time import time as GetTickCount CACHE_TIMEOUT = 5.0 STATUS_STATES = 'MAR!?IC' MODIFIED, ADDED, REMOVED, DELETED, UNKNOWN, IGNORED, UNCHANGED = STATUS_STATES NOT_IN_REPO = ' ' ROOT = "r" UNRESOLVED = 'U' # file status cache overlay_cache = {} cache_tick_count = 0 cache_root = None cache_pdir = None def add_dirs(list): dirs = set() if list: dirs.add('') for f in list: pdir = os.path.dirname(f) if pdir in dirs: continue while pdir: dirs.add(pdir) pdir = os.path.dirname(pdir) list.extend(dirs) def get_state(upath, repo=None): """ Get the state of a given path in source control. """ states = get_states(upath, repo) return states and states[0] or NOT_IN_REPO def get_states(upath, repo=None): """ Get the states of a given path in source control. """ global overlay_cache, cache_tick_count global cache_root, cache_pdir #print "called: _get_state(%s)" % path tc = GetTickCount() try: # handle some Asian charsets path = upath.encode('mbcs') except: path = upath # check if path is cached pdir = os.path.dirname(path) if cache_pdir == pdir and overlay_cache: if tc - cache_tick_count < CACHE_TIMEOUT: status = overlay_cache.get(path) if not status: if os.path.isdir(os.path.join(path, '.hg')): add(path, ROOT) status = ROOT else: status = overlay_cache.get(pdir, NOT_IN_REPO) print "%s: %s (cached)" % (path, status) return status else: print "Timed out!! ", overlay_cache.clear() cache_tick_count = GetTickCount() # path is a drive if path.endswith(":\\"): add(path, NOT_IN_REPO) return NOT_IN_REPO # open repo if cache_pdir == pdir: root = cache_root else: print "find new root ", root = thgutil.find_root(path) if root == path: add(path, ROOT) return ROOT cache_root = root cache_pdir = pdir print "_get_state: root = ", root if root is None: print "_get_state: not in repo" overlay_cache = {None: None} cache_tick_count = GetTickCount() return NOT_IN_REPO hgdir = os.path.join(root, '.hg', '') if pdir == hgdir[:-1] or pdir.startswith(hgdir): add(pdir, NOT_IN_REPO) return NOT_IN_REPO try: tc1 = GetTickCount() if not repo or (repo.root != root and repo.root != os.path.realpath(root)): repo = hg.repository(ui.ui(), path=root) print "hg.repository() took %g ticks" % (GetTickCount() - tc1) # check if to display overlay icons in this repo overlayopt = repo.ui.config('tortoisehg', 'overlayicons', ' ').lower() print "%s: repo overlayicons = " % path, overlayopt if overlayopt == 'localdisk': overlayopt = bool(thgutil.netdrive_status(path)) if not overlayopt or overlayopt in 'false off no'.split(): print "%s: overlayicons disabled" % path overlay_cache = {None: None} cache_tick_count = GetTickCount() return NOT_IN_REPO except RepoError: # We aren't in a working tree print "%s: not in repo" % pdir add(pdir, IGNORED) return IGNORED except StandardError, e: print "error while handling %s:" % pdir print e add(pdir, UNKNOWN) return UNKNOWN # get file status tc1 = GetTickCount() try: matcher = cmdutil.match(repo, [pdir]) repostate = repo.status(match=matcher, ignored=True, clean=True, unknown=True) except util.Abort, inst: print "abort: %s" % inst print "treat as unknown : %s" % path return UNKNOWN print "status() took %g ticks" % (GetTickCount() - tc1) mergestate = repo.dirstate.parents()[1] != node.nullid and \ hasattr(merge, 'mergestate') # cached file info tc = GetTickCount() overlay_cache = {} add(root, ROOT) states = STATUS_STATES if mergestate: mstate = merge.mergestate(repo) unresolved = [f for f in mstate if mstate[f] == 'u'] if unresolved: modified = repostate[0] modified[:] = set(modified) - set(unresolved) repostate.insert(0, unresolved) states = [UNRESOLVED] + states states = zip(repostate, states) states[-1], states[-2] = states[-2], states[-1] #clean before ignored for grp, st in states: add_dirs(grp) for f in grp: fpath = os.path.join(root, os.path.normpath(f)) add(fpath, st) status = overlay_cache.get(path, UNKNOWN) print "%s: %s" % (path, status) cache_tick_count = GetTickCount() return status def add(path, state): overlay_cache[path] = overlay_cache.get(path, '') + state