def get_hk_dbname(market, direction): dbname = '' if market == 'sse' and direction == 'north': dbname = HkexCrawler.get_dbname(ct.SH_MARKET_SYMBOL, ct.HK_MARKET_SYMBOL) elif market == 'sse' and direction == 'south': dbname = HkexCrawler.get_dbname(ct.HK_MARKET_SYMBOL, ct.SH_MARKET_SYMBOL) elif market == 'szse' and direction == 'south': dbname = HkexCrawler.get_dbname(ct.HK_MARKET_SYMBOL, ct.SZ_MARKET_SYMBOL) else: dbname = HkexCrawler.get_dbname(ct.SZ_MARKET_SYMBOL, ct.HK_MARKET_SYMBOL) return dbname
def __init__(self, item, dbinfo = ct.DB_INFO): super(HkexTradeOverviewPoster, self).__init__(item) self.mysql_reconnect_wait = 60 self.dbname = get_hk_dbname(market = item['market'], direction = item['direction']) self.table = HkexCrawler.get_capital_table(self.dbname) self.connect = pymysql.connect(host=dbinfo['host'], port=dbinfo['port'], db=self.dbname, user=dbinfo['user'], passwd=dbinfo['password'], charset=ct.UTF8) self.cursor = self.connect.cursor()