From: Christopher Allan Webber <cwebber@dustycloud.org>
Date: Tue, 29 Mar 2016 16:44:34 +0000 (-0700)
Subject: migrations: Only don't import "migrate" if it's not available
X-Git-Url: https://vcs.fsf.org/?a=commitdiff_plain;h=e47ed5c81ceb9979edc42a678d635efcc4607528;p=mediagoblin.git

migrations: Only don't import "migrate" if it's not available

* mediagoblin/db/migrations.py: Import migrate if available; if not
  print a warning.
---

diff --git a/mediagoblin/db/migrations.py b/mediagoblin/db/migrations.py
index 257da6ee..55d64294 100644
--- a/mediagoblin/db/migrations.py
+++ b/mediagoblin/db/migrations.py
@@ -14,13 +14,21 @@
 # You should have received a copy of the GNU Affero General Public License
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
+from __future__ import print_function
+
 import datetime
 import uuid
 
 import six
 
-if six.PY2:
+try:
     import migrate
+except ImportError:
+    # Apparently sqlalchemy-migrate is not installed, so we assume
+    # we must not need it
+    # TODO: Better error handling here, or require sqlalchemy-migrate
+    print("sqlalchemy-migrate not found... assuming we don't need it")
+    print("I hope you aren't running the legacy migrations!")
 
 import pytz
 import dateutil.tz