From c1798dbe1fdd89b40132ce80a0be0daa71779495 Mon Sep 17 00:00:00 2001
From: Fabian Affolter <mail@fabian-affolter.ch>
Date: Fri, 15 Jul 2016 00:15:53 +0200
Subject: [PATCH] Catch ImportError (#2526)

---
 homeassistant/scripts/db_migrator.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/homeassistant/scripts/db_migrator.py b/homeassistant/scripts/db_migrator.py
index b00ed163942..dfe31be7684 100644
--- a/homeassistant/scripts/db_migrator.py
+++ b/homeassistant/scripts/db_migrator.py
@@ -4,8 +4,13 @@ import argparse
 import os.path
 import sqlite3
 import sys
-from sqlalchemy import create_engine
-from sqlalchemy.orm import sessionmaker
+try:
+    from sqlalchemy import create_engine
+    from sqlalchemy.orm import sessionmaker
+except ImportError:
+    print('Fatal Error: SQLAlchemy is missing. Install it with '
+          '"pip3 install SQLAlchemy" before running this script')
+    sys.exit(1)
 from homeassistant.components.recorder import models
 import homeassistant.config as config_util
 import homeassistant.util.dt as dt_util