<filename>powerline_shell/segments/svn.py<gh_stars>0
import subprocess

from powerline_shell import utils


def _get_svn_revision():
    p = subprocess.Popen(["svn", "info", "--xml"],
                         stdout=subprocess.PIPE,
                         stderr=subprocess.PIPE,
                         env=utils.get_subprocess_env())
    for line in p.communicate()[0].decode("utf-8").splitlines():
        if "revision" in line:
            revision = line.split("=")[1].split('"')[1]
            break
    return revision


def parse_svn_stats(status):
    stats = utils.RepoStats()
    for line in status:
        if line[0] == "?":
            stats.new += 1
        elif line[0] == "C":
            stats.conflicted += 1
        elif line[0] in ["A", "D", "I", "M", "R", "!", "~"]:
            stats.changed += 1
    return stats


def _get_svn_status(output):
    """This function exists to enable mocking the `svn status` output in tests.
    """
    return output[0].decode("utf-8").splitlines()


def build_stats():
    try:
        p = subprocess.Popen(['svn', 'status'],
                             stdout=subprocess.PIPE, stderr=subprocess.PIPE,
                             env=utils.get_subprocess_env())
    except OSError:
        # Popen will throw an OSError if svn is not found
        return None, None
    pdata = p.communicate()
    if p.returncode != 0 or pdata[1][:22] == b'svn: warning: W155007:':
        return None, None
    status = _get_svn_status(pdata)
    stats = parse_svn_stats(status)
    revision = _get_svn_revision()
    return stats, revision


class Segment(utils.ThreadedSegment):
    def run(self):
        self.stats, self.revision = build_stats()

    def add_to_powerline(self):
        self.join()
        if not self.stats:
            return
        bg = self.powerline.theme.REPO_CLEAN_BG
        fg = self.powerline.theme.REPO_CLEAN_FG
        if self.stats.dirty:
            bg = self.powerline.theme.REPO_DIRTY_BG
            fg = self.powerline.theme.REPO_DIRTY_FG
        if self.powerline.segment_conf("vcs", "show_symbol"):
            symbol = utils.RepoStats().symbols["svn"] + " "
        else:
            symbol = ""
        self.powerline.append(symbol + "rev " + self.revision, fg, bg)
        self.stats.add_to_powerline(self.powerline)
