Use the distro package in dist_name if it exists or fall back to platform. Fixes #314

This commit is contained in:
Jacob Sayles
2020-06-01 10:18:54 -07:00
parent c0d61d86c7
commit e03167e476

View File

@@ -4,7 +4,6 @@ import contextlib
import datetime import datetime
import glob import glob
import os import os
import platform
import random import random
import shutil import shutil
import string import string
@@ -76,7 +75,14 @@ def exec_cmd(cmd, sudo_user=None, pinput=None, login=True, **kwargs):
def dist_name(): def dist_name():
"""Try to guess the distribution name.""" """Try to guess the distribution name."""
name, version, _id = platform.linux_distribution() try:
# Python 3.7 and up way
import distro
name, version, _id = distro.linux_distribution()
except ModuleNotFoundError as e:
# Python 3.6 and down way
import platform
name, version, _id = platform.linux_distribution()
return "unknown" if not name else name.lower() return "unknown" if not name else name.lower()