summaryrefslogtreecommitdiffstats
path: root/pre_commit/prefix.py
blob: 0e3ebbd8980775ddbcc3d6a65cb4a30186c81d78 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
import os.path
from typing import NamedTuple
from typing import Tuple


class Prefix(NamedTuple):
    prefix_dir: str

    def path(self, *parts: str) -> str:
        return os.path.normpath(os.path.join(self.prefix_dir, *parts))

    def exists(self, *parts: str) -> bool:
        return os.path.exists(self.path(*parts))

    def star(self, end: str) -> Tuple[str, ...]:
        paths = os.listdir(self.prefix_dir)
        return tuple(path for path in paths if path.endswith(end))