Source code for pipenv_pipes.picker.elements

# -*- coding: utf-8 -*-
from ..utils import collapse_path
from ..core import get_binary_version, read_project_dir_file
from .colors import colors


class Line():

    RIGHT_MARGIN = 2

    def __init__(self, text=None, color=None, pad=0):
        self._text = text
        self.color_name = color
        self.pad = pad

    @property
    def text(self):
        return ' ' * self.pad + self._text

    def render(self, screen, x, y):
        max_y, max_x = screen.getmaxyx()
        max_width = max_x - Line.RIGHT_MARGIN
        color_name = self.color_name
        color_pair = 0 if not color_name else colors[color_name].as_pair
        screen.addnstr(y, x, self.text, max_width, color_pair)


class EnvLine(Line):

    SELECTED_STR = '●'
    UNSELECTED_STR = ' ' * len(SELECTED_STR)
    HAS_DIR_STR = '*'
    UNSET_DIR_STR = '-- Not Set --'

    def __init__(self, env=None, **kwargs):
        self.env = env
        self.selected = kwargs.pop('selected')
        self.expanded = kwargs.pop('expanded')
        super().__init__(env.envname, **kwargs)

    @property
    def text(self):
        prefix = self.SELECTED_STR if self.selected else self.UNSELECTED_STR
        project_dir = read_project_dir_file(self.env.envpath)
        has_project_dir = bool(project_dir)
        if not has_project_dir:
            project_dir = self.UNSET_DIR_STR

        if self.expanded == 0:
            if has_project_dir:
                text = '{} {}'.format(self.env.envname, self.HAS_DIR_STR)
            else:
                text = self.env.envname
        if self.expanded == 1:
            binpath = get_binary_version(self.env.envpath)
            text = '{} ({})'.format(self.env.envname, binpath)
        if self.expanded == 2:
            text = collapse_path(self.env.envpath)
        if self.expanded == 3:
            text = project_dir

        return '{prefix} {text}'.format(prefix=prefix, text=text)