diff --git a/src/simple_db_migrate/main.py b/src/simple_db_migrate/main.py index e86a031..efa29f9 100644 --- a/src/simple_db_migrate/main.py +++ b/src/simple_db_migrate/main.py @@ -5,13 +5,14 @@ import sys class Main(object): + def __init__(self, options=None, args=None, mysql=None, db_migrate=None): self.__cli = CLI() self.__options = options self.__args = args self.__mysql = mysql - if self.__mysql is None: + if self.__mysql is None and not self.__options.create_migration: self.__mysql = MySQL(db_config_file=self.__options.db_config_file, drop_db_first=self.__options.drop_db_first) self.__db_migrate = db_migrate diff --git a/src/simple_db_migrate/mysql.py b/src/simple_db_migrate/mysql.py index 87fa02d..5f895e6 100644 --- a/src/simple_db_migrate/mysql.py +++ b/src/simple_db_migrate/mysql.py @@ -3,7 +3,7 @@ import sys class MySQL(object): - + def __init__(self, db_config_file="simple-db-migrate.conf", mysql_driver=MySQLdb, drop_db_first=False): self.__cli = CLI()