mercurial/cmdutil.py

34683:3d6d4b12128e

            self._addfileindir(filename, status)

        if status not in self.statuses:
            self.statuses.add(status)

    def _addfilestotersed(self, tersedict):
        """
        adds files to the their respective status list in the final tersed list

        path is the path of parent directory of the file
        files is a list of tuple where each tuple is (filename, status)
        tersedict is a dictonary which contains each status abbreviation as key and
        list of files and tersed dirs in that status as value
        """
        for f, st in self.files:
            tersedict[st].append(os.path.join(self.path, f))

    def _processtersestatus(self, tersedict, terseargs):
        """
        a recursive function which process status for a certain directory.

        self is an oject of dirnode class defined below. each object of dirnode
        class has a set of statuses which files in that directory has. This ease

        Following are the cases which can happen:

        1) All the files in the directory (including all the files in its
        subdirectories) share the same status and the user has asked us to terse
        that status. -> we add the directory name to status list and return

        2) If '1)' does not happen, we do following:

                a) Add all the files which are in this directory (only the ones in
                    this directory, not the subdirs) to their respective status list

                b) Recurse the function on all the subdirectories of this
                   directory
        """

            onlyst = self.statuses.pop()

            # Making sure we terse only when the status abbreviation is
            # passed as terse argument
            if onlyst in terseargs:
                tersedict[onlyst].append(self.path + pycompat.ossep)
                return

        # add the files to status list
        self._addfilestotersed(tersedict)

        #recurse on the subdirs
        for dirobj in self.subdirs.values():
            dirobj._processtersestatus(tersedict, terseargs)

def tersedir(statuslist, terseargs):
    """
    terses the status if all the files in a directory shares the same status

    stores the information required to know whether we can terse a certain
    directory or not.

    tersedict (defined in the function) is a dictionary which has one word key
    for each status and a list of files and dir in that status as the respective
    value. The dictionary is passed to other helper functions which builds it.
    """
    # the order matters here as that is used to produce final list
    allst = ('m', 'a', 'r', 'd', 'u', 'i', 'c')

    # checking the argument validity

        for f in getattr(statuslist, attrname):
            rootobj.addfile(f, attrname[0])
        tersedict[attrname[0]] = []

    # we won't be tersing the root dir, so add files in it
    rootobj._addfilestotersed(tersedict)

    # process each sub-directory and build tersedict
    for subdir in rootobj.subdirs.values():
        subdir._processtersestatus(tersedict, terseargs)

    tersedlist = []
    for st in allst:
        tersedict[st].sort()
        tersedlist.append(tersedict[st])

34684:5d98674df18a

            self._addfileindir(filename, status)

        if status not in self.statuses:
            self.statuses.add(status)

    def iterfilepaths(self):
        """
        adds files to the their respective status list in the final tersed list

        path is the path of parent directory of the file
        files is a list of tuple where each tuple is (filename, status)
        """
        for f, st in self.files:
            yield st, os.path.join(self.path, f)

    def tersewalk(self, terseargs):
        """
        a recursive function which process status for a certain directory.

        self is an oject of dirnode class defined below. each object of dirnode
        class has a set of statuses which files in that directory has. This ease

        Following are the cases which can happen:

        1) All the files in the directory (including all the files in its
        subdirectories) share the same status and the user has asked us to terse
        that status. -> yield (status, dirpath)

        2) If '1)' does not happen, we do following:

                a) Yield (status, filepath)  for all the files which are in this
                    directory (only the ones in this directory, not the subdirs)

                b) Recurse the function on all the subdirectories of this
                   directory
        """

            onlyst = self.statuses.pop()

            # Making sure we terse only when the status abbreviation is
            # passed as terse argument
            if onlyst in terseargs:
                yield onlyst, self.path + pycompat.ossep
                return

        # add the files to status list
        for st, fpath in self.iterfilepaths():
            yield st, fpath

        #recurse on the subdirs
        for dirobj in self.subdirs.values():
            for st, fpath in dirobj.tersewalk(terseargs):
                yield st, fpath

def tersedir(statuslist, terseargs):
    """
    terses the status if all the files in a directory shares the same status

    stores the information required to know whether we can terse a certain
    directory or not.

    tersedict (defined in the function) is a dictionary which has one word key
    for each status and a list of files and dir in that status as the respective
    value.
    """
    # the order matters here as that is used to produce final list
    allst = ('m', 'a', 'r', 'd', 'u', 'i', 'c')

    # checking the argument validity

        for f in getattr(statuslist, attrname):
            rootobj.addfile(f, attrname[0])
        tersedict[attrname[0]] = []

    # we won't be tersing the root dir, so add files in it
    for st, fpath in rootobj.iterfilepaths():
        tersedict[st].append(fpath)

    # process each sub-directory and build tersedict
    for subdir in rootobj.subdirs.values():
        for st, f in subdir.tersewalk(terseargs):
            tersedict[st].append(f)

    tersedlist = []
    for st in allst:
        tersedict[st].sort()
        tersedlist.append(tersedict[st])