Add Repo.get_shallow.

castedo · Apr 16, 2018 · 31e6508 · 31e6508
1 parent c2da1ee
commit 31e6508
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 0 deletions.
diff --git a/NEWS b/NEWS
@@ -11,6 +11,8 @@
   * Add `checkout` argument to `Repo.clone`.
     (Jelmer Vernooĳ, #503)
 
+  * Add `Repo.get_shallow` method. (Jelmer Vernooĳ)
+
  BUG FIXES
 
   * Fix handling of encoding for tags. (Jelmer Vernooĳ, #608)

diff --git a/dulwich/repo.py b/dulwich/repo.py
@@ -453,6 +453,13 @@ def get_config_stack(self):
         backends = [self.get_config()] + StackedConfig.default_backends()
         return StackedConfig(backends, writable=backends[0])
 
+    def get_shallow(self):
+        """Get the set of shallow commits.
+
+        :return: Set of shallow commits.
+        """
+        return set()
+
     def get_peeled(self, ref):
         """Get the peeled value of a ref.
 
@@ -907,6 +914,13 @@ def get_named_file(self, path, basedir=None):
                 return None
             raise
 
+    def get_shallow(self):
+        f = self.get_named_file('shallow')
+        if f is None:
+            return set()
+        with f:
+            return set(l.strip() for l in f)
+
     def index_path(self):
         """Return path to the index file."""
         return os.path.join(self.controldir(), INDEX_FILENAME)

diff --git a/dulwich/tests/test_repository.py b/dulwich/tests/test_repository.py
@@ -662,6 +662,13 @@ def setUp(self):
         self.assertEqual([], r[commit_sha].parents)
         self._root_commit = commit_sha
 
+    def test_shallow(self):
+        self.assertEqual(set(), self._repo.get_shallow())
+        with open(os.path.join(self._repo.path, '.git', 'shallow'), 'wb') as f:
+            f.write('a90fa2d900a17e99b433217e988c4eb4a2e9a097\n')
+        self.assertEqual({'a90fa2d900a17e99b433217e988c4eb4a2e9a097'},
+                         self._repo.get_shallow())
+
     def test_build_repo(self):
         r = self._repo
         self.assertEqual(b'ref: refs/heads/master', r.refs.read_ref(b'HEAD'))