Merge branch 'master' of git@stigma.nowoczesnapolska.org.pl:platforma
[redakcja.git] / lib / hg.py
1 # -*- coding: utf-8 -*-
2 import os
3 import codecs
4 from mercurial import localrepo, ui, match, node, encoding, util
5 import mercurial.merge, mercurial.error
6
7 encoding.encoding = 'utf-8'
8
9
10 class RepositoryDoesNotExist(Exception):
11     pass
12
13 class Repository(object):
14     """Abstrakcja repozytorium Mercurial. DziaƂa z Mercurial w wersji 1.3.1."""
15     
16     def __init__(self, path, create=False):
17         self.ui = ui.ui()
18         self.ui.config('ui', 'quiet', 'true')
19         self.ui.config('ui', 'interactive', 'false')
20         
21         self.real_path = os.path.realpath(path)
22         self.repo = self.open_repository(self.real_path, create)
23         self._pending_files = []
24     
25     def open_repository(self, path, create=False):
26         if os.path.isdir(path):
27             try:
28                 return localrepo.localrepository(self.ui, path)
29             except mercurial.error.RepoError:
30                 # dir is not an hg repo, we must init it
31                 if create:
32                     return localrepo.localrepository(self.ui, path, create=1)
33         elif create:
34             os.makedirs(path)
35             return localrepo.localrepository(self.ui, path, create=1)
36         raise RepositoryDoesNotExist("Repository %s does not exist." % path)
37         
38     def file_list(self, branch):
39         return self.in_branch(lambda: self._file_list(), branch)
40
41     def _file_list(self):
42         return list(self.repo[None])
43     
44     def get_file(self, path, branch):
45         return self.in_branch(lambda: self._get_file(path), branch)
46
47     def _get_file(self, path):
48         return self.repo.wread(path)
49     
50     def add_file(self, path, value, branch):
51         return self.in_branch(lambda: self._add_file(path, value), branch)
52
53     def _add_file(self, path, value):
54         return self.repo.wwrite(path, value.encode('utf-8'), [])
55
56     def _commit(self, message, user=None):
57         return self.repo.commit(text=message, user=user)
58     
59     def commit(self, message, branch, user=None):
60         return self.in_branch(lambda: self._commit(message, key=key, user=user), branch)
61
62     def in_branch(self, action, bname):
63         wlock = self.repo.wlock()
64         try:
65             old = self._switch_to_branch(bname)
66             try:
67                 # do some stuff
68                 return action()
69             finally:
70                 self._switch_to_branch(old)
71         finally:
72             wlock.release()
73
74     def _switch_to_branch(self, bname, create=True):
75         wlock = self.repo.wlock()
76         try:
77             current = self.repo[None].branch()
78             if current == bname:
79                 return current
80             try:
81                 tip = self.repo.branchtags()[bname]
82             except KeyError, ke:
83                 if not create: raise ke
84                 
85                 # create the branch on the fly
86
87                 # first switch to default branch
88                 default_tip = self.repo.branchtags()['default']
89                 mercurial.merge.update(self.repo, default_tip, False, True, None)
90
91                 # set the dirstate to new branch
92                 self.repo.dirstate.setbranch(bname)
93                 self._commit('Initial commit for automatic branch "%s".' % bname, user="django-admin")
94
95                 # collect the new tip
96                 tip = self.repo.branchtags()[bname]
97
98             upstats = mercurial.merge.update(self.repo, tip, False, True, None)
99             return current
100         except KeyError, ke:
101             raise RepositoryException("Can't switch to branch '%s': no such branch." % bname , ke)
102         except util.Abort, ae:
103             raise repositoryException("Can't switch to branch '%s': %s"  % (bname, ae.message), ae)
104         finally:
105             wlock.release()
106
107     def write_lock(self):
108         """Returns w write lock to the repository."""
109         return self.repo.wlock()
110
111
112 class RepositoryException(Exception):
113
114     def __init__(self, msg, cause=None):
115         Exception.__init__(self, msg)
116         self.cause = cause