/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 effbot/org/gzip_consumer.py

  • Committer: Martin Pool
  • Date: 2005-06-28 03:02:31 UTC
  • Revision ID: mbp@sourcefrog.net-20050628030231-d311e4ebcd467ef4
Merge John's import-speedup branch:

                                                                                         
  777 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 22:20:32 -0500
      revision-id: john@arbash-meinel.com-20050627032031-e82a50db3863b18e
      bzr selftest was not using the correct bzr

  776 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 22:20:22 -0500
      revision-id: john@arbash-meinel.com-20050627032021-c9f21fde989ddaee
      Add was using an old mutter

  775 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 22:02:33 -0500
      revision-id: john@arbash-meinel.com-20050627030233-9165cfe98fc63298
      Cleaned up to be less different

  774 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 21:54:53 -0500
      revision-id: john@arbash-meinel.com-20050627025452-4260d0e744edef43
      Allow BZR_PLUGIN_PATH='' to negate plugin loading.

  773 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 21:49:34 -0500
      revision-id: john@arbash-meinel.com-20050627024933-b7158f67b7b9eae5
      Finished the previous cleanup (allowing load_plugins to be called twice)

  772 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 21:45:08 -0500
      revision-id: john@arbash-meinel.com-20050627024508-723b1df510d196fc
      Work on making the tests pass. versioning.py is calling run_cmd directly, but plugins have been loaded.

  771 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 21:32:29 -0500
      revision-id: john@arbash-meinel.com-20050627023228-79972744d7c53e15
      Got it down a little bit more by removing import of tree and inventory.

  770 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 21:26:05 -0500
      revision-id: john@arbash-meinel.com-20050627022604-350b9773ef622f95
      Reducing the number of import from bzrlib/__init__.py and bzrlib/branch.py

  769 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 20:32:25 -0500
      revision-id: john@arbash-meinel.com-20050627013225-32dd044f10d23948
      Updated revision.py and xml.py to include SubElement.

  768 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 20:03:56 -0500
      revision-id: john@arbash-meinel.com-20050627010356-ee66919e1c377faf
      Minor typo

  767 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 20:03:13 -0500
      revision-id: john@arbash-meinel.com-20050627010312-40d024007eb85051
      Caching the import

  766 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 19:51:47 -0500
      revision-id: john@arbash-meinel.com-20050627005147-5281c99e48ed1834
      Created wrapper functions for lazy import of ElementTree

  765 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 19:46:37 -0500
      revision-id: john@arbash-meinel.com-20050627004636-bf432902004a94c5
      Removed all of the test imports of cElementTree

  764 John Arbash Meinel <john@arbash-meinel.com>       Sun 2005-06-26 19:43:59 -0500
      revision-id: john@arbash-meinel.com-20050627004358-d137fbe9570dd71b
      Trying to make bzr startup faster.

Show diffs side-by-side

added added

removed removed

Lines of Context:
effbot/org/gzip_consumer.py
 
1
# $Id: gzip_consumer.py 271 2004-10-09 10:50:59Z fredrik $
 
2
# gzip consumer
 
3
#
 
4
# Copyright (c) 2001-2004 by Fredrik Lundh.  All rights reserved.
 
5
#
 
6
 
 
7
##
 
8
# Consumer wrapper for GZIP streams.
 
9
 
 
10
class GzipConsumer:
 
11
 
 
12
    def __init__(self, consumer):
 
13
        self.__consumer = consumer
 
14
        self.__decoder = None
 
15
        self.__data = ""
 
16
 
 
17
    def __getattr__(self, key):
 
18
        return getattr(self.__consumer, key)
 
19
 
 
20
    def feed(self, data):
 
21
        if self.__decoder is None:
 
22
            # check if we have a full gzip header
 
23
            data = self.__data + data
 
24
            try:
 
25
                i = 10
 
26
                flag = ord(data[3])
 
27
                if flag & 4: # extra
 
28
                    x = ord(data[i]) + 256*ord(data[i+1])
 
29
                    i = i + 2 + x
 
30
                if flag & 8: # filename
 
31
                    while ord(data[i]):
 
32
                        i = i + 1
 
33
                    i = i + 1
 
34
                if flag & 16: # comment
 
35
                    while ord(data[i]):
 
36
                        i = i + 1
 
37
                    i = i + 1
 
38
                if flag & 2: # crc
 
39
                    i = i + 2
 
40
                if len(data) < i:
 
41
                    raise IndexError("not enough data")
 
42
                if data[:3] != "\x1f\x8b\x08":
 
43
                    raise IOError("invalid gzip data")
 
44
                data = data[i:]
 
45
            except IndexError:
 
46
                self.__data = data
 
47
                return # need more data
 
48
            import zlib
 
49
            self.__data = ""
 
50
            self.__decoder = zlib.decompressobj(-zlib.MAX_WBITS)
 
51
        data = self.__decoder.decompress(data)
 
52
        if data:
 
53
            self.__consumer.feed(data)
 
54
 
 
55
    def close(self):
 
56
        if self.__decoder:
 
57
            data = self.__decoder.flush()
 
58
            if data:
 
59
                self.__consumer.feed(data)
 
60
        self.__consumer.close()