From cbefac0904b0a77d60143e3f7c92c1a52b41893f Mon Sep 17 00:00:00 2001 From: Robert DeRose Date: Mon, 28 Mar 2016 13:44:25 -0400 Subject: [PATCH] Fix drop_database for SQLite The `drop_database` incorrectly reference the url.database value after it was set to `None`. It has been corrected to referrence `database` like the rest of the database types. --- sqlalchemy_utils/functions/database.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/sqlalchemy_utils/functions/database.py b/sqlalchemy_utils/functions/database.py index 20ffc9e..6557563 100644 --- a/sqlalchemy_utils/functions/database.py +++ b/sqlalchemy_utils/functions/database.py @@ -575,8 +575,9 @@ def drop_database(url): engine = sa.create_engine(url) - if engine.dialect.name == 'sqlite' and url.database != ':memory:': - os.remove(url.database) + if engine.dialect.name == 'sqlite' and database != ':memory:': + if database: + os.remove(database) elif engine.dialect.name == 'postgresql' and engine.driver == 'psycopg2': from psycopg2.extensions import ISOLATION_LEVEL_AUTOCOMMIT