Changeset d88ae5dd0edc…
Parent 71097e6909cf…
by
Changes to 2 files · Browse files at d88ae5dd0edc Showing diff from parent 71097e6909cf Diff from another changeset...
|
|
@@ -1,0 +1,114 @@ + # object_store.py -- Object store for git objects
+# Copyright (C) 2008 Jelmer Vernooij <jelmer@samba.org>
+#
+# This program is free software; you can redistribute it and/or
+# modify it under the terms of the GNU General Public License
+# as published by the Free Software Foundation; version 2
+# of the License.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
+# MA 02110-1301, USA.
+
+from objects import (
+ ShaFile,
+ )
+import os
+from pack import (
+ iter_sha1,
+ load_packs,
+ write_pack_index_v2,
+ PackData,
+ )
+PACKDIR = 'pack'
+
+class ObjectStore(object):
+
+ def __init__(self, path):
+ self.path = path
+ self._packs = None
+
+ def pack_dir(self):
+ return os.path.join(self.path, PACKDIR)
+
+ def __contains__(self, sha):
+ # TODO: This can be more efficient
+ try:
+ self[sha]
+ return True
+ except KeyError:
+ return False
+
+ @property
+ def packs(self):
+ """List with pack objects."""
+ if self._packs is None:
+ self._packs = list(load_packs(self.pack_dir()))
+ return self._packs
+
+ def _get_shafile(self, sha):
+ dir = sha[:2]
+ file = sha[2:]
+ # Check from object dir
+ path = os.path.join(self.path, dir, file)
+ if os.path.exists(path):
+ return ShaFile.from_file(path)
+ return None
+
+ def get_raw(self, sha):
+ """Obtain the raw text for an object.
+
+ :param sha: Sha for the object.
+ :return: tuple with object type and object contents.
+ """
+ for pack in self.packs:
+ if sha in pack:
+ return pack.get_raw(sha, self.get_raw)
+ # FIXME: Are pack deltas ever against on-disk shafiles ?
+ ret = self._get_shafile(sha)
+ if ret is not None:
+ return ret.as_raw_string()
+ raise KeyError(sha)
+
+ def __getitem__(self, sha):
+ assert len(sha) == 40, "Incorrect length sha: %s" % str(sha)
+ ret = self._get_shafile(sha)
+ if ret is not None:
+ return ret
+ # Check from packs
+ type, uncomp = self.get_raw(sha)
+ return ShaFile.from_raw_string(type, uncomp)
+
+ def move_in_pack(self, path):
+ """Move a specific file containing a pack into the pack directory.
+
+ :note: The file should be on the same file system as the
+ packs directory.
+
+ :param path: Path to the pack file.
+ """
+ p = PackData(path)
+ entries = p.sorted_entries(self.get_raw)
+ basename = os.path.join(self.pack_dir(),
+ "pack-%s" % iter_sha1(entry[0] for entry in entries))
+ write_pack_index_v2(basename+".idx", entries, p.calculate_checksum())
+ os.rename(path, basename + ".pack")
+
+ def add_pack(self):
+ """Add a new pack to this object store.
+
+ :return: Fileobject to write to and a commit function to
+ call when the pack is finished.
+ """
+ fd, path = tempfile.mkstemp(dir=self.pack_dir(), suffix=".pack")
+ f = os.fdopen(fd, 'w')
+ def commit():
+ if os.path.getsize(path) > 0:
+ self.move_in_pack(path)
+ return f, commit
|
|
@@ -27,22 +27,16 @@ NotGitRepository,
NotTreeError,
)
+from object_store import ObjectStore
from objects import (
ShaFile,
Commit,
Tree,
Blob,
)
-from pack import (
- iter_sha1,
- load_packs,
- write_pack_index_v2,
- PackData,
- )
import tempfile
OBJECTDIR = 'objects'
-PACKDIR = 'pack'
SYMREF = 'ref: '
@@ -293,87 +287,4 @@ create = init_bare
-class ObjectStore(object):
-
- def __init__(self, path):
- self.path = path
- self._packs = None
-
- def pack_dir(self):
- return os.path.join(self.path, PACKDIR)
-
- def __contains__(self, sha):
- # TODO: This can be more efficient
- try:
- self[sha]
- return True
- except KeyError:
- return False
-
- @property
- def packs(self):
- """List with pack objects."""
- if self._packs is None:
- self._packs = list(load_packs(self.pack_dir()))
- return self._packs
-
- def _get_shafile(self, sha):
- dir = sha[:2]
- file = sha[2:]
- # Check from object dir
- path = os.path.join(self.path, dir, file)
- if os.path.exists(path):
- return ShaFile.from_file(path)
- return None
-
- def get_raw(self, sha):
- """Obtain the raw text for an object.
-
- :param sha: Sha for the object.
- :return: tuple with object type and object contents.
- """
- for pack in self.packs:
- if sha in pack:
- return pack.get_raw(sha, self.get_raw)
- # FIXME: Are pack deltas ever against on-disk shafiles ?
- ret = self._get_shafile(sha)
- if ret is not None:
- return ret.as_raw_string()
- raise KeyError(sha)
-
- def __getitem__(self, sha):
- assert len(sha) == 40, "Incorrect length sha: %s" % str(sha)
- ret = self._get_shafile(sha)
- if ret is not None:
- return ret
- # Check from packs
- type, uncomp = self.get_raw(sha)
- return ShaFile.from_raw_string(type, uncomp)
-
- def move_in_pack(self, path):
- """Move a specific file containing a pack into the pack directory.
-
- :note: The file should be on the same file system as the
- packs directory.
-
- :param path: Path to the pack file.
- """
- p = PackData(path)
- entries = p.sorted_entries(self.get_raw)
- basename = os.path.join(self.pack_dir(),
- "pack-%s" % iter_sha1(entry[0] for entry in entries))
- write_pack_index_v2(basename+".idx", entries, p.calculate_checksum())
- os.rename(path, basename + ".pack")
-
- def add_pack(self):
- """Add a new pack to this object store.
-
- :return: Fileobject to write to and a commit function to
- call when the pack is finished.
- """
- fd, path = tempfile.mkstemp(dir=self.pack_dir(), suffix=".pack")
- f = os.fdopen(fd, 'w')
- def commit():
- if os.path.getsize(path) > 0:
- self.move_in_pack(path)
- return f, commit
+
|
Loading...