X-Git-Url: https://git.mdrn.pl/redakcja.git/blobdiff_plain/e8fe7696e2fc6334ef96495ea6d9d71354c53b6d..b9e5161a49d56f4bc177159c93d0f027e5ba4337:/lib/hg.py?ds=sidebyside

diff --git a/lib/hg.py b/lib/hg.py
index d8b4679a..2b885b3c 100644
--- a/lib/hg.py
+++ b/lib/hg.py
@@ -3,10 +3,8 @@ import os
 import codecs
 from mercurial import localrepo, ui, error, match, node, encoding
 
-
 encoding.encoding = 'utf-8'
 
-
 class RepositoryDoesNotExist(Exception):
     pass
 
@@ -110,4 +108,22 @@ class Repository(object):
             # reread keys
             # self._keys = self.get_persisted_objects_keys()
             # return node.hex(rev)
-            
\ No newline at end of file
+
+    def in_branch(self, branch_name, action):
+        wlock = self.repo.wlock()
+        try:
+            current_branch = self.repo[None].branch()
+            self.repo.dirstate.setbranch(branch_name)
+            try:
+                # do some stuff
+                action()
+            finally:
+                self.repo.dirstate.setbranch(current_branch)
+        finally:
+            wlock.release()
+
+    def write_lock(self):
+        """Returns w write lock to the repository."""
+        return self.repo.wlock()
+
+