mercurial/transaction.py

44432:a1908951ca42

# These are the file generators that should only be executed after the
# finalizers are done, since they rely on the output of the finalizers (like
# the changelog having been written).
postfinalizegenerators = {b'bookmarks', b'dirstate'}

gengroupall = b'all'
gengroupprefinalize = b'prefinalize'
gengrouppostfinalize = b'postfinalize'


def active(func):
    def _active(self, *args, **kwds):
        if self._count == 0:

    def removefilegenerator(self, genid):
        """reverse of addfilegenerator, remove a file generator function"""
        if genid in self._filegenerators:
            del self._filegenerators[genid]

    def _generatefiles(self, suffix=b'', group=gengroupall):
        # write files registered for generation
        any = False
        for id, entry in sorted(pycompat.iteritems(self._filegenerators)):
            any = True
            order, filenames, genfunc, location = entry

            # for generation at closing, check if it's before or after finalize
            postfinalize = group == gengrouppostfinalize
            if (
                group != gengroupall
                and (id in postfinalizegenerators) != postfinalize
            ):
                continue

            vfs = self._vfsmap[location]

    def close(self):
        '''commit the transaction'''
        if self._count == 1:
            self._validator(self)  # will raise exception if needed
            self._validator = None  # Help prevent cycles.
            self._generatefiles(group=gengroupprefinalize)
            while self._finalizecallback:
                callbacks = self._finalizecallback
                self._finalizecallback = {}
                categories = sorted(callbacks)
                for cat in categories:
                    callbacks[cat](self)
            # Prevent double usage and help clear cycles.
            self._finalizecallback = None
            self._generatefiles(group=gengrouppostfinalize)

        self._count -= 1
        if self._count != 0:
            return
        self._file.close()

44433:baf8c3f944eb

# These are the file generators that should only be executed after the
# finalizers are done, since they rely on the output of the finalizers (like
# the changelog having been written).
postfinalizegenerators = {b'bookmarks', b'dirstate'}

GEN_GROUP_ALL = b'all'
GEN_GROUP_PRE_FINALIZE = b'prefinalize'
GEN_GROUP_POST_FINALIZE = b'postfinalize'


def active(func):
    def _active(self, *args, **kwds):
        if self._count == 0:

    def removefilegenerator(self, genid):
        """reverse of addfilegenerator, remove a file generator function"""
        if genid in self._filegenerators:
            del self._filegenerators[genid]

    def _generatefiles(self, suffix=b'', group=GEN_GROUP_ALL):
        # write files registered for generation
        any = False
        for id, entry in sorted(pycompat.iteritems(self._filegenerators)):
            any = True
            order, filenames, genfunc, location = entry

            # for generation at closing, check if it's before or after finalize
            postfinalize = group == GEN_GROUP_POST_FINALIZE
            if (
                group != GEN_GROUP_ALL
                and (id in postfinalizegenerators) != postfinalize
            ):
                continue

            vfs = self._vfsmap[location]

    def close(self):
        '''commit the transaction'''
        if self._count == 1:
            self._validator(self)  # will raise exception if needed
            self._validator = None  # Help prevent cycles.
            self._generatefiles(group=GEN_GROUP_PRE_FINALIZE)
            while self._finalizecallback:
                callbacks = self._finalizecallback
                self._finalizecallback = {}
                categories = sorted(callbacks)
                for cat in categories:
                    callbacks[cat](self)
            # Prevent double usage and help clear cycles.
            self._finalizecallback = None
            self._generatefiles(group=GEN_GROUP_POST_FINALIZE)

        self._count -= 1
        if self._count != 0:
            return
        self._file.close()