forked from home-assistant/core
Update dependency check in __main__
This commit is contained in:
@@ -7,6 +7,14 @@ import argparse
|
|||||||
import importlib
|
import importlib
|
||||||
|
|
||||||
|
|
||||||
|
# Home Assistant dependencies, mapped module -> package name
|
||||||
|
DEPENDENCIES = {
|
||||||
|
'requests': 'requests',
|
||||||
|
'yaml': 'pyyaml',
|
||||||
|
'pytz': 'pytz',
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
def validate_python():
|
def validate_python():
|
||||||
""" Validate we're running the right Python version. """
|
""" Validate we're running the right Python version. """
|
||||||
major, minor = sys.version_info[:2]
|
major, minor = sys.version_info[:2]
|
||||||
@@ -20,13 +28,13 @@ def validate_dependencies():
|
|||||||
""" Validate all dependencies that HA uses. """
|
""" Validate all dependencies that HA uses. """
|
||||||
import_fail = False
|
import_fail = False
|
||||||
|
|
||||||
for module in ['requests']:
|
for module, name in DEPENDENCIES.items():
|
||||||
try:
|
try:
|
||||||
importlib.import_module(module)
|
importlib.import_module(module)
|
||||||
except ImportError:
|
except ImportError:
|
||||||
import_fail = True
|
import_fail = True
|
||||||
print(
|
print(
|
||||||
'Fatal Error: Unable to find dependency {}'.format(module))
|
'Fatal Error: Unable to find dependency {}'.format(name))
|
||||||
|
|
||||||
if import_fail:
|
if import_fail:
|
||||||
print(("Install dependencies by running: "
|
print(("Install dependencies by running: "
|
||||||
|
Reference in New Issue
Block a user