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()
