/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to transportgit.py

Import gettext.

Show diffs side-by-side

added added

removed removed

Lines of Context:
transportgit.py
18
18
 
19
19
from cStringIO import StringIO
20
20
 
 
21
import os
 
22
 
21
23
from dulwich.errors import (
22
24
    NotGitRepository,
23
25
    NoIndexPresent,
33
35
    MemoryPackIndex,
34
36
    PackData,
35
37
    Pack,
36
 
    ThinPackData,
37
38
    iter_sha1,
38
39
    load_pack_index_file,
39
40
    write_pack_data,
42
43
from dulwich.repo import (
43
44
    BaseRepo,
44
45
    RefsContainer,
 
46
    BASE_DIRECTORIES,
45
47
    INDEX_FILENAME,
46
48
    OBJECTDIR,
47
49
    REFSDIR,
52
54
    write_packed_refs,
53
55
    )
54
56
 
 
57
from bzrlib import (
 
58
    transport as _mod_transport,
 
59
    )
55
60
from bzrlib.errors import (
56
61
    FileExists,
57
62
    NoSuchFile,
93
98
 
94
99
    def allkeys(self):
95
100
        keys = set()
96
 
        if self.transport.has("HEAD"):
 
101
        try:
 
102
            self.transport.get_bytes("HEAD")
 
103
        except NoSuchFile:
 
104
            pass
 
105
        else:
97
106
            keys.add("HEAD")
98
107
        try:
99
108
            iter_files = list(self.transport.clone("refs").iter_files_recursive())
279
288
 
280
289
class TransportRepo(BaseRepo):
281
290
 
282
 
    def __init__(self, transport):
 
291
    def __init__(self, transport, bare):
283
292
        self.transport = transport
284
 
        try:
285
 
            if self.transport.has(".git/%s" % OBJECTDIR):
286
 
                self.bare = False
287
 
                self._controltransport = self.transport.clone('.git')
288
 
            elif self.transport.has_any(["info/refs", OBJECTDIR, REFSDIR]):
289
 
                self.bare = True
290
 
                self._controltransport = self.transport
291
 
            else:
292
 
                raise NotGitRepository(self.transport)
293
 
        except NoSuchFile:
294
 
            raise NotGitRepository(self.transport)
 
293
        self.bare = bare
 
294
        if self.bare:
 
295
            self._controltransport = self.transport
 
296
        else:
 
297
            self._controltransport = self.transport.clone('.git')
295
298
        object_store = TransportObjectStore(
296
299
            self._controltransport.clone(OBJECTDIR))
297
300
        super(TransportRepo, self).__init__(object_store, 
312
315
        except NoSuchFile:
313
316
            return None
314
317
 
 
318
    def _put_named_file(self, relpath, contents):
 
319
        self._controltransport.put_bytes(relpath, contents)
 
320
 
315
321
    def index_path(self):
316
322
        """Return the path to the index file."""
317
323
        return self._controltransport.local_abspath(INDEX_FILENAME)
332
338
    def __repr__(self):
333
339
        return "<%s for %r>" % (self.__class__.__name__, self.transport)
334
340
 
 
341
    @classmethod
 
342
    def init(cls, transport, bare=False):
 
343
        if not bare:
 
344
            transport.mkdir(".git")
 
345
            control_transport = transport.clone(".git")
 
346
        else:
 
347
            control_transport = transport
 
348
        for d in BASE_DIRECTORIES:
 
349
            control_transport.mkdir("/".join(d))
 
350
        control_transport.mkdir(OBJECTDIR)
 
351
        TransportObjectStore.init(control_transport.clone(OBJECTDIR))
 
352
        ret = cls(transport, bare)
 
353
        ret.refs.set_symbolic_ref("HEAD", "refs/heads/master")
 
354
        ret._init_files(bare)
 
355
        return ret
 
356
 
335
357
 
336
358
class TransportObjectStore(PackBasedObjectStore):
337
359
    """Git-style object store that exists on disk."""
344
366
        super(TransportObjectStore, self).__init__()
345
367
        self.transport = transport
346
368
        self.pack_transport = self.transport.clone(PACKDIR)
 
369
        self._alternates = None
347
370
 
348
371
    def __repr__(self):
349
372
        return "%s(%r)" % (self.__class__.__name__, self.transport)
351
374
    def _pack_cache_stale(self):
352
375
        return False # FIXME
353
376
 
 
377
    @property
 
378
    def alternates(self):
 
379
        if self._alternates is not None:
 
380
            return self._alternates
 
381
        self._alternates = []
 
382
        for path in self._read_alternate_paths():
 
383
            # FIXME: Check path
 
384
            t = _mod_transport.get_transport_from_path(path)
 
385
            self._alternates.append(self.__class__(t))
 
386
        return self._alternates
 
387
 
 
388
    def _read_alternate_paths(self):
 
389
        try:
 
390
            f = self.transport.get("info/alternates")
 
391
        except NoSuchFile:
 
392
            return []
 
393
        ret = []
 
394
        try:
 
395
            for l in f.readlines():
 
396
                l = l.rstrip("\n")
 
397
                if l[0] == "#":
 
398
                    continue
 
399
                if os.path.isabs(l):
 
400
                    continue
 
401
                ret.append(l)
 
402
            return ret
 
403
        finally:
 
404
            f.close()
 
405
 
354
406
    def _pack_names(self):
355
407
        try:
356
408
            f = self.transport.get('info/packs')
473
525
        :param path: Path to the pack file.
474
526
        """
475
527
        f.seek(0)
476
 
        data = ThinPackData.from_file(self.get_raw, f, len(f.getvalue()))
 
528
        data = PackData.from_file(self.get_raw, f, len(f.getvalue()))
477
529
        idx = MemoryPackIndex(data.sorted_entries(), data.get_stored_checksum())
478
530
        p = Pack.from_objects(data, idx)
479
531
 
480
532
        pack_sha = idx.objects_sha1()
481
533
 
482
 
        datafile = self.pack_transport.open_write_stream("pack-%s.pack" % pack_sha)
 
534
        datafile = self.pack_transport.open_write_stream(
 
535
                "pack-%s.pack" % pack_sha)
483
536
        try:
484
 
            entries, data_sum = write_pack_data(datafile, ((o, None) for o in p.iterobjects()), len(p))
 
537
            entries, data_sum = write_pack_data(datafile, p.pack_tuples())
485
538
        finally:
486
539
            datafile.close()
487
540
        entries.sort()
488
 
        idxfile = self.pack_transport.open_write_stream("pack-%s.idx" % pack_sha)
 
541
        idxfile = self.pack_transport.open_write_stream(
 
542
            "pack-%s.idx" % pack_sha)
489
543
        try:
490
544
            write_pack_index_v2(idxfile, data.sorted_entries(), data_sum)
491
545
        finally:
494
548
        self._add_known_pack(final_pack)
495
549
        return final_pack
496
550
 
497
 
 
498
 
 
499
551
    def add_pack(self):
500
552
        """Add a new pack to this object store. 
501
553