← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/store.py

  • Committer: mbp at sourcefrog
  • Date: 2005-03-28 10:04:39 UTC
  • Revision ID: mbp@sourcefrog.net-20050328100439-80f2cf8bb0e5a621
better messages from check command

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/store.py
1
 
 
 
1
#! /usr/bin/env python
 
2
# -*- coding: UTF-8 -*-
2
3
 
3
4
# This program is free software; you can redistribute it and/or modify
4
5
# it under the terms of the GNU General Public License as published by
22
23
__copyright__ = "Copyright (C) 2005 Canonical Ltd."
23
24
__author__ = "Martin Pool <mbp@canonical.com>"
24
25
 
25
 
import os, tempfile, types, osutils, gzip, errno
 
26
import os, tempfile, types, osutils
26
27
from stat import ST_SIZE
27
28
from StringIO import StringIO
28
29
from trace import mutter
29
30
 
 
31
 
30
32
######################################################################
31
33
# stores
32
34
 
57
59
    >>> st['123123'].read()
58
60
    'goodbye'
59
61
 
60
 
    TODO: Atomic add by writing to a temporary file and renaming.
 
62
    :todo: Atomic add by writing to a temporary file and renaming.
61
63
 
62
 
    TODO: Perhaps automatically transform to/from XML in a method?
 
64
    :todo: Perhaps automatically transform to/from XML in a method?
63
65
           Would just need to tell the constructor what class to
64
66
           use...
65
67
 
66
 
    TODO: Even within a simple disk store like this, we could
 
68
    :todo: Even within a simple disk store like this, we could
67
69
           gzip the files.  But since many are less than one disk
68
70
           block, that might not help a lot.
69
71
 
79
81
    def __repr__(self):
80
82
        return "%s(%r)" % (self.__class__.__name__, self._basedir)
81
83
 
82
 
    def add(self, f, fileid, compressed=True):
 
84
    def add(self, f, fileid):
83
85
        """Add contents of a file into the store.
84
86
 
85
 
        f -- An open file, or file-like object."""
 
87
        :param f: An open file, or file-like object."""
86
88
        # FIXME: Only works on smallish files
87
89
        # TODO: Can be optimized by copying at the same time as
88
90
        # computing the sum.
91
93
            content = f
92
94
        else:
93
95
            content = f.read()
94
 
 
95
 
        p = self._path(fileid)
96
 
        if os.access(p, os.F_OK) or os.access(p + '.gz', os.F_OK):
97
 
            bailout("store %r already contains id %r" % (self._basedir, fileid))
98
 
 
99
 
        if compressed:
100
 
            f = gzip.GzipFile(p + '.gz', 'wb')
101
 
            os.chmod(p + '.gz', 0444)
102
 
        else:
103
 
            f = file(p, 'wb')
104
 
            os.chmod(p, 0444)
105
 
            
106
 
        f.write(content)
107
 
        f.close()
 
96
        if fileid not in self:
 
97
            filename = self._path(fileid)
 
98
            f = file(filename, 'wb')
 
99
            f.write(content)
 
100
            ## f.flush()
 
101
            ## os.fsync(f.fileno())
 
102
            f.close()
 
103
            osutils.make_readonly(filename)
108
104
 
109
105
 
110
106
    def __contains__(self, fileid):
111
107
        """"""
112
 
        p = self._path(fileid)
113
 
        return (os.access(p, os.R_OK)
114
 
                or os.access(p + '.gz', os.R_OK))
 
108
        return os.access(self._path(fileid), os.R_OK)
115
109
 
116
 
    # TODO: Guard against the same thing being stored twice, compressed and uncompresse
117
110
 
118
111
    def __iter__(self):
119
 
        for f in os.listdir(self._basedir):
120
 
            if f[-3:] == '.gz':
121
 
                # TODO: case-insensitive?
122
 
                yield f[:-3]
123
 
            else:
124
 
                yield f
 
112
        return iter(os.listdir(self._basedir))
125
113
 
126
114
    def __len__(self):
127
115
        return len(os.listdir(self._basedir))
128
116
 
129
117
    def __getitem__(self, fileid):
130
118
        """Returns a file reading from a particular entry."""
131
 
        p = self._path(fileid)
132
 
        try:
133
 
            return gzip.GzipFile(p + '.gz', 'rb')
134
 
        except IOError, e:
135
 
            if e.errno == errno.ENOENT:
136
 
                return file(p, 'rb')
137
 
            else:
138
 
                raise e
 
119
        return file(self._path(fileid), 'rb')
139
120
 
140
121
    def total_size(self):
141
 
        """Return (count, bytes)
142
 
 
143
 
        This is the (compressed) size stored on disk, not the size of
144
 
        the content."""
 
122
        """Return (count, bytes)"""
145
123
        total = 0
146
124
        count = 0
147
125
        for fid in self:
148
126
            count += 1
149
 
            p = self._path(fid)
150
 
            try:
151
 
                total += os.stat(p)[ST_SIZE]
152
 
            except OSError:
153
 
                total += os.stat(p + '.gz')[ST_SIZE]
154
 
                
 
127
            total += os.stat(self._path(fid))[ST_SIZE]
155
128
        return count, total
156
129
 
 
130
    def delete_all(self):
 
131
        for fileid in self:
 
132
            self.delete(fileid)
 
133
 
 
134
    def delete(self, fileid):
 
135
        """Remove nominated store entry.
 
136
 
 
137
        Most stores will be add-only."""
 
138
        filename = self._path(fileid)
 
139
        ## osutils.make_writable(filename)
 
140
        os.remove(filename)
 
141
 
 
142
    def destroy(self):
 
143
        """Remove store; only allowed if it is empty."""
 
144
        os.rmdir(self._basedir)
 
145
        mutter("%r destroyed" % self)
157
146
 
158
147
 
159
148
 
167
156
        ImmutableStore.__init__(self, tempfile.mkdtemp())
168
157
 
169
158
    def __del__(self):
170
 
        for f in os.listdir(self._basedir):
171
 
            fpath = os.path.join(self._basedir, f)
172
 
            # needed on windows, and maybe some other filesystems
173
 
            os.chmod(fpath, 0600)
174
 
            os.remove(fpath)
175
 
        os.rmdir(self._basedir)
176
 
        mutter("%r destroyed" % self)
 
159
        self.delete_all()
 
160
        self.destroy()