mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-02 07:18:25 +01:00
Package added with get_package_manager function and other utils.
import changed accordingly in version Solves issue #2672
This commit is contained in:
@@ -18,7 +18,7 @@ from dvc.command.base import CmdBaseNoRepo, append_doc_link
|
||||
from dvc.version import __version__
|
||||
from dvc.exceptions import DvcException, NotDvcRepoError
|
||||
from dvc.system import System
|
||||
from dvc.updater import Updater
|
||||
from dvc.utils.pkg import get_package_manager
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
@@ -31,7 +31,7 @@ class CmdVersion(CmdBaseNoRepo):
|
||||
python_version = platform.python_version()
|
||||
platform_type = platform.platform()
|
||||
binary = is_binary()
|
||||
package_manager = Updater("dvc/")._get_package_manager()
|
||||
package_manager = get_package_manager()
|
||||
info = (
|
||||
"DVC version: {dvc_version}\n"
|
||||
"Python version: {python_version}\n"
|
||||
|
||||
@@ -0,0 +1,51 @@
|
||||
from dvc.utils import is_binary
|
||||
|
||||
|
||||
def get_linux():
|
||||
import distro
|
||||
|
||||
if not is_binary():
|
||||
return "pip"
|
||||
|
||||
package_managers = {
|
||||
"rhel": "yum",
|
||||
"centos": "yum",
|
||||
"fedora": "yum",
|
||||
"amazon": "yum",
|
||||
"opensuse": "yum",
|
||||
"ubuntu": "apt",
|
||||
"debian": "apt",
|
||||
}
|
||||
|
||||
return package_managers.get(distro.id())
|
||||
|
||||
|
||||
def get_darwin():
|
||||
if not is_binary():
|
||||
if __file__.startswith("/usr/local/Cellar"):
|
||||
return "formula"
|
||||
else:
|
||||
return "pip"
|
||||
return None
|
||||
|
||||
|
||||
def get_windows():
|
||||
return None if is_binary() else "pip"
|
||||
|
||||
|
||||
def get_package_manager():
|
||||
import platform
|
||||
from dvc.exceptions import DvcException
|
||||
|
||||
m = {
|
||||
"Windows": get_windows(),
|
||||
"Darwin": get_darwin(),
|
||||
"Linux": get_linux(),
|
||||
}
|
||||
|
||||
system = platform.system()
|
||||
func = m.get(system)
|
||||
if func is None:
|
||||
raise DvcException("not supported system '{}'".format(system))
|
||||
|
||||
return func
|
||||
Reference in New Issue
Block a user