def __init__(self, data_source, **kwargs): # Confirms required keys for datasource config info RDBSHub.__init__(self, data_source) # These attributes are required for certain base class methods self.data_source = data_source self.placeholder_char = '%s' self.quote_char = """'""" self.max_connect_attempts = 20 self.sleep_interval = 1 self.client_cursor = None if 'cursor' in kwargs: self.client_cursor = kwargs['cursor'] # Register return_type methods self.valid_return_types['iter'] = self.get_iter self.valid_return_types['dict'] = self.get_dict self.valid_return_types['dict_json'] = self.get_dict_json self.valid_return_types['tuple'] = self.get_tuple self.valid_return_types['tuple_json'] = self.get_tuple_json self.valid_return_types['set'] = self.get_set self.valid_return_types['table'] = self.get_table self.valid_return_types['table_json'] = self.get_table_json self.valid_return_types['set_json'] = self.get_set_json self.valid_return_types['rowcount'] = self.get_rowcount self.valid_return_types['callback'] = self.get_callback """ SQLHub.connection[ host_type ][ con_obj="Connection Object", cursor="Database cursor" ] """ self.connection = dict() # Configuration object for data source instance self.conf = self.get_data_source_config(self.data_source) # Load the procedures self.load_procs(self.data_source) __all__ = [ 'get_databases', # noqa 'use_database', 'escape_string', 'disconnect', 'execute', 'get_iter', 'get_dict', 'get_dict_json', 'get_list', 'get_list_json', 'get_set', 'get_set_json', 'get_callback' ]
def __init__(self, data_source, **kwargs): ##Confirms required keys for datasource config info## RDBSHub.__init__(self, data_source) ##These attributes are required for certain base class methods## self.data_source = data_source self.placeholder_char = "%s" self.quote_char = """'""" self.max_connect_attempts = 20 self.sleep_interval = 1 self.client_cursor = None if "cursor" in kwargs: self.client_cursor = kwargs["cursor"] ##Register return_type methods## self.valid_return_types["iter"] = self.get_iter self.valid_return_types["dict"] = self.get_dict self.valid_return_types["dict_json"] = self.get_dict_json self.valid_return_types["tuple"] = self.get_tuple self.valid_return_types["tuple_json"] = self.get_tuple_json self.valid_return_types["set"] = self.get_set self.valid_return_types["table"] = self.get_table self.valid_return_types["table_json"] = self.get_table_json self.valid_return_types["set_json"] = self.get_set_json self.valid_return_types["callback"] = self.get_callback """ SQLHub.connection[ host_type ][ con_obj="Connection Object", cursor="Database cursor" ] """ self.connection = dict() ##Configuration object for data source instance## self.conf = self.get_data_source_config(self.data_source) ##Load the procedures## self.load_procs(self.data_source) __all__ = [ "get_databases", "use_database", "escape_string", "disconnect", "execute", "get_iter", "get_dict", "get_dict_json", "get_list", "get_list_json", "get_set", "get_set_json", "get_callback", ]
def __init__(self, data_source, **kwargs): # Confirms required keys for datasource config info RDBSHub.__init__(self, data_source) # These attributes are required for certain base class methods self.data_source = data_source self.placeholder_char = '%s' self.quote_char = """'""" self.max_connect_attempts = 20 self.sleep_interval = 1 self.client_cursor = None if 'cursor' in kwargs: self.client_cursor = kwargs['cursor'] # Register return_type methods self.valid_return_types['iter'] = self.get_iter self.valid_return_types['dict'] = self.get_dict self.valid_return_types['dict_json'] = self.get_dict_json self.valid_return_types['tuple'] = self.get_tuple self.valid_return_types['tuple_json'] = self.get_tuple_json self.valid_return_types['set'] = self.get_set self.valid_return_types['table'] = self.get_table self.valid_return_types['table_json'] = self.get_table_json self.valid_return_types['set_json'] = self.get_set_json self.valid_return_types['rowcount'] = self.get_rowcount self.valid_return_types['callback'] = self.get_callback """ SQLHub.connection[ host_type ][ con_obj="Connection Object", cursor="Database cursor" ] """ self.connection = dict() # Configuration object for data source instance self.conf = self.get_data_source_config(self.data_source) # Load the procedures self.load_procs(self.data_source) __all__ = ['get_databases', # noqa 'use_database', 'escape_string', 'disconnect', 'execute', 'get_iter', 'get_dict', 'get_dict_json', 'get_list', 'get_list_json', 'get_set', 'get_set_json', 'get_callback']
def __init__(self, dataSource, **kwargs): ##Confirms required keys for datasource config info## RDBSHub.__init__(self, dataSource) ##These attributes are required for certain base class methods## self.dataSource = dataSource self.placeholderChar = '%s' self.quoteChar = """'""" self.maxConnectAttempts = 20 self.sleepInterval = 1 self.clientCursor = None if 'cursor' in kwargs: self.clientCursor = kwargs['cursor'] ##Register return_type methods## self.validReturnTypes['iter'] = self.getIter self.validReturnTypes['dict'] = self.getDict self.validReturnTypes['dict_json'] = self.getDictJson self.validReturnTypes['tuple'] = self.getTuple self.validReturnTypes['tuple_json'] = self.getTupleJson self.validReturnTypes['set'] = self.getSet self.validReturnTypes['table'] = self.getTable self.validReturnTypes['table_json'] = self.getTableJson self.validReturnTypes['set_json'] = self.getSetJson self.validReturnTypes['callback'] = self.getCallback """ SQLHub.connection[ hostType ][ con_obj="Connection Object", cursor="Database cursor" ] """ SQLHub.connection = dict() ##Configuration object for data source instance## self.conf = self.getDataSourceConfig(self.dataSource) ##Load the procedures## self.loadProcs(self.dataSource) __all__ = ['getDatabases', 'useDatabase', 'escapeString', 'disconnect', 'execute', 'getIter', 'getDict', 'getDictJson', 'getList', 'getListJson', 'getSet', 'getSetJson', 'getCallback']
def test_execute_rules(self): rh = RDBSHub(self.data_source) ### #See RDBSHub.set_execute_rules for test descriptions ### ######### #All tests should raise RDBSHubExecuteError ######### # 1.) make sure we recognize all of the args, chicken being the exception here args = dict(chicken=1, proc='fake.proc', return_type='tuple') self.__try_it(rh, args) # 2.) proc or sql must be provided or we have nothing to execute args = dict(return_type='tuple', db=self.db) self.__try_it(rh, args) # 3.) placeholders and replace must be set to lists args = dict(placeholders=dict()) self.__try_it(rh, args) args = dict(replace=dict()) self.__try_it(rh, args) # 4.) key_column is required if the return type is dict, dict_json, # set, or set_json for key in rh.return_type_key_columns: args = dict(return_type=key, proc='fake.proc') self.__try_it(rh, args) # 5.) If a return type of callback is selected a callback key must be # provided wih a function reference args = dict(return_type='callback', proc='fake.proc') self.__try_it(rh, args) # 6.) chunk_size must be provided with a chunk_source args = dict(chunk_size=100, proc='fake.proc') self.__try_it(rh, args) args = dict(chunk_source='table.column', proc='fake.proc') self.__try_it(rh, args)