X-Git-Url: https://git.mdrn.pl/redakcja.git/blobdiff_plain/352b8591bd1c7163835a6fa1db34d3e2861c1071..51d914bc0007135058a61623a2d057d2a7626a28:/lib/wlrepo/mercurial_backend/document.py diff --git a/lib/wlrepo/mercurial_backend/document.py b/lib/wlrepo/mercurial_backend/document.py index bced6741..3f94097a 100644 --- a/lib/wlrepo/mercurial_backend/document.py +++ b/lib/wlrepo/mercurial_backend/document.py @@ -5,16 +5,29 @@ __date__ = "$2009-09-25 09:35:06$" __doc__ = "Module documentation." import wlrepo +import mercurial.error class MercurialDocument(wlrepo.Document): def data(self, entry): - path = self._revision._docname + '.' + entry - return self._library._filectx(path, \ - self._revision.hgrev()).data() + path = self._library._sanitize_string(self.id + u'.' + entry) + try: + return self._library._filectx(path, \ + self._revision.hgrev()).data().decode('utf-8') + except mercurial.error.LookupError, e: + fl = [x.decode('utf-8') for x in self._revision._changectx] + raise wlrepo.EntryNotFound(self._revision, path.decode('utf-8'), fl) def quickwrite(self, entry, data, msg, user=None): user = user or self.owner + + if isinstance(data, unicode): + data = data.encode('utf-8') + + user = self._library._sanitize_string(user) + msg = self._library._sanitize_string(msg) + entry = self._library._sanitize_string(entry) + if user is None: raise ValueError("Can't determine user.") @@ -22,38 +35,46 @@ class MercurialDocument(wlrepo.Document): f = l._fileopen(r(entry), "w+") f.write(data) f.close() - l._fileadd(r(entry)) + l._fileadd(r(entry)) - return self.invoke_and_commit(write, lambda d: (msg, user)) + return self.invoke_and_commit(write, lambda d: (msg, \ + self._library._sanitize_string(self.owner)) ) - def invoke_and_commit(self, ops, before_commit): - lock = self._library._lock() + def invoke_and_commit(self, ops, commit_info): + lock = self._library.lock() try: self._library._checkout(self._revision.hgrev()) def entry_path(entry): - return self.id + '.' + entry + return self._library._sanitize_string(self.id + u'.' + entry) ops(self._library, entry_path) - message, user = before_commit(self) + message, user = commit_info(self) self._library._commit(message, user) - - return self._library.document(docid=self.id, user=self.owner) + try: + return self._library.document(docid=self.id, user=user) + except Exception, e: + # rollback the last commit + self._library._rollback() + raise e finally: - lock.release() - + lock.release() + # def commit(self, message, user): # """Make a new commit.""" # self.invoke_and_commit(message, user, lambda *a: True) def ismain(self): - return self._revision.user_name() is None + return self._revision.user_name is None def shared(self): if self.ismain(): return self - return self._library.document(docid=self._revision.document_name()) + + return self._library.document(docid=self.id) + def latest(self): + return self._library.document(docid=self.id, user=self.owner) def take(self, user): fullid = self._library.fulldocid(self.id, user) @@ -70,37 +91,34 @@ class MercurialDocument(wlrepo.Document): def update(self, user): """Update parts of the document.""" - lock = self.library._lock() + lock = self.library.lock() try: if self.ismain(): # main revision of the document - return True + return (True, False) if self._revision.has_children(): # can't update non-latest revision - return False + return (False, False) sv = self.shared() if not sv.ancestorof(self) and not self.parentof(sv): - self._revision.merge_with(sv._revision, user=user) + return self._revision.merge_with(sv._revision, user=user) - return True + return (False, False) finally: lock.release() def share(self, message): - lock = self.library._lock() + lock = self.library.lock() try: if self.ismain(): - return True # always shared - - user = self._revision.user_name() + return (True, False) # always shared + user = self._revision.user_name main = self.shared()._revision - local = self._revision - - no_changes = True + local = self._revision # Case 1: # * local @@ -117,8 +135,7 @@ class MercurialDocument(wlrepo.Document): if main.ancestorof(local): print "case 1" - main.merge_with(local, user=user, message=message) - no_changes = False + success, changed = main.merge_with(local, user=user, message=message) # Case 2: # # main * * local @@ -132,8 +149,7 @@ class MercurialDocument(wlrepo.Document): elif local.has_common_ancestor(main): print "case 2" if not local.parentof(main): - main.merge_with(local, user=user, message=message) - no_changes = False + success, changed = main.merge_with(local, user=user, message=message) # Case 3: # main * @@ -152,22 +168,33 @@ class MercurialDocument(wlrepo.Document): elif local.ancestorof(main): print "case 3" if not local.parentof(main): - local.merge_with(main, user=user, message='Local branch update.') - no_changes = False + success, changed = local.merge_with(main, user=user, \ + message='$AUTO$ Local branch update during share.') + else: print "case 4" - local.merge_with(main, user=user, message='Local branch update.') - local = self.shelf() - main.merge_with(local, user=user, message=message) + success, changed = local.merge_with(main, user=user, \ + message='$AUTO$ Local branch update during share.') + + if not success: + return False - print "no_changes: ", no_changes - return no_changes + if changed: + local = local.latest() + + success, changed = main.merge_with(local, user=user,\ + message=message) + + return success, changed finally: lock.release() - def __str__(self): + def __unicode__(self): return u"Document(%s:%s)" % (self.name, self.owner) + def __str__(self): + return self.__unicode__().encode('utf-8') + def __eq__(self, other): return (self._revision == other._revision) and (self.name == other.name)