def get_router(directory, database, verbose=0): from peewee_migrate import LOGGER from peewee_migrate.compat import exec_in from peewee_migrate.router import Router logging_level = VERBOSE[verbose] config = {} migrate_table = 'migratehistory' ignore = schema = None try: with open(os.path.join(directory, 'conf.py')) as cfg: exec_in(cfg.read(), config, config) database = config.get('DATABASE', database) ignore = config.get('IGNORE', ignore) schema = config.get('SCHEMA', schema) migrate_table = config.get('MIGRATE_TABLE', migrate_table) logging_level = config.get('LOGGING_LEVEL', logging_level).upper() except IOError: pass if isinstance(database, string_types): database = connect(database) LOGGER.setLevel(logging_level) try: return Router(database, migrate_table=migrate_table, migrate_dir=directory, ignore=ignore, schema=schema) except RuntimeError as exc: LOGGER.error(exc) return sys.exit(1)
def get_router(directory, database, verbose=0): from peewee_migrate import LOGGER from peewee_migrate.compat import exec_in from peewee_migrate.router import Router logging_level = VERBOSE[verbose] config = {} try: with open(os.path.join(directory, 'conf.py')) as cfg: exec_in(cfg.read(), config, config) database = config.get('DATABASE', database) logging_level = config.get('LOGGING_LEVEL', logging_level).upper() except IOError: pass if isinstance(database, string_types): database = connect(database) LOGGER.setLevel(logging_level) try: return Router(database, migrate_dir=directory) except RuntimeError as exc: LOGGER.error(exc) return sys.exit(1)
def run(self): """Run operations.""" for opn in self.ops: if isinstance(opn, Operation): LOGGER.info("%s %s", opn.method, opn.args) opn.run() else: opn() self.clean()