def __init__(self, database, name='huey', **kwargs): super(SqlStorage, self).__init__(name) if isinstance(database, Database): self.database = database else: # Treat database argument as a URL connection string. self.database = db_url_connect(database) self.KV, self.Schedule, self.Task = self.create_models() self.create_tables()
def _load_database(self, app): config_value = app.config['DATABASE'] if isinstance(config_value, dict): database = self._load_from_config_dict(dict(config_value)) else: # Assume a database connection URL. database = db_url_connect(config_value) if isinstance(self.database, Proxy): self.database.initialize(database) else: self.database = database
def _load_database(self, app, config_value): if isinstance(config_value, Database): database = config_value elif isinstance(config_value, dict): database = self._load_from_config_dict(dict(config_value)) else: # Assume a database connection URL. database = db_url_connect(config_value) if isinstance(self.database, Proxy): self.database.initialize(database) else: self.database = database
def __init__(self, name='huey', database=None, **kwargs): super(SqlStorage, self).__init__(name) if database is None: raise ConfigurationError('Use of SqlStorage requires a ' 'database= argument, which should be a ' 'peewee database or a connection string.') if isinstance(database, Database): self.database = database else: # Treat database argument as a URL connection string. self.database = db_url_connect(database) self.KV, self.Schedule, self.Task = self.create_models() self.create_tables()
def db_connect(db_url, meta_model=GlobalMeta): db = db_url_connect(db_url) meta_model.database.initialize(db) meta_model.database.create_tables(MODELS, safe=True) return db