152 lines
4.3 KiB
Python
152 lines
4.3 KiB
Python
import os
|
|
import subprocess
|
|
import sys
|
|
from contextlib import suppress
|
|
from typing import Optional
|
|
|
|
|
|
class Pip:
|
|
r"""Executes pip commands"""
|
|
_executable = sys.executable
|
|
_env = os.environ
|
|
_is_venv = False
|
|
|
|
"""Getter"""
|
|
@classmethod
|
|
def get_executable(cls) -> str:
|
|
return cls._executable
|
|
|
|
"""Setter"""
|
|
@classmethod
|
|
def set_executable(cls, executable: str):
|
|
r"""Sets the executable
|
|
|
|
Parameter
|
|
---------
|
|
executable: :class:`str`
|
|
The python command
|
|
"""
|
|
if executable is not None and executable != sys.executable:
|
|
cls._executable = executable
|
|
if os.path.islink(cls._executable):
|
|
cls._is_venv = True
|
|
path = os.path.dirname(os.path.dirname(cls._executable))
|
|
cls._env = os.environ
|
|
if sys.platform == 'win32':
|
|
cls._env['PATH'] = f'{path}\\bin' + os.pathsep + os.environ.get('PATH', '')
|
|
else:
|
|
cls._env['PATH'] = f'{path}/bin' + os.pathsep + os.environ.get('PATH', '')
|
|
cls._env['VIRTUAL_ENV'] = path
|
|
|
|
@classmethod
|
|
def reset_executable(cls):
|
|
r"""Resets the executable to system standard"""
|
|
cls._executable = sys.executable
|
|
cls._is_venv = False
|
|
|
|
"""Public utils functions"""
|
|
@classmethod
|
|
def get_package(cls, package: str) -> Optional[str]:
|
|
r"""Gets given package py local pip list
|
|
|
|
Parameter
|
|
---------
|
|
package: :class:`str`
|
|
|
|
Returns
|
|
-------
|
|
The package name as string
|
|
"""
|
|
result = None
|
|
with suppress(Exception):
|
|
args = [cls._executable, "-m", "pip", "show", package]
|
|
if cls._is_venv:
|
|
args = ["pip", "show", package]
|
|
|
|
result = subprocess.check_output(
|
|
args,
|
|
stderr=subprocess.DEVNULL, env=cls._env
|
|
)
|
|
|
|
if result is None:
|
|
return None
|
|
|
|
new_package: list[str] = str(result, 'utf-8').lower().split('\n')
|
|
new_version = ''
|
|
|
|
for atr in new_package:
|
|
if 'version' in atr:
|
|
new_version = atr.split(': ')[1]
|
|
|
|
if new_version != '':
|
|
return f'{package}=={new_version}'
|
|
|
|
return package
|
|
|
|
@classmethod
|
|
def get_outdated(cls) -> bytes:
|
|
r"""Gets table of outdated packages
|
|
|
|
Returns
|
|
-------
|
|
Bytes string of the command result
|
|
"""
|
|
args = [cls._executable, "-m", "pip", "list", "--outdated"]
|
|
if cls._is_venv:
|
|
args = ["pip", "list", "--outdated"]
|
|
|
|
return subprocess.check_output(args, env=cls._env)
|
|
|
|
@classmethod
|
|
def install(cls, package: str, *args, source: str = None, stdout=None, stderr=None):
|
|
r"""Installs given package
|
|
|
|
Parameter
|
|
---------
|
|
package: :class:`str`
|
|
The name of the package
|
|
args: :class:`list`
|
|
Arguments for the command
|
|
source: :class:`str`
|
|
Extra index URL
|
|
stdout: :class:`str`
|
|
Stdout of subprocess.run
|
|
stderr: :class:`str`
|
|
Stderr of subprocess.run
|
|
"""
|
|
pip_args = [cls._executable, "-m", "pip", "install"]
|
|
if cls._is_venv:
|
|
pip_args = ["pip", "install"]
|
|
|
|
for arg in args:
|
|
pip_args.append(arg)
|
|
|
|
if source is not None:
|
|
pip_args.append(f'--extra-index-url')
|
|
pip_args.append(source)
|
|
|
|
pip_args.append(package)
|
|
subprocess.run(pip_args, stdout=stdout, stderr=stderr, env=cls._env)
|
|
|
|
@classmethod
|
|
def uninstall(cls, package: str, stdout=None, stderr=None):
|
|
r"""Uninstalls given package
|
|
|
|
Parameter
|
|
---------
|
|
package: :class:`str`
|
|
The name of the package
|
|
stdout: :class:`str`
|
|
Stdout of subprocess.run
|
|
stderr: :class:`str`
|
|
Stderr of subprocess.run
|
|
"""
|
|
args = [cls._executable, "-m", "pip", "uninstall", "--yes", package]
|
|
if cls._is_venv:
|
|
args = ["pip", "uninstall", "--yes", package]
|
|
|
|
subprocess.run(
|
|
args,
|
|
stdout=stdout, stderr=stderr, env=cls._env
|
|
)
|