| hg-reviewers |
This holds instance state that is passed in at init time. It
doesn't need to be exposed as part of the public interface.
| Automatic diff as part of commit; lint not applicable. |
| Automatic diff as part of commit; unit tests not applicable. |
| if validator is None: | if validator is None: | ||||
| validator = lambda tr: None | validator = lambda tr: None | ||||
| self._validator = validator | self._validator = validator | ||||
| # A callback to do something just after releasing transaction. | # A callback to do something just after releasing transaction. | ||||
| if releasefn is None: | if releasefn is None: | ||||
| releasefn = lambda tr, success: None | releasefn = lambda tr, success: None | ||||
| self._releasefn = releasefn | self._releasefn = releasefn | ||||
| self.checkambigfiles = set() | self._checkambigfiles = set() | ||||
| if checkambigfiles: | if checkambigfiles: | ||||
| self.checkambigfiles.update(checkambigfiles) | self._checkambigfiles.update(checkambigfiles) | ||||
| self.names = [name] | self.names = [name] | ||||
| # A dict dedicated to precisely tracking the changes introduced in the | # A dict dedicated to precisely tracking the changes introduced in the | ||||
| # transaction. | # transaction. | ||||
| self.changes = {} | self.changes = {} | ||||
| # a dict of arguments to be passed to hooks | # a dict of arguments to be passed to hooks | ||||
| try: | try: | ||||
| for name in filenames: | for name in filenames: | ||||
| name += suffix | name += suffix | ||||
| if suffix: | if suffix: | ||||
| self.registertmp(name, location=location) | self.registertmp(name, location=location) | ||||
| checkambig = False | checkambig = False | ||||
| else: | else: | ||||
| self.addbackup(name, location=location) | self.addbackup(name, location=location) | ||||
| checkambig = (name, location) in self.checkambigfiles | checkambig = (name, location) in self._checkambigfiles | ||||
| files.append(vfs(name, 'w', atomictemp=True, | files.append(vfs(name, 'w', atomictemp=True, | ||||
| checkambig=checkambig)) | checkambig=checkambig)) | ||||
| genfunc(*files) | genfunc(*files) | ||||
| finally: | finally: | ||||
| for f in files: | for f in files: | ||||
| f.close() | f.close() | ||||
| return any | return any | ||||
| try: | try: | ||||
| for cat in sorted(self._abortcallback): | for cat in sorted(self._abortcallback): | ||||
| self._abortcallback[cat](self) | self._abortcallback[cat](self) | ||||
| # Prevent double usage and help clear cycles. | # Prevent double usage and help clear cycles. | ||||
| self._abortcallback = None | self._abortcallback = None | ||||
| _playback(self._journal, self.report, self.opener, self._vfsmap, | _playback(self._journal, self.report, self.opener, self._vfsmap, | ||||
| self.entries, self._backupentries, False, | self.entries, self._backupentries, False, | ||||
| checkambigfiles=self.checkambigfiles) | checkambigfiles=self._checkambigfiles) | ||||
| self.report(_("rollback completed\n")) | self.report(_("rollback completed\n")) | ||||
| except BaseException: | except BaseException: | ||||
| self.report(_("rollback failed - please run hg recover\n")) | self.report(_("rollback failed - please run hg recover\n")) | ||||
| finally: | finally: | ||||
| self._journal = None | self._journal = None | ||||
| self._releasefn(self, False) # notify failure of transaction | self._releasefn(self, False) # notify failure of transaction | ||||
| self._releasefn = None # Help prevent cycles. | self._releasefn = None # Help prevent cycles. | ||||