wc_linemode.py

import ranger.api
from ranger.core.linemode import LinemodeBase

from .wc_helper import WCHelper


@ranger.api.register_linemode
class WCLinemode(LinemodeBase):
    name = "wc"
    uses_metadata = False

    def __init__(self):
        "Initialize the ranger wordcount linemode plugin"
        self.wc_helper = WCHelper()

    def call_wc(self, filename: str) -> str:
        "Call wc and return the results"
        return self.wc_helper.call_wc(filename)

    def get_wc_string(self, res: str) -> str:
        "Convert the wc result to a string"
        return self.wc_helper.get_wc_string(res)

    def get_wc(self, res: str) -> int:
        "Parse the word count from the wc process call"
        return self.wc_helper.get_wc(res)

    def filetitle(self, fobj, metadata) -> str:
        return fobj.relative_path

    def infostring(self, fobj, metadata) -> str:
        """
        Add the number of words in the file to the linemode
        Or return the empty string
        """
        return self.wc_helper.infostring(fobj, metadata)