##// END OF EJS Templates
workingfilectx.cmp: invert boolean return value...
Nicolas Dumazet -
r11538:16fe9880 stable
parent child Browse files
Show More
@@ -1,1078 +1,1078 b''
1 1 # context.py - changeset and file context objects for mercurial
2 2 #
3 3 # Copyright 2006, 2007 Matt Mackall <mpm@selenic.com>
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
8 8 from node import nullid, nullrev, short, hex
9 9 from i18n import _
10 10 import ancestor, bdiff, error, util, subrepo, patch
11 11 import os, errno, stat
12 12
13 13 propertycache = util.propertycache
14 14
15 15 class changectx(object):
16 16 """A changecontext object makes access to data related to a particular
17 17 changeset convenient."""
18 18 def __init__(self, repo, changeid=''):
19 19 """changeid is a revision number, node, or tag"""
20 20 if changeid == '':
21 21 changeid = '.'
22 22 self._repo = repo
23 23 if isinstance(changeid, (long, int)):
24 24 self._rev = changeid
25 25 self._node = self._repo.changelog.node(changeid)
26 26 else:
27 27 self._node = self._repo.lookup(changeid)
28 28 self._rev = self._repo.changelog.rev(self._node)
29 29
30 30 def __str__(self):
31 31 return short(self.node())
32 32
33 33 def __int__(self):
34 34 return self.rev()
35 35
36 36 def __repr__(self):
37 37 return "<changectx %s>" % str(self)
38 38
39 39 def __hash__(self):
40 40 try:
41 41 return hash(self._rev)
42 42 except AttributeError:
43 43 return id(self)
44 44
45 45 def __eq__(self, other):
46 46 try:
47 47 return self._rev == other._rev
48 48 except AttributeError:
49 49 return False
50 50
51 51 def __ne__(self, other):
52 52 return not (self == other)
53 53
54 54 def __nonzero__(self):
55 55 return self._rev != nullrev
56 56
57 57 @propertycache
58 58 def _changeset(self):
59 59 return self._repo.changelog.read(self.node())
60 60
61 61 @propertycache
62 62 def _manifest(self):
63 63 return self._repo.manifest.read(self._changeset[0])
64 64
65 65 @propertycache
66 66 def _manifestdelta(self):
67 67 return self._repo.manifest.readdelta(self._changeset[0])
68 68
69 69 @propertycache
70 70 def _parents(self):
71 71 p = self._repo.changelog.parentrevs(self._rev)
72 72 if p[1] == nullrev:
73 73 p = p[:-1]
74 74 return [changectx(self._repo, x) for x in p]
75 75
76 76 @propertycache
77 77 def substate(self):
78 78 return subrepo.state(self)
79 79
80 80 def __contains__(self, key):
81 81 return key in self._manifest
82 82
83 83 def __getitem__(self, key):
84 84 return self.filectx(key)
85 85
86 86 def __iter__(self):
87 87 for f in sorted(self._manifest):
88 88 yield f
89 89
90 90 def changeset(self):
91 91 return self._changeset
92 92 def manifest(self):
93 93 return self._manifest
94 94 def manifestnode(self):
95 95 return self._changeset[0]
96 96
97 97 def rev(self):
98 98 return self._rev
99 99 def node(self):
100 100 return self._node
101 101 def hex(self):
102 102 return hex(self._node)
103 103 def user(self):
104 104 return self._changeset[1]
105 105 def date(self):
106 106 return self._changeset[2]
107 107 def files(self):
108 108 return self._changeset[3]
109 109 def description(self):
110 110 return self._changeset[4]
111 111 def branch(self):
112 112 return self._changeset[5].get("branch")
113 113 def extra(self):
114 114 return self._changeset[5]
115 115 def tags(self):
116 116 return self._repo.nodetags(self._node)
117 117
118 118 def parents(self):
119 119 """return contexts for each parent changeset"""
120 120 return self._parents
121 121
122 122 def p1(self):
123 123 return self._parents[0]
124 124
125 125 def p2(self):
126 126 if len(self._parents) == 2:
127 127 return self._parents[1]
128 128 return changectx(self._repo, -1)
129 129
130 130 def children(self):
131 131 """return contexts for each child changeset"""
132 132 c = self._repo.changelog.children(self._node)
133 133 return [changectx(self._repo, x) for x in c]
134 134
135 135 def ancestors(self):
136 136 for a in self._repo.changelog.ancestors(self._rev):
137 137 yield changectx(self._repo, a)
138 138
139 139 def descendants(self):
140 140 for d in self._repo.changelog.descendants(self._rev):
141 141 yield changectx(self._repo, d)
142 142
143 143 def _fileinfo(self, path):
144 144 if '_manifest' in self.__dict__:
145 145 try:
146 146 return self._manifest[path], self._manifest.flags(path)
147 147 except KeyError:
148 148 raise error.LookupError(self._node, path,
149 149 _('not found in manifest'))
150 150 if '_manifestdelta' in self.__dict__ or path in self.files():
151 151 if path in self._manifestdelta:
152 152 return self._manifestdelta[path], self._manifestdelta.flags(path)
153 153 node, flag = self._repo.manifest.find(self._changeset[0], path)
154 154 if not node:
155 155 raise error.LookupError(self._node, path,
156 156 _('not found in manifest'))
157 157
158 158 return node, flag
159 159
160 160 def filenode(self, path):
161 161 return self._fileinfo(path)[0]
162 162
163 163 def flags(self, path):
164 164 try:
165 165 return self._fileinfo(path)[1]
166 166 except error.LookupError:
167 167 return ''
168 168
169 169 def filectx(self, path, fileid=None, filelog=None):
170 170 """get a file context from this changeset"""
171 171 if fileid is None:
172 172 fileid = self.filenode(path)
173 173 return filectx(self._repo, path, fileid=fileid,
174 174 changectx=self, filelog=filelog)
175 175
176 176 def ancestor(self, c2):
177 177 """
178 178 return the ancestor context of self and c2
179 179 """
180 180 # deal with workingctxs
181 181 n2 = c2._node
182 182 if n2 == None:
183 183 n2 = c2._parents[0]._node
184 184 n = self._repo.changelog.ancestor(self._node, n2)
185 185 return changectx(self._repo, n)
186 186
187 187 def walk(self, match):
188 188 fset = set(match.files())
189 189 # for dirstate.walk, files=['.'] means "walk the whole tree".
190 190 # follow that here, too
191 191 fset.discard('.')
192 192 for fn in self:
193 193 for ffn in fset:
194 194 # match if the file is the exact name or a directory
195 195 if ffn == fn or fn.startswith("%s/" % ffn):
196 196 fset.remove(ffn)
197 197 break
198 198 if match(fn):
199 199 yield fn
200 200 for fn in sorted(fset):
201 201 if match.bad(fn, 'No such file in rev ' + str(self)) and match(fn):
202 202 yield fn
203 203
204 204 def sub(self, path):
205 205 return subrepo.subrepo(self, path)
206 206
207 207 def diff(self, ctx2=None, match=None, **opts):
208 208 """Returns a diff generator for the given contexts and matcher"""
209 209 if ctx2 is None:
210 210 ctx2 = self.p1()
211 211 if ctx2 is not None and not isinstance(ctx2, changectx):
212 212 ctx2 = self._repo[ctx2]
213 213 diffopts = patch.diffopts(self._repo.ui, opts)
214 214 return patch.diff(self._repo, ctx2.node(), self.node(),
215 215 match=match, opts=diffopts)
216 216
217 217 class filectx(object):
218 218 """A filecontext object makes access to data related to a particular
219 219 filerevision convenient."""
220 220 def __init__(self, repo, path, changeid=None, fileid=None,
221 221 filelog=None, changectx=None):
222 222 """changeid can be a changeset revision, node, or tag.
223 223 fileid can be a file revision or node."""
224 224 self._repo = repo
225 225 self._path = path
226 226
227 227 assert (changeid is not None
228 228 or fileid is not None
229 229 or changectx is not None), \
230 230 ("bad args: changeid=%r, fileid=%r, changectx=%r"
231 231 % (changeid, fileid, changectx))
232 232
233 233 if filelog:
234 234 self._filelog = filelog
235 235
236 236 if changeid is not None:
237 237 self._changeid = changeid
238 238 if changectx is not None:
239 239 self._changectx = changectx
240 240 if fileid is not None:
241 241 self._fileid = fileid
242 242
243 243 @propertycache
244 244 def _changectx(self):
245 245 return changectx(self._repo, self._changeid)
246 246
247 247 @propertycache
248 248 def _filelog(self):
249 249 return self._repo.file(self._path)
250 250
251 251 @propertycache
252 252 def _changeid(self):
253 253 if '_changectx' in self.__dict__:
254 254 return self._changectx.rev()
255 255 else:
256 256 return self._filelog.linkrev(self._filerev)
257 257
258 258 @propertycache
259 259 def _filenode(self):
260 260 if '_fileid' in self.__dict__:
261 261 return self._filelog.lookup(self._fileid)
262 262 else:
263 263 return self._changectx.filenode(self._path)
264 264
265 265 @propertycache
266 266 def _filerev(self):
267 267 return self._filelog.rev(self._filenode)
268 268
269 269 @propertycache
270 270 def _repopath(self):
271 271 return self._path
272 272
273 273 def __nonzero__(self):
274 274 try:
275 275 self._filenode
276 276 return True
277 277 except error.LookupError:
278 278 # file is missing
279 279 return False
280 280
281 281 def __str__(self):
282 282 return "%s@%s" % (self.path(), short(self.node()))
283 283
284 284 def __repr__(self):
285 285 return "<filectx %s>" % str(self)
286 286
287 287 def __hash__(self):
288 288 try:
289 289 return hash((self._path, self._filenode))
290 290 except AttributeError:
291 291 return id(self)
292 292
293 293 def __eq__(self, other):
294 294 try:
295 295 return (self._path == other._path
296 296 and self._filenode == other._filenode)
297 297 except AttributeError:
298 298 return False
299 299
300 300 def __ne__(self, other):
301 301 return not (self == other)
302 302
303 303 def filectx(self, fileid):
304 304 '''opens an arbitrary revision of the file without
305 305 opening a new filelog'''
306 306 return filectx(self._repo, self._path, fileid=fileid,
307 307 filelog=self._filelog)
308 308
309 309 def filerev(self):
310 310 return self._filerev
311 311 def filenode(self):
312 312 return self._filenode
313 313 def flags(self):
314 314 return self._changectx.flags(self._path)
315 315 def filelog(self):
316 316 return self._filelog
317 317
318 318 def rev(self):
319 319 if '_changectx' in self.__dict__:
320 320 return self._changectx.rev()
321 321 if '_changeid' in self.__dict__:
322 322 return self._changectx.rev()
323 323 return self._filelog.linkrev(self._filerev)
324 324
325 325 def linkrev(self):
326 326 return self._filelog.linkrev(self._filerev)
327 327 def node(self):
328 328 return self._changectx.node()
329 329 def hex(self):
330 330 return hex(self.node())
331 331 def user(self):
332 332 return self._changectx.user()
333 333 def date(self):
334 334 return self._changectx.date()
335 335 def files(self):
336 336 return self._changectx.files()
337 337 def description(self):
338 338 return self._changectx.description()
339 339 def branch(self):
340 340 return self._changectx.branch()
341 341 def extra(self):
342 342 return self._changectx.extra()
343 343 def manifest(self):
344 344 return self._changectx.manifest()
345 345 def changectx(self):
346 346 return self._changectx
347 347
348 348 def data(self):
349 349 return self._filelog.read(self._filenode)
350 350 def path(self):
351 351 return self._path
352 352 def size(self):
353 353 return self._filelog.size(self._filerev)
354 354
355 355 def cmp(self, text):
356 356 return self._filelog.cmp(self._filenode, text)
357 357
358 358 def renamed(self):
359 359 """check if file was actually renamed in this changeset revision
360 360
361 361 If rename logged in file revision, we report copy for changeset only
362 362 if file revisions linkrev points back to the changeset in question
363 363 or both changeset parents contain different file revisions.
364 364 """
365 365
366 366 renamed = self._filelog.renamed(self._filenode)
367 367 if not renamed:
368 368 return renamed
369 369
370 370 if self.rev() == self.linkrev():
371 371 return renamed
372 372
373 373 name = self.path()
374 374 fnode = self._filenode
375 375 for p in self._changectx.parents():
376 376 try:
377 377 if fnode == p.filenode(name):
378 378 return None
379 379 except error.LookupError:
380 380 pass
381 381 return renamed
382 382
383 383 def parents(self):
384 384 p = self._path
385 385 fl = self._filelog
386 386 pl = [(p, n, fl) for n in self._filelog.parents(self._filenode)]
387 387
388 388 r = self._filelog.renamed(self._filenode)
389 389 if r:
390 390 pl[0] = (r[0], r[1], None)
391 391
392 392 return [filectx(self._repo, p, fileid=n, filelog=l)
393 393 for p, n, l in pl if n != nullid]
394 394
395 395 def children(self):
396 396 # hard for renames
397 397 c = self._filelog.children(self._filenode)
398 398 return [filectx(self._repo, self._path, fileid=x,
399 399 filelog=self._filelog) for x in c]
400 400
401 401 def annotate(self, follow=False, linenumber=None):
402 402 '''returns a list of tuples of (ctx, line) for each line
403 403 in the file, where ctx is the filectx of the node where
404 404 that line was last changed.
405 405 This returns tuples of ((ctx, linenumber), line) for each line,
406 406 if "linenumber" parameter is NOT "None".
407 407 In such tuples, linenumber means one at the first appearance
408 408 in the managed file.
409 409 To reduce annotation cost,
410 410 this returns fixed value(False is used) as linenumber,
411 411 if "linenumber" parameter is "False".'''
412 412
413 413 def decorate_compat(text, rev):
414 414 return ([rev] * len(text.splitlines()), text)
415 415
416 416 def without_linenumber(text, rev):
417 417 return ([(rev, False)] * len(text.splitlines()), text)
418 418
419 419 def with_linenumber(text, rev):
420 420 size = len(text.splitlines())
421 421 return ([(rev, i) for i in xrange(1, size + 1)], text)
422 422
423 423 decorate = (((linenumber is None) and decorate_compat) or
424 424 (linenumber and with_linenumber) or
425 425 without_linenumber)
426 426
427 427 def pair(parent, child):
428 428 for a1, a2, b1, b2 in bdiff.blocks(parent[1], child[1]):
429 429 child[0][b1:b2] = parent[0][a1:a2]
430 430 return child
431 431
432 432 getlog = util.lrucachefunc(lambda x: self._repo.file(x))
433 433 def getctx(path, fileid):
434 434 log = path == self._path and self._filelog or getlog(path)
435 435 return filectx(self._repo, path, fileid=fileid, filelog=log)
436 436 getctx = util.lrucachefunc(getctx)
437 437
438 438 def parents(f):
439 439 # we want to reuse filectx objects as much as possible
440 440 p = f._path
441 441 if f._filerev is None: # working dir
442 442 pl = [(n.path(), n.filerev()) for n in f.parents()]
443 443 else:
444 444 pl = [(p, n) for n in f._filelog.parentrevs(f._filerev)]
445 445
446 446 if follow:
447 447 r = f.renamed()
448 448 if r:
449 449 pl[0] = (r[0], getlog(r[0]).rev(r[1]))
450 450
451 451 return [getctx(p, n) for p, n in pl if n != nullrev]
452 452
453 453 # use linkrev to find the first changeset where self appeared
454 454 if self.rev() != self.linkrev():
455 455 base = self.filectx(self.filerev())
456 456 else:
457 457 base = self
458 458
459 459 # find all ancestors
460 460 needed = {base: 1}
461 461 visit = [base]
462 462 files = [base._path]
463 463 while visit:
464 464 f = visit.pop(0)
465 465 for p in parents(f):
466 466 if p not in needed:
467 467 needed[p] = 1
468 468 visit.append(p)
469 469 if p._path not in files:
470 470 files.append(p._path)
471 471 else:
472 472 # count how many times we'll use this
473 473 needed[p] += 1
474 474
475 475 # sort by revision (per file) which is a topological order
476 476 visit = []
477 477 for f in files:
478 478 visit.extend(n for n in needed if n._path == f)
479 479
480 480 hist = {}
481 481 for f in sorted(visit, key=lambda x: x.rev()):
482 482 curr = decorate(f.data(), f)
483 483 for p in parents(f):
484 484 curr = pair(hist[p], curr)
485 485 # trim the history of unneeded revs
486 486 needed[p] -= 1
487 487 if not needed[p]:
488 488 del hist[p]
489 489 hist[f] = curr
490 490
491 491 return zip(hist[f][0], hist[f][1].splitlines(True))
492 492
493 493 def ancestor(self, fc2, actx=None):
494 494 """
495 495 find the common ancestor file context, if any, of self, and fc2
496 496
497 497 If actx is given, it must be the changectx of the common ancestor
498 498 of self's and fc2's respective changesets.
499 499 """
500 500
501 501 if actx is None:
502 502 actx = self.changectx().ancestor(fc2.changectx())
503 503
504 504 # the trivial case: changesets are unrelated, files must be too
505 505 if not actx:
506 506 return None
507 507
508 508 # the easy case: no (relevant) renames
509 509 if fc2.path() == self.path() and self.path() in actx:
510 510 return actx[self.path()]
511 511 acache = {}
512 512
513 513 # prime the ancestor cache for the working directory
514 514 for c in (self, fc2):
515 515 if c._filerev is None:
516 516 pl = [(n.path(), n.filenode()) for n in c.parents()]
517 517 acache[(c._path, None)] = pl
518 518
519 519 flcache = {self._repopath:self._filelog, fc2._repopath:fc2._filelog}
520 520 def parents(vertex):
521 521 if vertex in acache:
522 522 return acache[vertex]
523 523 f, n = vertex
524 524 if f not in flcache:
525 525 flcache[f] = self._repo.file(f)
526 526 fl = flcache[f]
527 527 pl = [(f, p) for p in fl.parents(n) if p != nullid]
528 528 re = fl.renamed(n)
529 529 if re:
530 530 pl.append(re)
531 531 acache[vertex] = pl
532 532 return pl
533 533
534 534 a, b = (self._path, self._filenode), (fc2._path, fc2._filenode)
535 535 v = ancestor.ancestor(a, b, parents)
536 536 if v:
537 537 f, n = v
538 538 return filectx(self._repo, f, fileid=n, filelog=flcache[f])
539 539
540 540 return None
541 541
542 542 def ancestors(self):
543 543 seen = set(str(self))
544 544 visit = [self]
545 545 while visit:
546 546 for parent in visit.pop(0).parents():
547 547 s = str(parent)
548 548 if s not in seen:
549 549 visit.append(parent)
550 550 seen.add(s)
551 551 yield parent
552 552
553 553 class workingctx(changectx):
554 554 """A workingctx object makes access to data related to
555 555 the current working directory convenient.
556 556 date - any valid date string or (unixtime, offset), or None.
557 557 user - username string, or None.
558 558 extra - a dictionary of extra values, or None.
559 559 changes - a list of file lists as returned by localrepo.status()
560 560 or None to use the repository status.
561 561 """
562 562 def __init__(self, repo, text="", user=None, date=None, extra=None,
563 563 changes=None):
564 564 self._repo = repo
565 565 self._rev = None
566 566 self._node = None
567 567 self._text = text
568 568 if date:
569 569 self._date = util.parsedate(date)
570 570 if user:
571 571 self._user = user
572 572 if changes:
573 573 self._status = list(changes[:4])
574 574 self._unknown = changes[4]
575 575 self._ignored = changes[5]
576 576 self._clean = changes[6]
577 577 else:
578 578 self._unknown = None
579 579 self._ignored = None
580 580 self._clean = None
581 581
582 582 self._extra = {}
583 583 if extra:
584 584 self._extra = extra.copy()
585 585 if 'branch' not in self._extra:
586 586 branch = self._repo.dirstate.branch()
587 587 try:
588 588 branch = branch.decode('UTF-8').encode('UTF-8')
589 589 except UnicodeDecodeError:
590 590 raise util.Abort(_('branch name not in UTF-8!'))
591 591 self._extra['branch'] = branch
592 592 if self._extra['branch'] == '':
593 593 self._extra['branch'] = 'default'
594 594
595 595 def __str__(self):
596 596 return str(self._parents[0]) + "+"
597 597
598 598 def __nonzero__(self):
599 599 return True
600 600
601 601 def __contains__(self, key):
602 602 return self._repo.dirstate[key] not in "?r"
603 603
604 604 @propertycache
605 605 def _manifest(self):
606 606 """generate a manifest corresponding to the working directory"""
607 607
608 608 if self._unknown is None:
609 609 self.status(unknown=True)
610 610
611 611 man = self._parents[0].manifest().copy()
612 612 copied = self._repo.dirstate.copies()
613 613 if len(self._parents) > 1:
614 614 man2 = self.p2().manifest()
615 615 def getman(f):
616 616 if f in man:
617 617 return man
618 618 return man2
619 619 else:
620 620 getman = lambda f: man
621 621 def cf(f):
622 622 f = copied.get(f, f)
623 623 return getman(f).flags(f)
624 624 ff = self._repo.dirstate.flagfunc(cf)
625 625 modified, added, removed, deleted = self._status
626 626 unknown = self._unknown
627 627 for i, l in (("a", added), ("m", modified), ("u", unknown)):
628 628 for f in l:
629 629 orig = copied.get(f, f)
630 630 man[f] = getman(orig).get(orig, nullid) + i
631 631 try:
632 632 man.set(f, ff(f))
633 633 except OSError:
634 634 pass
635 635
636 636 for f in deleted + removed:
637 637 if f in man:
638 638 del man[f]
639 639
640 640 return man
641 641
642 642 @propertycache
643 643 def _status(self):
644 644 return self._repo.status()[:4]
645 645
646 646 @propertycache
647 647 def _user(self):
648 648 return self._repo.ui.username()
649 649
650 650 @propertycache
651 651 def _date(self):
652 652 return util.makedate()
653 653
654 654 @propertycache
655 655 def _parents(self):
656 656 p = self._repo.dirstate.parents()
657 657 if p[1] == nullid:
658 658 p = p[:-1]
659 659 self._parents = [changectx(self._repo, x) for x in p]
660 660 return self._parents
661 661
662 662 def status(self, ignored=False, clean=False, unknown=False):
663 663 """Explicit status query
664 664 Unless this method is used to query the working copy status, the
665 665 _status property will implicitly read the status using its default
666 666 arguments."""
667 667 stat = self._repo.status(ignored=ignored, clean=clean, unknown=unknown)
668 668 self._unknown = self._ignored = self._clean = None
669 669 if unknown:
670 670 self._unknown = stat[4]
671 671 if ignored:
672 672 self._ignored = stat[5]
673 673 if clean:
674 674 self._clean = stat[6]
675 675 self._status = stat[:4]
676 676 return stat
677 677
678 678 def manifest(self):
679 679 return self._manifest
680 680 def user(self):
681 681 return self._user or self._repo.ui.username()
682 682 def date(self):
683 683 return self._date
684 684 def description(self):
685 685 return self._text
686 686 def files(self):
687 687 return sorted(self._status[0] + self._status[1] + self._status[2])
688 688
689 689 def modified(self):
690 690 return self._status[0]
691 691 def added(self):
692 692 return self._status[1]
693 693 def removed(self):
694 694 return self._status[2]
695 695 def deleted(self):
696 696 return self._status[3]
697 697 def unknown(self):
698 698 assert self._unknown is not None # must call status first
699 699 return self._unknown
700 700 def ignored(self):
701 701 assert self._ignored is not None # must call status first
702 702 return self._ignored
703 703 def clean(self):
704 704 assert self._clean is not None # must call status first
705 705 return self._clean
706 706 def branch(self):
707 707 return self._extra['branch']
708 708 def extra(self):
709 709 return self._extra
710 710
711 711 def tags(self):
712 712 t = []
713 713 [t.extend(p.tags()) for p in self.parents()]
714 714 return t
715 715
716 716 def children(self):
717 717 return []
718 718
719 719 def flags(self, path):
720 720 if '_manifest' in self.__dict__:
721 721 try:
722 722 return self._manifest.flags(path)
723 723 except KeyError:
724 724 return ''
725 725
726 726 orig = self._repo.dirstate.copies().get(path, path)
727 727
728 728 def findflag(ctx):
729 729 mnode = ctx.changeset()[0]
730 730 node, flag = self._repo.manifest.find(mnode, orig)
731 731 ff = self._repo.dirstate.flagfunc(lambda x: flag or '')
732 732 try:
733 733 return ff(path)
734 734 except OSError:
735 735 pass
736 736
737 737 flag = findflag(self._parents[0])
738 738 if flag is None and len(self.parents()) > 1:
739 739 flag = findflag(self._parents[1])
740 740 if flag is None or self._repo.dirstate[path] == 'r':
741 741 return ''
742 742 return flag
743 743
744 744 def filectx(self, path, filelog=None):
745 745 """get a file context from the working directory"""
746 746 return workingfilectx(self._repo, path, workingctx=self,
747 747 filelog=filelog)
748 748
749 749 def ancestor(self, c2):
750 750 """return the ancestor context of self and c2"""
751 751 return self._parents[0].ancestor(c2) # punt on two parents for now
752 752
753 753 def walk(self, match):
754 754 return sorted(self._repo.dirstate.walk(match, self.substate.keys(),
755 755 True, False))
756 756
757 757 def dirty(self, missing=False):
758 758 "check whether a working directory is modified"
759 759 # check subrepos first
760 760 for s in self.substate:
761 761 if self.sub(s).dirty():
762 762 return True
763 763 # check current working dir
764 764 return (self.p2() or self.branch() != self.p1().branch() or
765 765 self.modified() or self.added() or self.removed() or
766 766 (missing and self.deleted()))
767 767
768 768 def add(self, list):
769 769 wlock = self._repo.wlock()
770 770 ui, ds = self._repo.ui, self._repo.dirstate
771 771 try:
772 772 rejected = []
773 773 for f in list:
774 774 p = self._repo.wjoin(f)
775 775 try:
776 776 st = os.lstat(p)
777 777 except:
778 778 ui.warn(_("%s does not exist!\n") % f)
779 779 rejected.append(f)
780 780 continue
781 781 if st.st_size > 10000000:
782 782 ui.warn(_("%s: up to %d MB of RAM may be required "
783 783 "to manage this file\n"
784 784 "(use 'hg revert %s' to cancel the "
785 785 "pending addition)\n")
786 786 % (f, 3 * st.st_size // 1000000, f))
787 787 if not (stat.S_ISREG(st.st_mode) or stat.S_ISLNK(st.st_mode)):
788 788 ui.warn(_("%s not added: only files and symlinks "
789 789 "supported currently\n") % f)
790 790 rejected.append(p)
791 791 elif ds[f] in 'amn':
792 792 ui.warn(_("%s already tracked!\n") % f)
793 793 elif ds[f] == 'r':
794 794 ds.normallookup(f)
795 795 else:
796 796 ds.add(f)
797 797 return rejected
798 798 finally:
799 799 wlock.release()
800 800
801 801 def forget(self, list):
802 802 wlock = self._repo.wlock()
803 803 try:
804 804 for f in list:
805 805 if self._repo.dirstate[f] != 'a':
806 806 self._repo.ui.warn(_("%s not added!\n") % f)
807 807 else:
808 808 self._repo.dirstate.forget(f)
809 809 finally:
810 810 wlock.release()
811 811
812 812 def remove(self, list, unlink=False):
813 813 if unlink:
814 814 for f in list:
815 815 try:
816 816 util.unlink(self._repo.wjoin(f))
817 817 except OSError, inst:
818 818 if inst.errno != errno.ENOENT:
819 819 raise
820 820 wlock = self._repo.wlock()
821 821 try:
822 822 for f in list:
823 823 if unlink and os.path.exists(self._repo.wjoin(f)):
824 824 self._repo.ui.warn(_("%s still exists!\n") % f)
825 825 elif self._repo.dirstate[f] == 'a':
826 826 self._repo.dirstate.forget(f)
827 827 elif f not in self._repo.dirstate:
828 828 self._repo.ui.warn(_("%s not tracked!\n") % f)
829 829 else:
830 830 self._repo.dirstate.remove(f)
831 831 finally:
832 832 wlock.release()
833 833
834 834 def undelete(self, list):
835 835 pctxs = self.parents()
836 836 wlock = self._repo.wlock()
837 837 try:
838 838 for f in list:
839 839 if self._repo.dirstate[f] != 'r':
840 840 self._repo.ui.warn(_("%s not removed!\n") % f)
841 841 else:
842 842 fctx = f in pctxs[0] and pctxs[0] or pctxs[1]
843 843 t = fctx.data()
844 844 self._repo.wwrite(f, t, fctx.flags())
845 845 self._repo.dirstate.normal(f)
846 846 finally:
847 847 wlock.release()
848 848
849 849 def copy(self, source, dest):
850 850 p = self._repo.wjoin(dest)
851 851 if not (os.path.exists(p) or os.path.islink(p)):
852 852 self._repo.ui.warn(_("%s does not exist!\n") % dest)
853 853 elif not (os.path.isfile(p) or os.path.islink(p)):
854 854 self._repo.ui.warn(_("copy failed: %s is not a file or a "
855 855 "symbolic link\n") % dest)
856 856 else:
857 857 wlock = self._repo.wlock()
858 858 try:
859 859 if self._repo.dirstate[dest] in '?r':
860 860 self._repo.dirstate.add(dest)
861 861 self._repo.dirstate.copy(source, dest)
862 862 finally:
863 863 wlock.release()
864 864
865 865 class workingfilectx(filectx):
866 866 """A workingfilectx object makes access to data related to a particular
867 867 file in the working directory convenient."""
868 868 def __init__(self, repo, path, filelog=None, workingctx=None):
869 869 """changeid can be a changeset revision, node, or tag.
870 870 fileid can be a file revision or node."""
871 871 self._repo = repo
872 872 self._path = path
873 873 self._changeid = None
874 874 self._filerev = self._filenode = None
875 875
876 876 if filelog:
877 877 self._filelog = filelog
878 878 if workingctx:
879 879 self._changectx = workingctx
880 880
881 881 @propertycache
882 882 def _changectx(self):
883 883 return workingctx(self._repo)
884 884
885 885 def __nonzero__(self):
886 886 return True
887 887
888 888 def __str__(self):
889 889 return "%s@%s" % (self.path(), self._changectx)
890 890
891 891 def data(self):
892 892 return self._repo.wread(self._path)
893 893 def renamed(self):
894 894 rp = self._repo.dirstate.copied(self._path)
895 895 if not rp:
896 896 return None
897 897 return rp, self._changectx._parents[0]._manifest.get(rp, nullid)
898 898
899 899 def parents(self):
900 900 '''return parent filectxs, following copies if necessary'''
901 901 def filenode(ctx, path):
902 902 return ctx._manifest.get(path, nullid)
903 903
904 904 path = self._path
905 905 fl = self._filelog
906 906 pcl = self._changectx._parents
907 907 renamed = self.renamed()
908 908
909 909 if renamed:
910 910 pl = [renamed + (None,)]
911 911 else:
912 912 pl = [(path, filenode(pcl[0], path), fl)]
913 913
914 914 for pc in pcl[1:]:
915 915 pl.append((path, filenode(pc, path), fl))
916 916
917 917 return [filectx(self._repo, p, fileid=n, filelog=l)
918 918 for p, n, l in pl if n != nullid]
919 919
920 920 def children(self):
921 921 return []
922 922
923 923 def size(self):
924 924 return os.stat(self._repo.wjoin(self._path)).st_size
925 925 def date(self):
926 926 t, tz = self._changectx.date()
927 927 try:
928 928 return (int(os.lstat(self._repo.wjoin(self._path)).st_mtime), tz)
929 929 except OSError, err:
930 930 if err.errno != errno.ENOENT:
931 931 raise
932 932 return (t, tz)
933 933
934 934 def cmp(self, text):
935 return self._repo.wread(self._path) == text
935 return self._repo.wread(self._path) != text
936 936
937 937 class memctx(object):
938 938 """Use memctx to perform in-memory commits via localrepo.commitctx().
939 939
940 940 Revision information is supplied at initialization time while
941 941 related files data and is made available through a callback
942 942 mechanism. 'repo' is the current localrepo, 'parents' is a
943 943 sequence of two parent revisions identifiers (pass None for every
944 944 missing parent), 'text' is the commit message and 'files' lists
945 945 names of files touched by the revision (normalized and relative to
946 946 repository root).
947 947
948 948 filectxfn(repo, memctx, path) is a callable receiving the
949 949 repository, the current memctx object and the normalized path of
950 950 requested file, relative to repository root. It is fired by the
951 951 commit function for every file in 'files', but calls order is
952 952 undefined. If the file is available in the revision being
953 953 committed (updated or added), filectxfn returns a memfilectx
954 954 object. If the file was removed, filectxfn raises an
955 955 IOError. Moved files are represented by marking the source file
956 956 removed and the new file added with copy information (see
957 957 memfilectx).
958 958
959 959 user receives the committer name and defaults to current
960 960 repository username, date is the commit date in any format
961 961 supported by util.parsedate() and defaults to current date, extra
962 962 is a dictionary of metadata or is left empty.
963 963 """
964 964 def __init__(self, repo, parents, text, files, filectxfn, user=None,
965 965 date=None, extra=None):
966 966 self._repo = repo
967 967 self._rev = None
968 968 self._node = None
969 969 self._text = text
970 970 self._date = date and util.parsedate(date) or util.makedate()
971 971 self._user = user
972 972 parents = [(p or nullid) for p in parents]
973 973 p1, p2 = parents
974 974 self._parents = [changectx(self._repo, p) for p in (p1, p2)]
975 975 files = sorted(set(files))
976 976 self._status = [files, [], [], [], []]
977 977 self._filectxfn = filectxfn
978 978
979 979 self._extra = extra and extra.copy() or {}
980 980 if 'branch' not in self._extra:
981 981 self._extra['branch'] = 'default'
982 982 elif self._extra.get('branch') == '':
983 983 self._extra['branch'] = 'default'
984 984
985 985 def __str__(self):
986 986 return str(self._parents[0]) + "+"
987 987
988 988 def __int__(self):
989 989 return self._rev
990 990
991 991 def __nonzero__(self):
992 992 return True
993 993
994 994 def __getitem__(self, key):
995 995 return self.filectx(key)
996 996
997 997 def p1(self):
998 998 return self._parents[0]
999 999 def p2(self):
1000 1000 return self._parents[1]
1001 1001
1002 1002 def user(self):
1003 1003 return self._user or self._repo.ui.username()
1004 1004 def date(self):
1005 1005 return self._date
1006 1006 def description(self):
1007 1007 return self._text
1008 1008 def files(self):
1009 1009 return self.modified()
1010 1010 def modified(self):
1011 1011 return self._status[0]
1012 1012 def added(self):
1013 1013 return self._status[1]
1014 1014 def removed(self):
1015 1015 return self._status[2]
1016 1016 def deleted(self):
1017 1017 return self._status[3]
1018 1018 def unknown(self):
1019 1019 return self._status[4]
1020 1020 def ignored(self):
1021 1021 return self._status[5]
1022 1022 def clean(self):
1023 1023 return self._status[6]
1024 1024 def branch(self):
1025 1025 return self._extra['branch']
1026 1026 def extra(self):
1027 1027 return self._extra
1028 1028 def flags(self, f):
1029 1029 return self[f].flags()
1030 1030
1031 1031 def parents(self):
1032 1032 """return contexts for each parent changeset"""
1033 1033 return self._parents
1034 1034
1035 1035 def filectx(self, path, filelog=None):
1036 1036 """get a file context from the working directory"""
1037 1037 return self._filectxfn(self._repo, self, path)
1038 1038
1039 1039 def commit(self):
1040 1040 """commit context to the repo"""
1041 1041 return self._repo.commitctx(self)
1042 1042
1043 1043 class memfilectx(object):
1044 1044 """memfilectx represents an in-memory file to commit.
1045 1045
1046 1046 See memctx for more details.
1047 1047 """
1048 1048 def __init__(self, path, data, islink=False, isexec=False, copied=None):
1049 1049 """
1050 1050 path is the normalized file path relative to repository root.
1051 1051 data is the file content as a string.
1052 1052 islink is True if the file is a symbolic link.
1053 1053 isexec is True if the file is executable.
1054 1054 copied is the source file path if current file was copied in the
1055 1055 revision being committed, or None."""
1056 1056 self._path = path
1057 1057 self._data = data
1058 1058 self._flags = (islink and 'l' or '') + (isexec and 'x' or '')
1059 1059 self._copied = None
1060 1060 if copied:
1061 1061 self._copied = (copied, nullid)
1062 1062
1063 1063 def __nonzero__(self):
1064 1064 return True
1065 1065 def __str__(self):
1066 1066 return "%s@%s" % (self.path(), self._changectx)
1067 1067 def path(self):
1068 1068 return self._path
1069 1069 def data(self):
1070 1070 return self._data
1071 1071 def flags(self):
1072 1072 return self._flags
1073 1073 def isexec(self):
1074 1074 return 'x' in self._flags
1075 1075 def islink(self):
1076 1076 return 'l' in self._flags
1077 1077 def renamed(self):
1078 1078 return self._copied
General Comments 0
You need to be logged in to leave comments. Login now