diff --git a/migrate/versioning/script/py.py b/migrate/versioning/script/py.py index 93ad776..9aeb3f4 100644 --- a/migrate/versioning/script/py.py +++ b/migrate/versioning/script/py.py @@ -12,6 +12,7 @@ from migrate.versioning.config import operations from migrate.versioning.template import Template from migrate.versioning.script import base from migrate.versioning.util import import_path, load_model, with_engine +from migrate.changeset.exceptions import * log = logging.getLogger(__name__) @@ -140,7 +141,7 @@ class PythonScript(base.BaseScript): script_func(engine) except TypeError: warnings.warn("upgrade/downgrade functions must accept engine" - " parameter (since version > 0.5.4)", exceptions.MigrateDeprecationWarning) + " parameter (since version > 0.5.4)", MigrateDeprecationWarning) raise @property diff --git a/migrate/versioning/util/__init__.py b/migrate/versioning/util/__init__.py index 5aefbae..4d39314 100644 --- a/migrate/versioning/util/__init__.py +++ b/migrate/versioning/util/__init__.py @@ -14,8 +14,8 @@ from sqlalchemy.pool import StaticPool from migrate.versioning import exceptions from migrate.versioning.util.keyedinstance import KeyedInstance from migrate.versioning.util.importpath import import_path -from migrate.changeset.exceptions import * +from migrate.changeset import * log = logging.getLogger(__name__) diff --git a/setup.py b/setup.py index 85c04ef..c0b6f6b 100644 --- a/setup.py +++ b/setup.py @@ -9,7 +9,7 @@ except ImportError: use_setuptools() from setuptools import setup, find_packages -required_deps = ['sqlalchemy >= 0.5', 'decorator', 'tempita >= 0.4'] +required_deps = ['SQLAlchemy >= 0.5', 'decorator', 'Tempita >= 0.4', 'setuptools'] readme_file = open(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'README'))