~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/branch.py

  • Committer: Martin Pool
  • Date: 2005-06-27 01:26:11 UTC
  • Revision ID: mbp@sourcefrog.net-20050627012611-4effb7007553fde1
- tweak rsync upload script

Show diffs side-by-side

added added

removed removed

Lines of Context:
15
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
17
 
18
 
import sys
19
 
import os
 
18
import sys, os, os.path, random, time, sha, sets, types, re, shutil, tempfile
 
19
import traceback, socket, fnmatch, difflib, time
 
20
from binascii import hexlify
20
21
 
21
22
import bzrlib
22
 
from bzrlib.trace import mutter, note
23
 
from bzrlib.osutils import isdir, quotefn, compact_date, rand_bytes, \
24
 
     splitpath, \
25
 
     sha_file, appendpath, file_kind
26
 
from bzrlib.errors import BzrError, InvalidRevisionNumber, InvalidRevisionId
27
 
import bzrlib.errors
28
 
from bzrlib.textui import show_status
29
 
from bzrlib.revision import Revision
30
 
from bzrlib.xml import unpack_xml
31
 
from bzrlib.delta import compare_trees
32
 
from bzrlib.tree import EmptyTree, RevisionTree
33
 
        
 
23
from inventory import Inventory
 
24
from trace import mutter, note
 
25
from tree import Tree, EmptyTree, RevisionTree
 
26
from inventory import InventoryEntry, Inventory
 
27
from osutils import isdir, quotefn, isfile, uuid, sha_file, username, \
 
28
     format_date, compact_date, pumpfile, user_email, rand_bytes, splitpath, \
 
29
     joinpath, sha_file, sha_string, file_kind, local_time_offset, appendpath
 
30
from store import ImmutableStore
 
31
from revision import Revision
 
32
from errors import BzrError
 
33
from textui import show_status
 
34
 
34
35
BZR_BRANCH_FORMAT = "Bazaar-NG branch, format 0.0.4\n"
35
36
## TODO: Maybe include checks for common corruption of newlines, etc?
36
37
 
37
38
 
38
 
# TODO: Some operations like log might retrieve the same revisions
39
 
# repeatedly to calculate deltas.  We could perhaps have a weakref
40
 
# cache in memory to make this faster.
41
 
 
42
 
# TODO: please move the revision-string syntax stuff out of the branch
43
 
# object; it's clutter
44
 
 
45
39
 
46
40
def find_branch(f, **args):
47
41
    if f and (f.startswith('http://') or f.startswith('https://')):
51
45
        return Branch(f, **args)
52
46
 
53
47
 
54
 
def find_cached_branch(f, cache_root, **args):
55
 
    from remotebranch import RemoteBranch
56
 
    br = find_branch(f, **args)
57
 
    def cacheify(br, store_name):
58
 
        from meta_store import CachedStore
59
 
        cache_path = os.path.join(cache_root, store_name)
60
 
        os.mkdir(cache_path)
61
 
        new_store = CachedStore(getattr(br, store_name), cache_path)
62
 
        setattr(br, store_name, new_store)
63
 
 
64
 
    if isinstance(br, RemoteBranch):
65
 
        cacheify(br, 'inventory_store')
66
 
        cacheify(br, 'text_store')
67
 
        cacheify(br, 'revision_store')
68
 
    return br
69
 
 
70
48
 
71
49
def _relpath(base, path):
72
50
    """Return path relative to base, or raise exception.
104
82
    It is not necessary that f exists.
105
83
 
106
84
    Basically we keep looking up until we find the control directory or
107
 
    run into the root.  If there isn't one, raises NotBranchError.
108
 
    """
 
85
    run into the root."""
109
86
    if f == None:
110
87
        f = os.getcwd()
111
88
    elif hasattr(os.path, 'realpath'):
124
101
        head, tail = os.path.split(f)
125
102
        if head == f:
126
103
            # reached the root, whatever that may be
127
 
            raise bzrlib.errors.NotBranchError('%r is not in a branch' % orig_f)
 
104
            raise BzrError('%r is not in a branch' % orig_f)
128
105
        f = head
129
 
 
130
 
 
131
 
 
132
 
# XXX: move into bzrlib.errors; subclass BzrError    
 
106
    
133
107
class DivergedBranches(Exception):
134
108
    def __init__(self, branch1, branch2):
135
109
        self.branch1 = branch1
137
111
        Exception.__init__(self, "These branches have diverged.")
138
112
 
139
113
 
 
114
class NoSuchRevision(BzrError):
 
115
    def __init__(self, branch, revision):
 
116
        self.branch = branch
 
117
        self.revision = revision
 
118
        msg = "Branch %s has no revision %d" % (branch, revision)
 
119
        BzrError.__init__(self, msg)
 
120
 
 
121
 
140
122
######################################################################
141
123
# branch objects
142
124
 
161
143
    _lock_count = None
162
144
    _lock = None
163
145
    
164
 
    # Map some sort of prefix into a namespace
165
 
    # stuff like "revno:10", "revid:", etc.
166
 
    # This should match a prefix with a function which accepts
167
 
    REVISION_NAMESPACES = {}
168
 
 
169
146
    def __init__(self, base, init=False, find_root=True):
170
147
        """Create new branch object at a particular location.
171
148
 
181
158
        In the test suite, creation of new trees is tested using the
182
159
        `ScratchBranch` class.
183
160
        """
184
 
        from bzrlib.store import ImmutableStore
185
161
        if init:
186
162
            self.base = os.path.realpath(base)
187
163
            self._make_control()
273
249
 
274
250
    def controlfilename(self, file_or_path):
275
251
        """Return location relative to branch."""
276
 
        if isinstance(file_or_path, basestring):
 
252
        if isinstance(file_or_path, types.StringTypes):
277
253
            file_or_path = [file_or_path]
278
254
        return os.path.join(self.base, bzrlib.BZRDIR, *file_or_path)
279
255
 
306
282
 
307
283
 
308
284
    def _make_control(self):
309
 
        from bzrlib.inventory import Inventory
310
 
        from bzrlib.xml import pack_xml
311
 
        
312
285
        os.mkdir(self.controlfilename([]))
313
286
        self.controlfile('README', 'w').write(
314
287
            "This is a Bazaar-NG control directory.\n"
318
291
            os.mkdir(self.controlfilename(d))
319
292
        for f in ('revision-history', 'merged-patches',
320
293
                  'pending-merged-patches', 'branch-name',
321
 
                  'branch-lock',
322
 
                  'pending-merges'):
 
294
                  'branch-lock'):
323
295
            self.controlfile(f, 'w').write('')
324
296
        mutter('created control directory in ' + self.base)
325
 
 
326
 
        # if we want per-tree root ids then this is the place to set
327
 
        # them; they're not needed for now and so ommitted for
328
 
        # simplicity.
329
 
        pack_xml(Inventory(), self.controlfile('inventory','w'))
 
297
        Inventory().write_xml(self.controlfile('inventory','w'))
330
298
 
331
299
 
332
300
    def _check_format(self):
347
315
                           ['use a different bzr version',
348
316
                            'or remove the .bzr directory and "bzr init" again'])
349
317
 
350
 
    def get_root_id(self):
351
 
        """Return the id of this branches root"""
352
 
        inv = self.read_working_inventory()
353
 
        return inv.root.file_id
354
318
 
355
 
    def set_root_id(self, file_id):
356
 
        inv = self.read_working_inventory()
357
 
        orig_root_id = inv.root.file_id
358
 
        del inv._byid[inv.root.file_id]
359
 
        inv.root.file_id = file_id
360
 
        inv._byid[inv.root.file_id] = inv.root
361
 
        for fid in inv:
362
 
            entry = inv[fid]
363
 
            if entry.parent_id in (None, orig_root_id):
364
 
                entry.parent_id = inv.root.file_id
365
 
        self._write_inventory(inv)
366
319
 
367
320
    def read_working_inventory(self):
368
321
        """Read the working inventory."""
369
 
        from bzrlib.inventory import Inventory
370
 
        from bzrlib.xml import unpack_xml
371
 
        from time import time
372
 
        before = time()
 
322
        before = time.time()
 
323
        # ElementTree does its own conversion from UTF-8, so open in
 
324
        # binary.
373
325
        self.lock_read()
374
326
        try:
375
 
            # ElementTree does its own conversion from UTF-8, so open in
376
 
            # binary.
377
 
            inv = unpack_xml(Inventory,
378
 
                             self.controlfile('inventory', 'rb'))
 
327
            inv = Inventory.read_xml(self.controlfile('inventory', 'rb'))
379
328
            mutter("loaded inventory of %d items in %f"
380
 
                   % (len(inv), time() - before))
 
329
                   % (len(inv), time.time() - before))
381
330
            return inv
382
331
        finally:
383
332
            self.unlock()
389
338
        That is to say, the inventory describing changes underway, that
390
339
        will be committed to the next revision.
391
340
        """
392
 
        from bzrlib.atomicfile import AtomicFile
393
 
        from bzrlib.xml import pack_xml
394
 
        
395
341
        self.lock_write()
396
342
        try:
 
343
            from bzrlib.atomicfile import AtomicFile
 
344
 
397
345
            f = AtomicFile(self.controlfilename('inventory'), 'wb')
398
346
            try:
399
 
                pack_xml(inv, f)
 
347
                inv.write_xml(f)
400
348
                f.commit()
401
349
            finally:
402
350
                f.close()
438
386
        """
439
387
        # TODO: Re-adding a file that is removed in the working copy
440
388
        # should probably put it back with the previous ID.
441
 
        if isinstance(files, basestring):
442
 
            assert(ids is None or isinstance(ids, basestring))
 
389
        if isinstance(files, types.StringTypes):
 
390
            assert(ids is None or isinstance(ids, types.StringTypes))
443
391
            files = [files]
444
392
            if ids is not None:
445
393
                ids = [ids]
494
442
            # use inventory as it was in that revision
495
443
            file_id = tree.inventory.path2id(file)
496
444
            if not file_id:
497
 
                raise BzrError("%r is not present in revision %s" % (file, revno))
 
445
                raise BzrError("%r is not present in revision %d" % (file, revno))
498
446
            tree.print_file(file_id)
499
447
        finally:
500
448
            self.unlock()
516
464
        """
517
465
        ## TODO: Normalize names
518
466
        ## TODO: Remove nested loops; better scalability
519
 
        if isinstance(files, basestring):
 
467
        if isinstance(files, types.StringTypes):
520
468
            files = [files]
521
469
 
522
470
        self.lock_write()
547
495
 
548
496
    # FIXME: this doesn't need to be a branch method
549
497
    def set_inventory(self, new_inventory_list):
550
 
        from bzrlib.inventory import Inventory, InventoryEntry
551
 
        inv = Inventory(self.get_root_id())
 
498
        inv = Inventory()
552
499
        for path, file_id, parent, kind in new_inventory_list:
553
500
            name = os.path.basename(path)
554
501
            if name == "":
576
523
        return self.working_tree().unknowns()
577
524
 
578
525
 
579
 
    def append_revision(self, *revision_ids):
 
526
    def append_revision(self, revision_id):
580
527
        from bzrlib.atomicfile import AtomicFile
581
528
 
582
 
        for revision_id in revision_ids:
583
 
            mutter("add {%s} to revision-history" % revision_id)
584
 
 
585
 
        rev_history = self.revision_history()
586
 
        rev_history.extend(revision_ids)
 
529
        mutter("add {%s} to revision-history" % revision_id)
 
530
        rev_history = self.revision_history() + [revision_id]
587
531
 
588
532
        f = AtomicFile(self.controlfilename('revision-history'))
589
533
        try:
594
538
            f.close()
595
539
 
596
540
 
597
 
    def get_revision_xml(self, revision_id):
598
 
        """Return XML file object for revision object."""
599
 
        if not revision_id or not isinstance(revision_id, basestring):
600
 
            raise InvalidRevisionId(revision_id)
601
 
 
602
 
        self.lock_read()
603
 
        try:
604
 
            try:
605
 
                return self.revision_store[revision_id]
606
 
            except IndexError:
607
 
                raise bzrlib.errors.NoSuchRevision(self, revision_id)
608
 
        finally:
609
 
            self.unlock()
610
 
 
611
 
 
612
541
    def get_revision(self, revision_id):
613
542
        """Return the Revision object for a named revision"""
614
 
        xml_file = self.get_revision_xml(revision_id)
615
 
 
616
 
        try:
617
 
            r = unpack_xml(Revision, xml_file)
618
 
        except SyntaxError, e:
619
 
            raise bzrlib.errors.BzrError('failed to unpack revision_xml',
620
 
                                         [revision_id,
621
 
                                          str(e)])
622
 
            
 
543
        if not revision_id or not isinstance(revision_id, basestring):
 
544
            raise ValueError('invalid revision-id: %r' % revision_id)
 
545
        r = Revision.read_xml(self.revision_store[revision_id])
623
546
        assert r.revision_id == revision_id
624
547
        return r
625
548
 
626
 
 
627
 
    def get_revision_delta(self, revno):
628
 
        """Return the delta for one revision.
629
 
 
630
 
        The delta is relative to its mainline predecessor, or the
631
 
        empty tree for revision 1.
632
 
        """
633
 
        assert isinstance(revno, int)
634
 
        rh = self.revision_history()
635
 
        if not (1 <= revno <= len(rh)):
636
 
            raise InvalidRevisionNumber(revno)
637
 
 
638
 
        # revno is 1-based; list is 0-based
639
 
 
640
 
        new_tree = self.revision_tree(rh[revno-1])
641
 
        if revno == 1:
642
 
            old_tree = EmptyTree()
643
 
        else:
644
 
            old_tree = self.revision_tree(rh[revno-2])
645
 
 
646
 
        return compare_trees(old_tree, new_tree)
647
 
 
648
 
        
649
 
 
650
549
    def get_revision_sha1(self, revision_id):
651
550
        """Hash the stored value of a revision, and return it."""
652
551
        # In the future, revision entries will be signed. At that
655
554
        # the revision, (add signatures/remove signatures) and still
656
555
        # have all hash pointers stay consistent.
657
556
        # But for now, just hash the contents.
658
 
        return bzrlib.osutils.sha_file(self.get_revision_xml(revision_id))
 
557
        return sha_file(self.revision_store[revision_id])
659
558
 
660
559
 
661
560
    def get_inventory(self, inventory_id):
664
563
        TODO: Perhaps for this and similar methods, take a revision
665
564
               parameter which can be either an integer revno or a
666
565
               string hash."""
667
 
        from bzrlib.inventory import Inventory
668
 
        from bzrlib.xml import unpack_xml
669
 
 
670
 
        return unpack_xml(Inventory, self.get_inventory_xml(inventory_id))
671
 
 
672
 
 
673
 
    def get_inventory_xml(self, inventory_id):
674
 
        """Get inventory XML as a file object."""
675
 
        return self.inventory_store[inventory_id]
676
 
            
 
566
        i = Inventory.read_xml(self.inventory_store[inventory_id])
 
567
        return i
677
568
 
678
569
    def get_inventory_sha1(self, inventory_id):
679
570
        """Return the sha1 hash of the inventory entry
680
571
        """
681
 
        return sha_file(self.get_inventory_xml(inventory_id))
 
572
        return sha_file(self.inventory_store[inventory_id])
682
573
 
683
574
 
684
575
    def get_revision_inventory(self, revision_id):
685
576
        """Return inventory of a past revision."""
686
 
        # bzr 0.0.6 imposes the constraint that the inventory_id
687
 
        # must be the same as its revision, so this is trivial.
688
577
        if revision_id == None:
689
 
            from bzrlib.inventory import Inventory
690
 
            return Inventory(self.get_root_id())
 
578
            return Inventory()
691
579
        else:
692
 
            return self.get_inventory(revision_id)
 
580
            return self.get_inventory(self.get_revision(revision_id).inventory_id)
693
581
 
694
582
 
695
583
    def revision_history(self):
750
638
                return r+1, my_history[r]
751
639
        return None, None
752
640
 
 
641
    def enum_history(self, direction):
 
642
        """Return (revno, revision_id) for history of branch.
 
643
 
 
644
        direction
 
645
            'forward' is from earliest to latest
 
646
            'reverse' is from latest to earliest
 
647
        """
 
648
        rh = self.revision_history()
 
649
        if direction == 'forward':
 
650
            i = 1
 
651
            for rid in rh:
 
652
                yield i, rid
 
653
                i += 1
 
654
        elif direction == 'reverse':
 
655
            i = len(rh)
 
656
            while i > 0:
 
657
                yield i, rh[i-1]
 
658
                i -= 1
 
659
        else:
 
660
            raise ValueError('invalid history direction', direction)
 
661
 
753
662
 
754
663
    def revno(self):
755
664
        """Return current revision number for this branch.
843
752
 
844
753
        pb.update('comparing histories')
845
754
        revision_ids = self.missing_revisions(other, stop_revision)
846
 
 
847
 
        if hasattr(other.revision_store, "prefetch"):
848
 
            other.revision_store.prefetch(revision_ids)
849
 
        if hasattr(other.inventory_store, "prefetch"):
850
 
            inventory_ids = [other.get_revision(r).inventory_id
851
 
                             for r in revision_ids]
852
 
            other.inventory_store.prefetch(inventory_ids)
853
 
                
854
755
        revisions = []
855
 
        needed_texts = set()
 
756
        needed_texts = sets.Set()
856
757
        i = 0
857
758
        for rev_id in revision_ids:
858
759
            i += 1
887
788
        commit(self, *args, **kw)
888
789
        
889
790
 
890
 
    def lookup_revision(self, revision):
891
 
        """Return the revision identifier for a given revision information."""
892
 
        revno, info = self.get_revision_info(revision)
893
 
        return info
894
 
 
895
 
    def get_revision_info(self, revision):
896
 
        """Return (revno, revision id) for revision identifier.
897
 
 
898
 
        revision can be an integer, in which case it is assumed to be revno (though
899
 
            this will translate negative values into positive ones)
900
 
        revision can also be a string, in which case it is parsed for something like
901
 
            'date:' or 'revid:' etc.
902
 
        """
903
 
        if revision is None:
904
 
            return 0, None
905
 
        revno = None
906
 
        try:# Convert to int if possible
907
 
            revision = int(revision)
908
 
        except ValueError:
909
 
            pass
910
 
        revs = self.revision_history()
911
 
        if isinstance(revision, int):
912
 
            if revision == 0:
913
 
                return 0, None
914
 
            # Mabye we should do this first, but we don't need it if revision == 0
915
 
            if revision < 0:
916
 
                revno = len(revs) + revision + 1
917
 
            else:
918
 
                revno = revision
919
 
        elif isinstance(revision, basestring):
920
 
            for prefix, func in Branch.REVISION_NAMESPACES.iteritems():
921
 
                if revision.startswith(prefix):
922
 
                    revno = func(self, revs, revision)
923
 
                    break
924
 
            else:
925
 
                raise BzrError('No namespace registered for string: %r' % revision)
926
 
 
927
 
        if revno is None or revno <= 0 or revno > len(revs):
928
 
            raise BzrError("no such revision %s" % revision)
929
 
        return revno, revs[revno-1]
930
 
 
931
 
    def _namespace_revno(self, revs, revision):
932
 
        """Lookup a revision by revision number"""
933
 
        assert revision.startswith('revno:')
934
 
        try:
935
 
            return int(revision[6:])
936
 
        except ValueError:
937
 
            return None
938
 
    REVISION_NAMESPACES['revno:'] = _namespace_revno
939
 
 
940
 
    def _namespace_revid(self, revs, revision):
941
 
        assert revision.startswith('revid:')
942
 
        try:
943
 
            return revs.index(revision[6:]) + 1
944
 
        except ValueError:
945
 
            return None
946
 
    REVISION_NAMESPACES['revid:'] = _namespace_revid
947
 
 
948
 
    def _namespace_last(self, revs, revision):
949
 
        assert revision.startswith('last:')
950
 
        try:
951
 
            offset = int(revision[5:])
952
 
        except ValueError:
953
 
            return None
954
 
        else:
955
 
            if offset <= 0:
956
 
                raise BzrError('You must supply a positive value for --revision last:XXX')
957
 
            return len(revs) - offset + 1
958
 
    REVISION_NAMESPACES['last:'] = _namespace_last
959
 
 
960
 
    def _namespace_tag(self, revs, revision):
961
 
        assert revision.startswith('tag:')
962
 
        raise BzrError('tag: namespace registered, but not implemented.')
963
 
    REVISION_NAMESPACES['tag:'] = _namespace_tag
964
 
 
965
 
    def _namespace_date(self, revs, revision):
966
 
        assert revision.startswith('date:')
967
 
        import datetime
968
 
        # Spec for date revisions:
969
 
        #   date:value
970
 
        #   value can be 'yesterday', 'today', 'tomorrow' or a YYYY-MM-DD string.
971
 
        #   it can also start with a '+/-/='. '+' says match the first
972
 
        #   entry after the given date. '-' is match the first entry before the date
973
 
        #   '=' is match the first entry after, but still on the given date.
974
 
        #
975
 
        #   +2005-05-12 says find the first matching entry after May 12th, 2005 at 0:00
976
 
        #   -2005-05-12 says find the first matching entry before May 12th, 2005 at 0:00
977
 
        #   =2005-05-12 says find the first match after May 12th, 2005 at 0:00 but before
978
 
        #       May 13th, 2005 at 0:00
979
 
        #
980
 
        #   So the proper way of saying 'give me all entries for today' is:
981
 
        #       -r {date:+today}:{date:-tomorrow}
982
 
        #   The default is '=' when not supplied
983
 
        val = revision[5:]
984
 
        match_style = '='
985
 
        if val[:1] in ('+', '-', '='):
986
 
            match_style = val[:1]
987
 
            val = val[1:]
988
 
 
989
 
        today = datetime.datetime.today().replace(hour=0,minute=0,second=0,microsecond=0)
990
 
        if val.lower() == 'yesterday':
991
 
            dt = today - datetime.timedelta(days=1)
992
 
        elif val.lower() == 'today':
993
 
            dt = today
994
 
        elif val.lower() == 'tomorrow':
995
 
            dt = today + datetime.timedelta(days=1)
996
 
        else:
997
 
            import re
998
 
            # This should be done outside the function to avoid recompiling it.
999
 
            _date_re = re.compile(
1000
 
                    r'(?P<date>(?P<year>\d\d\d\d)-(?P<month>\d\d)-(?P<day>\d\d))?'
1001
 
                    r'(,|T)?\s*'
1002
 
                    r'(?P<time>(?P<hour>\d\d):(?P<minute>\d\d)(:(?P<second>\d\d))?)?'
1003
 
                )
1004
 
            m = _date_re.match(val)
1005
 
            if not m or (not m.group('date') and not m.group('time')):
1006
 
                raise BzrError('Invalid revision date %r' % revision)
1007
 
 
1008
 
            if m.group('date'):
1009
 
                year, month, day = int(m.group('year')), int(m.group('month')), int(m.group('day'))
1010
 
            else:
1011
 
                year, month, day = today.year, today.month, today.day
1012
 
            if m.group('time'):
1013
 
                hour = int(m.group('hour'))
1014
 
                minute = int(m.group('minute'))
1015
 
                if m.group('second'):
1016
 
                    second = int(m.group('second'))
1017
 
                else:
1018
 
                    second = 0
1019
 
            else:
1020
 
                hour, minute, second = 0,0,0
1021
 
 
1022
 
            dt = datetime.datetime(year=year, month=month, day=day,
1023
 
                    hour=hour, minute=minute, second=second)
1024
 
        first = dt
1025
 
        last = None
1026
 
        reversed = False
1027
 
        if match_style == '-':
1028
 
            reversed = True
1029
 
        elif match_style == '=':
1030
 
            last = dt + datetime.timedelta(days=1)
1031
 
 
1032
 
        if reversed:
1033
 
            for i in range(len(revs)-1, -1, -1):
1034
 
                r = self.get_revision(revs[i])
1035
 
                # TODO: Handle timezone.
1036
 
                dt = datetime.datetime.fromtimestamp(r.timestamp)
1037
 
                if first >= dt and (last is None or dt >= last):
1038
 
                    return i+1
1039
 
        else:
1040
 
            for i in range(len(revs)):
1041
 
                r = self.get_revision(revs[i])
1042
 
                # TODO: Handle timezone.
1043
 
                dt = datetime.datetime.fromtimestamp(r.timestamp)
1044
 
                if first <= dt and (last is None or dt <= last):
1045
 
                    return i+1
1046
 
    REVISION_NAMESPACES['date:'] = _namespace_date
 
791
    def lookup_revision(self, revno):
 
792
        """Return revision hash for revision number."""
 
793
        if revno == 0:
 
794
            return None
 
795
 
 
796
        try:
 
797
            # list is 0-based; revisions are 1-based
 
798
            return self.revision_history()[revno-1]
 
799
        except IndexError:
 
800
            raise BzrError("no such revision %s" % revno)
 
801
 
1047
802
 
1048
803
    def revision_tree(self, revision_id):
1049
804
        """Return Tree for a revision on this branch.
1237
992
                f.close()
1238
993
 
1239
994
 
1240
 
    def pending_merges(self):
1241
 
        """Return a list of pending merges.
1242
 
 
1243
 
        These are revisions that have been merged into the working
1244
 
        directory but not yet committed.
1245
 
        """
1246
 
        cfn = self.controlfilename('pending-merges')
1247
 
        if not os.path.exists(cfn):
1248
 
            return []
1249
 
        p = []
1250
 
        for l in self.controlfile('pending-merges', 'r').readlines():
1251
 
            p.append(l.rstrip('\n'))
1252
 
        return p
1253
 
 
1254
 
 
1255
 
    def add_pending_merge(self, revision_id):
1256
 
        from bzrlib.revision import validate_revision_id
1257
 
 
1258
 
        validate_revision_id(revision_id)
1259
 
 
1260
 
        p = self.pending_merges()
1261
 
        if revision_id in p:
1262
 
            return
1263
 
        p.append(revision_id)
1264
 
        self.set_pending_merges(p)
1265
 
 
1266
 
 
1267
 
    def set_pending_merges(self, rev_list):
1268
 
        from bzrlib.atomicfile import AtomicFile
1269
 
        self.lock_write()
1270
 
        try:
1271
 
            f = AtomicFile(self.controlfilename('pending-merges'))
1272
 
            try:
1273
 
                for l in rev_list:
1274
 
                    print >>f, l
1275
 
                f.commit()
1276
 
            finally:
1277
 
                f.close()
1278
 
        finally:
1279
 
            self.unlock()
1280
 
 
1281
 
 
1282
995
 
1283
996
class ScratchBranch(Branch):
1284
997
    """Special test class: a branch that cleans up after itself.
1298
1011
 
1299
1012
        If any files are listed, they are created in the working copy.
1300
1013
        """
1301
 
        from tempfile import mkdtemp
1302
1014
        init = False
1303
1015
        if base is None:
1304
 
            base = mkdtemp()
 
1016
            base = tempfile.mkdtemp()
1305
1017
            init = True
1306
1018
        Branch.__init__(self, base, init=init)
1307
1019
        for d in dirs:
1320
1032
        >>> os.path.isfile(os.path.join(clone.base, "file1"))
1321
1033
        True
1322
1034
        """
1323
 
        from shutil import copytree
1324
 
        from tempfile import mkdtemp
1325
 
        base = mkdtemp()
 
1035
        base = tempfile.mkdtemp()
1326
1036
        os.rmdir(base)
1327
 
        copytree(self.base, base, symlinks=True)
 
1037
        shutil.copytree(self.base, base, symlinks=True)
1328
1038
        return ScratchBranch(base=base)
1329
1039
        
1330
1040
    def __del__(self):
1332
1042
 
1333
1043
    def destroy(self):
1334
1044
        """Destroy the test branch, removing the scratch directory."""
1335
 
        from shutil import rmtree
1336
1045
        try:
1337
1046
            if self.base:
1338
1047
                mutter("delete ScratchBranch %s" % self.base)
1339
 
                rmtree(self.base)
 
1048
                shutil.rmtree(self.base)
1340
1049
        except OSError, e:
1341
1050
            # Work around for shutil.rmtree failing on Windows when
1342
1051
            # readonly files are encountered
1344
1053
            for root, dirs, files in os.walk(self.base, topdown=False):
1345
1054
                for name in files:
1346
1055
                    os.chmod(os.path.join(root, name), 0700)
1347
 
            rmtree(self.base)
 
1056
            shutil.rmtree(self.base)
1348
1057
        self.base = None
1349
1058
 
1350
1059
    
1375
1084
    cope with just randomness because running uuidgen every time is
1376
1085
    slow."""
1377
1086
    import re
1378
 
    from binascii import hexlify
1379
 
    from time import time
1380
1087
 
1381
1088
    # get last component
1382
1089
    idx = name.rfind('/')
1394
1101
    name = re.sub(r'[^\w.]', '', name)
1395
1102
 
1396
1103
    s = hexlify(rand_bytes(8))
1397
 
    return '-'.join((name, compact_date(time()), s))
1398
 
 
1399
 
 
1400
 
def gen_root_id():
1401
 
    """Return a new tree-root file id."""
1402
 
    return gen_file_id('TREE_ROOT')
1403
 
 
 
1104
    return '-'.join((name, compact_date(time.time()), s))