from ..command import Command

__all__ = ("PoetryCommand",)


class PoetryCommand(Command):
    EXECUTOR: str = "poetry"

    def __init__(self, *args: str) -> None:
        super().__init__(self.EXECUTOR, *args)

    @classmethod
    def env_use(cls, python_version: str) -> "PoetryCommand":
        return cls("env", "use", python_version)

    @classmethod
    def install(cls) -> "PoetryCommand":
        return cls("install")