def test_get_all_tables(self, mock_read_sql_query): res = [1., 2., 3.] mock_read_sql_query.return_value = pd.DataFrame({'Tables_in_stock': Series(res)}) test_cm = cmysql.CMySQL(ct.DB_INFO) self.assertTrue(test_cm.get_all_tables() == res) mock_read_sql_query.return_value = pd.DataFrame({'Tables_in_stock': Series()}) test_cm = cmysql.CMySQL(ct.DB_INFO) self.assertTrue(len(test_cm.get_all_tables()) == 0)
def __init__(self, dbinfo = ct.DB_INFO, without_init = False, redis_host = None): self.table = ct.CALENDAR_TABLE self.trigger = ct.SYNCCAL2REDIS self.redis = create_redis_obj() if redis_host is None else create_redis_obj(host = redis_host) self.mysql_client = cmysql.CMySQL(dbinfo, iredis = self.redis) if without_init == False: if not self.init(): raise Exception("calendar table init failed")
def __init__(self, dbinfo=ct.DB_INFO, redis_host=None): self.table = ct.COMBINATION_INFO_TABLE self.logger = getLogger(__name__) self.redis = create_redis_obj( ) if redis_host is None else create_redis_obj(host=redis_host) self.mysql_client = cmysql.CMySQL(dbinfo, iredis=self.redis) if not self.init(): raise Exception("init combination table failed")
def __init__(self, dbinfo): self.table = ct.STOCK_INFO_TABLE self.redis = create_redis_obj() self.mysql_client = cmysql.CMySQL(dbinfo) self.mysql_dbs = self.mysql_client.get_all_databases() #self.trigger = ct.SYNCSTOCK2REDIS #if not self.create(): raise Exception("create stock info table:%s failed" % self.table) if not self.init(): raise Exception("init stock info table failed")
def test_set_get(self, mock_to_sql, mock_read_sql_query): res = [1., 2., 3.] data = pd.DataFrame({'Tables_in_stock': Series(res)}) mock_to_sql.return_value = None mock_read_sql_query.return_value = data test_cm = cmysql.CMySQL(ct.DB_INFO) data1 = test_cm.get("sql") self.assertTrue(test_cm.set(data, "table") == None) self.assertTrue(test_cm.get("sql").equals(data))
def __init__(self, dbinfo, code): self.code = code self.dbname = self.get_dbname(code) self.redis = create_redis_obj() self.data_type_dict = {9: "day"} self.influx_client = CInflux(ct.IN_DB_INFO, self.dbname) self.mysql_client = cmysql.CMySQL(dbinfo, self.dbname) if not self.create(): raise Exception("create combination table failed")
def __init__(self, dbinfo=ct.DB_INFO, redis_host=None, stocks_dir='/data/tdx/history/days', stock_path='/data/tdx/base/stocks.csv'): self.table = ct.STOCK_INFO_TABLE self.redis = create_redis_obj( ) if redis_host is None else create_redis_obj(host=redis_host) self.mysql_client = cmysql.CMySQL(dbinfo, iredis=self.redis) self.mysql_dbs = self.mysql_client.get_all_databases() self.stocks_dir = stocks_dir self.stock_path = stock_path
def __init__(self, dbinfo=ct.DB_INFO, without_init=False): self.table = ct.CALENDAR_TABLE self.trigger = ct.SYNCCAL2REDIS self.mysql_client = cmysql.CMySQL(dbinfo) self.redis = create_redis_obj() if without_init == False: if not self.create(): raise Exception("create calendar table failed") if not self.init(True): raise Exception("calendar table init failed") # here must be first init and second register, for init will delete table # which will delete trigger if not self.register(): raise Exception("create calendar trigger failed")
def __init__(self, dbinfo=ct.DB_INFO, redis_host=None): self.table = ct.INDUSTRY_INFO self.redis = create_redis_obj( ) if redis_host is None else create_redis_obj(redis_host) self.mysql_client = cmysql.CMySQL(dbinfo, iredis=self.redis) if not self.init(): raise Exception("init combination table failed")
def __init__(self, dbinfo): self.table = ct.COMBINATION_INFO_TABLE self.redis = create_redis_obj() self.mysql_client = cmysql.CMySQL(dbinfo) self.mysql_dbs = self.mysql_client.get_all_databases() if not self.init(): raise Exception("init combination table failed")
def __init__(self, dbinfo=ct.DB_INFO, redis_host=None): self.table = ct.STOCK_INFO_TABLE self.redis = create_redis_obj( ) if redis_host is None else create_redis_obj(host=redis_host) self.mysql_client = cmysql.CMySQL(dbinfo, iredis=self.redis) self.mysql_dbs = self.mysql_client.get_all_databases()